allow unit tests to specify code as Source objects, add test for multiple executions of the same source object

Author: lukasstadler

graalpython/com.oracle.graal.python.test/src/com/oracle/graal/python/test/PythonTests.java

@@ -192,8 +192,15 @@ public static void assertPrints(String expected, Path scriptName) {
     public static void assertPrints(String expected, String code) {
         final ByteArrayOutputStream byteArray = new ByteArrayOutputStream();
         final PrintStream printStream = new PrintStream(byteArray);
-        String source = code;
-        PythonTests.runScript(new String[0], source, printStream, System.err);
+        PythonTests.runScript(new String[0], code, printStream, System.err);
+        String result = byteArray.toString().replaceAll("\r\n", "\n");
+        assertEquals(expected.replaceAll(" at 0x[0-9a-f]*>", " at 0xabcd>"), result.replaceAll(" at 0x[0-9a-f]*>", " at 0xabcd>"));
+    }
+
+    public static void assertPrints(String expected, org.graalvm.polyglot.Source code) {
+        final ByteArrayOutputStream byteArray = new ByteArrayOutputStream();
+        final PrintStream printStream = new PrintStream(byteArray);
+        PythonTests.runScript(new String[0], code, printStream, System.err);
         String result = byteArray.toString().replaceAll("\r\n", "\n");
         assertEquals(expected.replaceAll(" at 0x[0-9a-f]*>", " at 0xabcd>"), result.replaceAll(" at 0x[0-9a-f]*>", " at 0xabcd>"));
     }
@@ -311,6 +318,15 @@ public static void runScript(String[] args, String source, OutputStream out, Out
         }
     }
 
+    public static void runScript(String[] args, org.graalvm.polyglot.Source source, OutputStream out, OutputStream err) {
+        try {
+            enterContext(args);
+            context.eval(source);
+        } finally {
+            flush(out, err);
+        }
+    }
+
     public static void runScript(String[] args, String source, OutputStream out, OutputStream err, Runnable cb) {
         try {
             enterContext(args);

graalpython/com.oracle.graal.python.test/src/com/oracle/graal/python/test/basic/HelloWorldTests.java

@@ -34,4 +34,11 @@ public class HelloWorldTests {
     public void helloworld() {
         assertPrints("hello world\n", "print(\"hello world\")");
     }
+
+    @Test
+    public void helloworldAgain() {
+        org.graalvm.polyglot.Source source = org.graalvm.polyglot.Source.create("python", "try: print(value)\nexcept:print('hello')\nvalue='world'");
+        assertPrints("hello\n", source);
+        assertPrints("hello\n", source);
+    }
 }