package com.oracle.truffle.js.scriptengine.test;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail;
import static org.junit.Assume.assumeFalse;
import java.util.function.Predicate;
import javax.script.Bindings;
import javax.script.ScriptContext;
import javax.script.ScriptEngine;
import javax.script.ScriptEngineManager;
import javax.script.ScriptException;
import javax.script.SimpleBindings;
import org.graalvm.polyglot.HostAccess;
import org.junit.Test;
import com.oracle.truffle.js.scriptengine.GraalJSScriptEngine;
public class TestBindings {
private static final String SCRIPT_CONTEXT_GLOBAL_BINDINGS_IMPORT_FUNCTION_NAME = "importScriptEngineGlobalBindings";
private static final String argsName = "arguments";
private static final String[] defaultArgs = new String[]{"arg0", "arg1"};
private static final String varName = "boundVarName";
private static final String defaultVarValue = "default";
private static final String updatedVarValue = "updated";
private final ScriptEngineManager manager = new ScriptEngineManager();
private ScriptEngine getEngine() {
return manager.getEngineByName(TestEngine.TESTED_ENGINE_NAME);
}
@Test
public void enginePut() throws ScriptException {
ScriptEngine engine = getEngine();
engine.put(varName, defaultVarValue);
boolean result = (boolean) engine.eval(varName + " === '" + defaultVarValue + "';");
assertTrue(result);
}
@Test
public void bindingsPut() throws ScriptException {
ScriptEngine engine = getEngine();
Bindings bindings = engine.getBindings(ScriptContext.ENGINE_SCOPE);
bindings.put(varName, defaultVarValue);
boolean result = (boolean) engine.eval(varName + " === '" + defaultVarValue + "';");
assertTrue(result);
}
@Test
public void bindingsPutGlobal() throws ScriptException {
ScriptEngine engine = getEngine();
Bindings bindings = engine.getBindings(ScriptContext.GLOBAL_SCOPE);
bindings.put(varName, defaultVarValue);
boolean result = (boolean) engine.eval(varName + " === '" + defaultVarValue + "';");
assertTrue(result);
}
@Test
public void bindingsUpdateGlobal() throws ScriptException {
ScriptEngine engine = getEngine();
Bindings bindings = engine.getBindings(ScriptContext.GLOBAL_SCOPE);
bindings.put(varName, defaultVarValue);
boolean result = (boolean) engine.eval(varName + " === '" + defaultVarValue + "';");
assertTrue(result);
bindings.put(varName, updatedVarValue);
result = (boolean) engine.eval(varName + " === '" + updatedVarValue + "';");
assertTrue(result);
}
@Test
public void bindingsRemoveGlobal() throws ScriptException {
ScriptEngine engine = getEngine();
Bindings bindings = engine.getBindings(ScriptContext.GLOBAL_SCOPE);
bindings.put(varName, defaultVarValue);
boolean result = (boolean) engine.eval(varName + " === '" + defaultVarValue + "';");
assertTrue(result);
bindings.remove(varName);
result = (boolean) engine.eval("typeof " + varName + " === 'undefined';");
assertTrue(result);
}
@Test
public void updateEngineBindings() throws ScriptException {
ScriptEngine engine = getEngine();
Bindings bindings = engine.getBindings(ScriptContext.ENGINE_SCOPE);
bindings.put(varName, defaultVarValue);
engine.eval(varName + " = '" + updatedVarValue + "';");
assertEquals(updatedVarValue, bindings.get(varName));
assertEquals(updatedVarValue, engine.get(varName));
}
@Test
public void updateGlobalBindings() throws ScriptException {
ScriptEngine engine = getEngine();
Bindings bindings = engine.getBindings(ScriptContext.GLOBAL_SCOPE);
bindings.put(varName, defaultVarValue);
engine.eval(varName + " = '" + updatedVarValue + "';");
assertEquals(defaultVarValue, bindings.get(varName));
assertEquals(updatedVarValue, engine.get(varName));
}
@Test
public void overrideEngineBindings1() throws ScriptException {
ScriptEngine engine = getEngine();
engine.eval("var " + varName + " = '" + defaultVarValue + "';");
engine.put(varName, updatedVarValue);
assertEquals(updatedVarValue, engine.eval(varName));
}
@Test
public void overrideEngineBindings2() throws ScriptException {
ScriptEngine engine = getEngine();
engine.eval("var " + varName + " = '" + defaultVarValue + "';");
Bindings bindings = new SimpleBindings();
bindings.put(varName, updatedVarValue);
engine.setBindings(bindings, ScriptContext.ENGINE_SCOPE);
assertEquals(updatedVarValue, engine.eval(varName));
}
@Test
public void overrideGlobalBindings() throws ScriptException {
ScriptEngine engine = getEngine();
engine.eval("var " + varName + " = '" + defaultVarValue + "';");
engine.getBindings(ScriptContext.GLOBAL_SCOPE).put(varName, updatedVarValue);
assertEquals(defaultVarValue, engine.eval(varName));
}
@Test
public void setBindings1() throws ScriptException {
Bindings bindings = new SimpleBindings();
bindings.put(varName, defaultVarValue);
ScriptEngine engine = getEngine();
engine.setBindings(bindings, ScriptContext.ENGINE_SCOPE);
assertEquals(true, engine.eval(varName + " === '" + defaultVarValue + "';"));
}
@Test
public void setBindings2() throws ScriptException {
Bindings bindings = new SimpleBindings();
bindings.put(varName, defaultVarValue);
ScriptEngine engine = getEngine();
engine.setBindings(bindings, ScriptContext.ENGINE_SCOPE);
engine.eval(varName + " = '" + updatedVarValue + "';");
assertEquals(bindings.get(varName), defaultVarValue);
assertEquals(engine.get(varName), defaultVarValue);
}
@Test
public void setBindings3() throws ScriptException {
ScriptEngine engine = getEngine();
engine.put(varName, null);
assertNull(engine.get(varName));
assertNull(engine.eval(varName));
assertEquals(true, engine.eval(varName + " === null;"));
}
@Test
public void engineGetDeclared() throws ScriptException {
ScriptEngine engine = getEngine();
Bindings bindings = engine.getBindings(ScriptContext.ENGINE_SCOPE);
int sizeBefore = bindings.size();
engine.eval("var " + varName + " = '" + defaultVarValue + "';");
assertEquals(defaultVarValue, engine.get(varName));
int sizeAfter = bindings.size();
assertEquals(1, sizeAfter - sizeBefore);
}
@Test
public void clearBindings1() {
ScriptEngine engine = getEngine();
Bindings bindings = engine.getBindings(ScriptContext.ENGINE_SCOPE);
int sizeBeforePut = bindings.size();
bindings.put(varName, defaultVarValue);
int sizeAfterPut = bindings.size();
assertEquals(1, sizeAfterPut - sizeBeforePut);
assertEquals(defaultVarValue, bindings.get(varName));
int sizeBeforeClear = bindings.size();
bindings.clear();
int sizeAfterClear = bindings.size();
assertEquals("One binding cleared", 1, sizeBeforeClear - sizeAfterClear);
assertEquals(null, bindings.get(varName));
}
@Test
public void clearBindings2() throws ScriptException {
ScriptEngine engine = getEngine();
Bindings bindings = engine.getBindings(ScriptContext.ENGINE_SCOPE);
int sizeBeforePut = bindings.size();
bindings.put(varName, defaultVarValue);
int sizeAfterPut = bindings.size();
assertEquals(1, sizeAfterPut - sizeBeforePut);
engine.eval("var " + varName + " = '" + updatedVarValue + "';");
assertEquals(sizeAfterPut, bindings.size());
assertEquals(updatedVarValue, bindings.get(varName));
int sizeBeforeClear = bindings.size();
bindings.clear();
int sizeAfterClear = bindings.size();
assertEquals("One binding cleared", 1, sizeBeforeClear - sizeAfterClear);
assertEquals(null, bindings.get(varName));
}
@Test
public void clearNoBindings() throws ScriptException {
ScriptEngine engine = getEngine();
engine.eval("var " + varName + " = '" + defaultVarValue + "';");
Bindings bindings = engine.getBindings(ScriptContext.ENGINE_SCOPE);
int sizeBeforeClear = bindings.size();
bindings.clear();
int sizeAfterClear = bindings.size();
assertEquals("No bindings cleared", sizeBeforeClear, sizeAfterClear);
assertEquals(defaultVarValue, bindings.get(varName));
}
@Test
public void engineOverrideGlobal() throws ScriptException {
ScriptEngine engine = getEngine();
Bindings engineB = engine.createBindings();
Bindings globalB = engine.createBindings();
engineB.put(varName, updatedVarValue);
globalB.put(varName, defaultVarValue);
engine.setBindings(engineB, ScriptContext.ENGINE_SCOPE);
engine.setBindings(globalB, ScriptContext.GLOBAL_SCOPE);
assertEquals(updatedVarValue, engine.eval(varName));
}
@Test
public void setArgs1() throws ScriptException {
ScriptEngine engine = getEngine();
engine.put("polyglot.js.allowHostAccess", true);
engine.put(argsName, defaultArgs);
assertEquals(defaultArgs[0], engine.eval(argsName + "[0];"));
assertEquals(defaultArgs[1], engine.eval(argsName + "[1];"));
assertEquals(true, engine.eval("(function() { for (var idx in this) { if (idx == '" + argsName + "') { return false; } }; return true; })();"));
}
@Test
public void setArgs2() throws ScriptException {
ScriptEngine engine = getEngine();
Bindings bindings = new SimpleBindings();
bindings.put("polyglot.js.allowHostAccess", true);
bindings.put(argsName, defaultArgs);
engine.setBindings(bindings, ScriptContext.ENGINE_SCOPE);
assertEquals(defaultArgs[0], engine.eval(argsName + "[0];"));
assertEquals(defaultArgs[1], engine.eval(argsName + "[1];"));
}
@Test
public void importFunctionInvisible() throws ScriptException {
ScriptEngine engine = getEngine();
boolean result = (boolean) engine.eval("typeof " + SCRIPT_CONTEXT_GLOBAL_BINDINGS_IMPORT_FUNCTION_NAME + " === 'undefined';");
assertTrue(result);
}
@Test
public void importFunctionOverwrite() throws ScriptException {
ScriptEngine engine = getEngine();
Bindings bindings = engine.getBindings(ScriptContext.ENGINE_SCOPE);
bindings.put(SCRIPT_CONTEXT_GLOBAL_BINDINGS_IMPORT_FUNCTION_NAME, defaultVarValue);
Bindings globalBindings = engine.getBindings(ScriptContext.GLOBAL_SCOPE);
globalBindings.put(varName, defaultVarValue);
boolean result = (boolean) engine.eval(SCRIPT_CONTEXT_GLOBAL_BINDINGS_IMPORT_FUNCTION_NAME + " === '" + defaultVarValue + "';");
assertTrue(result);
result = (boolean) engine.eval(varName + " === '" + defaultVarValue + "';");
assertTrue(result);
}
@Test
public void testJSMapInScope() throws ScriptException {
ScriptEngine engine = getEngine();
Bindings global = engine.getBindings(ScriptContext.GLOBAL_SCOPE);
engine.eval("jsObj = {a: 1}", global);
Object globalResult = engine.eval("JSON.stringify(jsObj)", global);
assertNotNull(globalResult);
}
@Test(expected = ScriptException.class)
public void testNoHostAccessByDefault() throws ScriptException {
assumeFalse(Boolean.getBoolean("polyglot.js.nashorn-compat"));
ScriptEngine engine = getEngine();
Bindings bindings = engine.getBindings(ScriptContext.ENGINE_SCOPE);
bindings.put("javaObj", new Object());
engine.eval("(javaObj instanceof Java.type('java.lang.Object'));");
fail();
}
@Test
public void testSetContextOptionsViaGraalJSBindings() throws ScriptException {
ScriptEngine engine = getEngine();
Bindings bindings = engine.getBindings(ScriptContext.ENGINE_SCOPE);
bindings.put("polyglot.js.allowHostAccess", true);
bindings.put("polyglot.js.allowHostClassLookup", true);
bindings.put("javaObj", new Object());
assertTrue((boolean) engine.eval("(javaObj instanceof Java.type('java.lang.Object'));"));
}
@Test
public void testSetContextOptionsViaGraalJSBindings2() throws ScriptException {
ScriptEngine engine = getEngine();
Bindings bindings = engine.getBindings(ScriptContext.ENGINE_SCOPE);
bindings.put("polyglot.js.allowHostAccess", true);
bindings.put("polyglot.js.allowHostClassLookup", (Predicate<String>) s -> true);
bindings.put("javaObj", new Object());
assertTrue((boolean) engine.eval("(javaObj instanceof Java.type('java.lang.Object'));"));
}
@Test(expected = IllegalStateException.class)
public void testTrySetContextOptionsAfterInitViaGraalJSBindings1() throws ScriptException {
ScriptEngine engine = getEngine();
engine.eval("dummy = 42;");
Bindings bindings = engine.getBindings(ScriptContext.ENGINE_SCOPE);
bindings.put("polyglot.js.allowHostAccess", true);
fail();
}
@Test(expected = IllegalStateException.class)
public void testTrySetContextOptionsAfterInitViaGraalJSBindings2() {
ScriptEngine engine = getEngine();
Bindings bindings = engine.getBindings(ScriptContext.ENGINE_SCOPE);
bindings.put("javaObj", new Object());
bindings.put("polyglot.js.allowHostAccess", true);
fail();
}
@Test
public void testSetContextOptionsViaSimpleBindings() throws ScriptException {
ScriptEngine engine = getEngine();
Bindings bindings = new SimpleBindings();
bindings.put("polyglot.js.allowHostAccess", true);
bindings.put("polyglot.js.allowHostClassLookup", true);
bindings.put("javaObj", new Object());
assertTrue((boolean) engine.eval("(javaObj instanceof Java.type('java.lang.Object'));", bindings));
}
@Test(expected = IllegalStateException.class)
public void testTrySetContextOptionsAfterInitViaSimpleBindings() throws ScriptException {
ScriptEngine engine = getEngine();
Bindings bindings = new SimpleBindings();
engine.eval("dummy = 42;", bindings);
bindings.put("polyglot.js.allowHostAccess", true);
bindings.put("polyglot.js.allowHostClassLookup", true);
engine.eval("(javaObj instanceof Java.type('java.lang.Object'));", bindings);
fail();
}
@Test
public void testSetContextBuilder() throws ScriptException {
ScriptEngine engine = GraalJSScriptEngine.create(null, TestUtil.newContextBuilder().allowHostAccess(HostAccess.ALL).allowHostClassLookup(s -> true));
engine.put("javaObj", new Object());
assertTrue((boolean) engine.eval("(javaObj instanceof Java.type('java.lang.Object'));"));
}
@Test
public void testEnableNashornCompat() throws ScriptException {
ScriptEngine engine = getEngine();
Bindings bindings = engine.getBindings(ScriptContext.ENGINE_SCOPE);
bindings.put("polyglot.js.nashorn-compat", true);
bindings.put("javaObj", new Object());
assertTrue((boolean) engine.eval("(javaObj instanceof Java.type('java.lang.Object'));", bindings));
engine.eval("JavaImporter");
}
@Test
public void globalPutEngineGet() {
ScriptEngine engine = getEngine();
Bindings globalBindings = engine.getBindings(ScriptContext.GLOBAL_SCOPE);
Bindings engineBindings = engine.getBindings(ScriptContext.ENGINE_SCOPE);
globalBindings.put("foo", "bar");
assertEquals("bar", engineBindings.get("foo"));
}
}