src/third_party/mozjs/js/src/jsapi-tests/testEnclosingFunction.cpp

/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 4 -*-
 * vim: set ts=8 sts=4 et sw=4 tw=99:
 *
 * Test script cloning.
 */
/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#include "tests.h"
#include "jsfriendapi.h"
#include "jsdbgapi.h"

using namespace js;

JSScript *found = NULL;

JSBool
CheckEnclosing(JSContext *cx, unsigned argc, Value *vp)
{
    CallArgs args = CallArgsFromVp(argc, vp);

    found = js::GetOutermostEnclosingFunctionOfScriptedCaller(cx);

    args.rval().set(UndefinedValue());
    return true;
}

BEGIN_TEST(test_enclosingFunction)
{
    CHECK(JS_DefineFunction(cx, global, "checkEnclosing", CheckEnclosing, 0, 0));

    EXEC("checkEnclosing()");
    CHECK(found == NULL);

    RootedFunction fun(cx);

    const char s1chars[] = "checkEnclosing()";
    fun = JS_CompileFunction(cx, global, "s1", 0, NULL, s1chars, strlen(s1chars), __FILE__, __LINE__);
    CHECK(fun);
    EXEC("s1()");
    CHECK(found == JS_GetFunctionScript(cx, fun));

    const char s2chars[] = "return function() { checkEnclosing() }";
    fun = JS_CompileFunction(cx, global, "s2", 0, NULL, s2chars, strlen(s2chars), __FILE__, __LINE__);
    CHECK(fun);
    EXEC("s2()()");
    CHECK(found == JS_GetFunctionScript(cx, fun));

    const char s3chars[] = "return function() { let (x) { function g() { checkEnclosing() } return g() } }";
    fun = JS_CompileFunction(cx, global, "s3", 0, NULL, s3chars, strlen(s3chars), __FILE__, __LINE__);
    CHECK(fun);
    EXEC("s3()()");
    CHECK(found == JS_GetFunctionScript(cx, fun));

    return true;
}
END_TEST(test_enclosingFunction)