tests.java.org.python.jsr223.ScriptEngineTest Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of jython Show documentation
Show all versions of jython Show documentation
Jython is an implementation of the high-level, dynamic, object-oriented
language Python written in 100% Pure Java, and seamlessly integrated with
the Java platform. It thus allows you to run Python on any Java platform.
package org.python.jsr223;
import java.io.IOException;
import java.io.StringReader;
import java.math.BigInteger;
import java.util.Arrays;
import javax.script.Bindings;
import javax.script.Compilable;
import javax.script.CompiledScript;
import javax.script.Invocable;
import javax.script.ScriptContext;
import javax.script.ScriptEngine;
import javax.script.ScriptEngineManager;
import javax.script.ScriptException;
import javax.script.SimpleScriptContext;
import org.junit.Assert;
import org.python.core.Options;
import org.python.core.PyList;
import org.python.core.PyString;
import junit.framework.TestCase;
public class ScriptEngineTest extends TestCase {
public void testEvalString() throws ScriptException {
ScriptEngineManager manager = new ScriptEngineManager();
ScriptEngine pythonEngine = manager.getEngineByName("python");
ScriptContext context = pythonEngine.getContext();
context.setAttribute(ScriptEngine.FILENAME, "sample.py", ScriptContext.ENGINE_SCOPE);
assertNull(pythonEngine.eval("x = 5"));
assertEquals(5, pythonEngine.eval("x"));
assertEquals("sample.py", pythonEngine.eval("__file__"));
pythonEngine.eval("import sys");
assertEquals(Arrays.asList("sample.py"), pythonEngine.eval("sys.argv"));
}
public void testEvalStringArgv() throws ScriptException {
ScriptEngineManager manager = new ScriptEngineManager();
ScriptEngine pythonEngine = manager.getEngineByName("python");
ScriptContext context = pythonEngine.getContext();
context.setAttribute(ScriptEngine.FILENAME, "sample.py", ScriptContext.ENGINE_SCOPE);
context.setAttribute(ScriptEngine.ARGV, new String[] {"foo", "bar"}, ScriptContext.ENGINE_SCOPE);
assertNull(pythonEngine.eval("x = 5"));
assertEquals(5, pythonEngine.eval("x"));
assertEquals("sample.py", pythonEngine.eval("__file__"));
pythonEngine.eval("import sys");
assertEquals(Arrays.asList("sample.py", "foo", "bar"), pythonEngine.eval("sys.argv"));
}
public void testEvalStringNoFilenameWithArgv() throws ScriptException {
ScriptEngineManager manager = new ScriptEngineManager();
ScriptEngine pythonEngine = manager.getEngineByName("python");
ScriptContext context = pythonEngine.getContext();
context.setAttribute(ScriptEngine.ARGV, new String[] {"foo", "bar"}, ScriptContext.ENGINE_SCOPE);
assertNull(pythonEngine.eval("x = 5"));
assertEquals(5, pythonEngine.eval("x"));
boolean gotExpectedException = false;
try {
pythonEngine.eval("__file__");
} catch (ScriptException e) {
assertTrue(e.getMessage().startsWith("NameError: "));
gotExpectedException = true;
}
if (!gotExpectedException) {
fail("Excepted __file__ to be undefined");
}
pythonEngine.eval("import sys");
assertEquals(Arrays.asList("foo", "bar"), pythonEngine.eval("sys.argv"));
}
public void testSyntaxError() {
ScriptEngineManager manager = new ScriptEngineManager();
ScriptEngine pythonEngine = manager.getEngineByName("python");
try {
pythonEngine.eval("5q");
} catch (ScriptException e) {
assertEquals(e.getColumnNumber(), 1);
assertEquals(e.getLineNumber(), 1);
assertTrue(e.getMessage().startsWith("SyntaxError: "));
return;
}
assertTrue("Expected a ScriptException", false);
}
public void testPythonException() {
ScriptEngineManager manager = new ScriptEngineManager();
ScriptEngine pythonEngine = manager.getEngineByName("python");
try {
pythonEngine.eval("pass\ndel undefined");
} catch (ScriptException e) {
assertEquals(e.getLineNumber(), 2);
assertTrue(e.getMessage().startsWith("NameError: "));
return;
}
assertTrue("Expected a ScriptException", false);
}
public void testScriptFilename() {
ScriptEngineManager manager = new ScriptEngineManager();
ScriptEngine pythonEngine = manager.getEngineByName("python");
SimpleScriptContext scriptContext = new SimpleScriptContext();
scriptContext.setAttribute(ScriptEngine.FILENAME, "sample.py", ScriptContext.ENGINE_SCOPE);
try {
pythonEngine.eval("foo", scriptContext);
} catch (ScriptException e) {
assertEquals("sample.py", e.getFileName());
return;
}
assertTrue("Expected a ScriptException", false);
}
public void testCompileEvalString() throws ScriptException {
ScriptEngineManager manager = new ScriptEngineManager();
ScriptEngine pythonEngine = manager.getEngineByName("python");
ScriptContext context = pythonEngine.getContext();
context.setAttribute(ScriptEngine.FILENAME, "sample.py", ScriptContext.ENGINE_SCOPE);
CompiledScript five = ((Compilable)pythonEngine).compile("5");
assertEquals(5, five.eval());
assertEquals("sample.py", pythonEngine.eval("__file__"));
pythonEngine.eval("import sys");
assertEquals(Arrays.asList("sample.py"), pythonEngine.eval("sys.argv"));
}
public void testEvalReader() throws ScriptException {
ScriptEngineManager manager = new ScriptEngineManager();
ScriptEngine pythonEngine = manager.getEngineByName("python");
ScriptContext context = pythonEngine.getContext();
context.setAttribute(ScriptEngine.FILENAME, "sample.py", ScriptContext.ENGINE_SCOPE);
assertNull(pythonEngine.eval(new StringReader("x = 5")));
assertEquals(5, pythonEngine.eval(new StringReader("x")));
assertEquals("sample.py", pythonEngine.eval("__file__"));
pythonEngine.eval("import sys");
assertEquals(Arrays.asList("sample.py"), pythonEngine.eval("sys.argv"));
}
public void testCompileEvalReader() throws ScriptException {
ScriptEngineManager manager = new ScriptEngineManager();
ScriptEngine pythonEngine = manager.getEngineByName("python");
ScriptContext context = pythonEngine.getContext();
context.setAttribute(ScriptEngine.FILENAME, "sample.py", ScriptContext.ENGINE_SCOPE);
CompiledScript five = ((Compilable)pythonEngine).compile(new StringReader("5"));
assertEquals(5, five.eval());
assertEquals("sample.py", pythonEngine.eval("__file__"));
pythonEngine.eval("import sys");
assertEquals(Arrays.asList("sample.py"), pythonEngine.eval("sys.argv"));
}
public void testBindings() throws ScriptException {
ScriptEngineManager manager = new ScriptEngineManager();
ScriptEngine pythonEngine = manager.getEngineByName("python");
pythonEngine.put("a", 42);
assertEquals(42, pythonEngine.eval("a"));
assertNull(pythonEngine.eval("x = 5"));
assertEquals(5, pythonEngine.get("x"));
assertNull(pythonEngine.eval("del x"));
assertNull(pythonEngine.get("x"));
}
static class ThreadLocalBindingsTest implements Runnable {
ScriptEngine engine;
int value;
Object x;
Object name;
Throwable exception;
public ThreadLocalBindingsTest(ScriptEngine engine, int value) {
this.engine = engine;
this.value = value;
}
//@formatter:off
static final String script = String.join("\n", new String[] {
"try:",
" a",
"except NameError:",
" pass",
"else:",
" raise Exception('a is defined', a)"});
//@formatter:on
@Override
public void run() {
try {
Bindings bindings = engine.createBindings();
assertNull(engine.eval(script, bindings));
bindings.put("x", value);
x = engine.eval("x", bindings);
name = engine.eval("__name__", bindings);
} catch (Throwable e) {
e.printStackTrace();
exception = e;
}
}
}
/**
* Test that, with the use of a {@code Bindings} argument to {@code ScriptEngine.eval}, the
* interpreter is presented with a distinct name space, whether in a thread or not, and
* that __name__ == "__main__" in the engine-scoped name space (only).
*
* @throws Exception
*/
public void testThreadLocalBindings() throws ScriptException, InterruptedException {
ScriptEngineManager manager = new ScriptEngineManager();
ScriptEngine pythonEngine = manager.getEngineByName("python");
pythonEngine.put("a", 42);
pythonEngine.put("x", 15);
// Examine name space of the engine with Bindings
ThreadLocalBindingsTest test = new ThreadLocalBindingsTest(pythonEngine, -7);
test.run(); // This does not start a thread
assertNull(test.exception);
assertEquals(-7, test.x);
assertEquals("__builtin__", test.name);
// Examine name space of the engine with Bindings and in a thread
ThreadLocalBindingsTest test2 = new ThreadLocalBindingsTest(pythonEngine, -22);
Thread thread = new Thread(test2);
thread.start(); // This *does* start a thread
thread.join();
assertNull(test2.exception);
assertEquals(-22, test2.x);
assertEquals("__builtin__", test2.name);
// Test name space of the pythonEngine without Bindings is unaffected
assertEquals(15, pythonEngine.get("x"));
assertNull(pythonEngine.eval("del x"));
assertNull(pythonEngine.get("x"));
assertEquals("__main__", pythonEngine.eval("__name__"));
}
public void testInvoke() throws ScriptException, NoSuchMethodException {
ScriptEngineManager manager = new ScriptEngineManager();
ScriptEngine pythonEngine = manager.getEngineByName("python");
Invocable invocableEngine = (Invocable)pythonEngine;
assertNull(pythonEngine.eval("def f(x): return abs(x)"));
assertEquals(5, invocableEngine.invokeFunction("f", -5));
assertEquals("spam", invocableEngine.invokeMethod(new PyString(" spam "), "strip"));
assertEquals("spam", invocableEngine.invokeMethod(" spam ", "strip"));
}
public void testInvokeFunctionNoSuchMethod() throws ScriptException {
ScriptEngineManager manager = new ScriptEngineManager();
Invocable invocableEngine = (Invocable)manager.getEngineByName("python");
try {
invocableEngine.invokeFunction("undefined");
} catch (NoSuchMethodException e) {
return;
}
assertTrue("Expected a NoSuchMethodException", false);
}
public void testInvokeMethodNoSuchMethod() throws ScriptException {
ScriptEngineManager manager = new ScriptEngineManager();
Invocable invocableEngine = (Invocable)manager.getEngineByName("python");
try {
invocableEngine.invokeMethod("eggs", "undefined");
fail("Expected a NoSuchMethodException");
} catch (NoSuchMethodException e) {
assertEquals("undefined", e.getMessage());
}
}
public void testGetInterface() throws ScriptException, IOException {
ScriptEngineManager manager = new ScriptEngineManager();
ScriptEngine pythonEngine = manager.getEngineByName("python");
Invocable invocableEngine = (Invocable)pythonEngine;
assertNull(pythonEngine.eval("def read(cb): return 1"));
Readable readable = invocableEngine.getInterface(Readable.class);
assertEquals(1, readable.read(null));
assertNull(pythonEngine.eval("class C(object):\n" + " def read(self, cb): return 2\n"
+ "c = C()"));
readable = invocableEngine.getInterface(pythonEngine.get("c"), Readable.class);
assertEquals(2, readable.read(null));
}
public void testInvokeMethodNoSuchArgs() throws ScriptException, NoSuchMethodException {
ScriptEngineManager manager = new ScriptEngineManager();
ScriptEngine pythonEngine = manager.getEngineByName("python");
Invocable invocableEngine = (Invocable)pythonEngine;
Object newStringCapitalize = invocableEngine.invokeMethod("test", "capitalize");
assertEquals(newStringCapitalize, "Test");
}
public void testPdb() {
ScriptEngineManager manager = new ScriptEngineManager();
ScriptEngine pythonEngine = manager.getEngineByName("python");
// String from issue 1674
String pdbString = "from pdb import set_trace; set_trace()";
try {
pythonEngine.eval(pdbString);
fail("bdb.BdbQuit expected");
} catch (ScriptException e) {
assertTrue(e.getMessage().startsWith("bdb.BdbQuit"));
}
}
// FIXME PyScriptEngineScope lacks items(), iteritems(), and other dict methods
// This should be added in a future release
public void testScope_repr() throws ScriptException {
ScriptEngineManager manager = new ScriptEngineManager();
ScriptEngine pythonEngine = manager.getEngineByName("python");
pythonEngine.eval("a = 4");
pythonEngine.eval("b = 'hi'");
String repr = (String)pythonEngine.eval("repr(locals())");
// locals() contains builtins as of 2.7.0, so we need to selectively test
Assert.assertTrue(repr.contains("'a': 4"));
Assert.assertTrue(repr.contains("'b': u'hi'"));
}
public void testScope_iter() throws ScriptException {
ScriptEngineManager manager = new ScriptEngineManager();
ScriptEngine pythonEngine = manager.getEngineByName("python");
pythonEngine.eval("a = 4");
pythonEngine.eval("b = 'hi'");
PyList locals = (PyList) pythonEngine.eval("sorted((item for item in locals()))");
Assert.assertTrue(locals.contains("a"));
Assert.assertTrue(locals.contains("b"));
Assert.assertTrue(locals.contains("__name__"));
}
public void testScope_lookup() throws ScriptException {
ScriptEngineManager manager = new ScriptEngineManager();
ScriptEngine pythonEngine = manager.getEngineByName("python");
pythonEngine.eval("a = 4");
pythonEngine.eval("b = 'hi'");
pythonEngine.eval("var_a = locals()['a']");
pythonEngine.eval("arepr = repr(var_a)");
assertEquals("4", pythonEngine.get("arepr"));
}
public void testIssue1681() throws ScriptException {
ScriptEngineManager manager = new ScriptEngineManager();
ScriptEngine pythonEngine = manager.getEngineByName("python");
pythonEngine.eval("from org.python.jsr223 import PythonCallable\n"
+ "class MyPythonCallable(PythonCallable):\n"
+ " def getAString(self): return 'a string'\n\n"
+ "result = MyPythonCallable().getAString()\n" //
+ "test = MyPythonCallable()\n" //
+ "result2 = test.getAString()");
assertEquals("a string", pythonEngine.get("result"));
assertEquals("a string", pythonEngine.get("result2"));
}
public void testIssue1698() throws ScriptException {
ScriptEngineManager manager = new ScriptEngineManager();
ScriptEngine pythonEngine = manager.getEngineByName("python");
pythonEngine.eval("import warnings");
// Would previously fail
pythonEngine.eval("warnings.warn('test')");
}
public void testSiteImportedByDefault() throws ScriptException {
ScriptEngineManager manager = new ScriptEngineManager();
ScriptEngine pythonEngine = manager.getEngineByName("python");
pythonEngine.eval("import sys");
pythonEngine.eval("'site' in sys.modules");
}
public void testSiteCanBeNotImported() throws ScriptException {
try {
Options.importSite = false;
ScriptEngineManager manager = new ScriptEngineManager();
ScriptEngine pythonEngine = manager.getEngineByName("python");
pythonEngine.eval("import sys");
pythonEngine.eval("'site' not in sys.modules");
} finally {
Options.importSite = true;
}
}
public void testIssue2090() throws ScriptException {
ScriptEngineManager manager = new ScriptEngineManager();
ScriptEngine pythonEngine = manager.getEngineByName("python");
pythonEngine.eval("a = 10L\n" + "b = a-1");
Object r = pythonEngine.get("b");
assertEquals(new BigInteger("9"), r);
}
}