package org.mozilla.javascript.tests;
import org.mozilla.javascript.Context;
import org.mozilla.javascript.ContextFactory;
import org.mozilla.javascript.Script;
import org.mozilla.javascript.Scriptable;
import junit.framework.TestCase;
/**
* Takes care that the name of the method generated for a function "looks like" the original function name.
* See https://bugzilla.mozilla.org/show_bug.cgi?id=460726
* @author Marc Guillemot
*/
public class GeneratedMethodNameTest extends TestCase
{
public void testStandardFunction() throws Exception {
final String scriptCode = "function myFunc() {\n"
+ " var m = javaNameGetter.readCurrentFunctionJavaName();\n"
+ " if (m != 'myFunc') throw 'got ' + m;"
+ "}\n"
+ "myFunc();";
doTest(scriptCode);
}
public void testFunctionDollar() throws Exception {
final String scriptCode = "function $() {\n"
+ " var m = javaNameGetter.readCurrentFunctionJavaName();\n"
+ " if (m != '$') throw 'got ' + m;"
+ "}\n"
+ "$();";
doTest(scriptCode);
}
public void testScriptName() throws Exception {
final String scriptCode =
"var m = javaNameGetter.readCurrentFunctionJavaName();\n"
+ "if (m != 'script') throw 'got ' + m;";
doTest(scriptCode);
}
public void testConstructor() throws Exception {
final String scriptCode = "function myFunc() {\n"
+ " var m = javaNameGetter.readCurrentFunctionJavaName();\n"
+ " if (m != 'myFunc') throw 'got ' + m;"
+ "}\n"
+ "new myFunc();";
doTest(scriptCode);
}
public void testAnonymousFunction() throws Exception {
final String scriptCode = "var myFunc = function() {\n"
+ " var m = javaNameGetter.readCurrentFunctionJavaName();\n"
+ " if (m != 'anonymous') throw 'got ' + m;"
+ "}\n"
+ "myFunc();";
doTest(scriptCode);
}
public class JavaNameGetter {
public String readCurrentFunctionJavaName() {
final Throwable t = new RuntimeException();
// remove prefix and suffix of method name
return t.getStackTrace()[8].getMethodName().
replaceFirst("_[^_]*_(.*)_[^_]*", "$1");
}
}
public void doTest(final String scriptCode) throws Exception {
final Context cx = ContextFactory.getGlobal().enterContext();
try {
Scriptable topScope = cx.initStandardObjects();
topScope.put("javaNameGetter", topScope, new JavaNameGetter());
Script script = cx.compileString(scriptCode, "myScript", 1, null);
script.exec(cx, topScope);
} finally {
Context.exit();
}
}
}