Add test for ThreadPool.

Author: fangerer

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

@@ -45,6 +45,8 @@
 import java.nio.file.Files;
 import java.nio.file.Path;
 import java.nio.file.Paths;
+import java.util.Collections;
+import java.util.Map;
 
 import org.graalvm.polyglot.Context;
 import org.graalvm.polyglot.Engine;
@@ -97,10 +99,14 @@ public class PythonTests {
     }
 
     public static void enterContext(String... newArgs) {
+        enterContext(Collections.emptyMap(), newArgs);
+    }
+
+    public static void enterContext(Map<String, String> options, String[] args) {
         PythonTests.outArray.reset();
         PythonTests.errArray.reset();
         Context prevContext = context;
-        context = Context.newBuilder().engine(engine).allowExperimentalOptions(true).allowAllAccess(true).arguments("python", newArgs).option("python.Executable", executable).build();
+        context = Context.newBuilder().engine(engine).allowExperimentalOptions(true).allowAllAccess(true).options(options).arguments("python", args).option("python.Executable", executable).build();
         context.initialize("python");
         if (prevContext != null) {
             closeContext(prevContext);
@@ -333,6 +339,15 @@ public static Value runScript(String[] args, File path, OutputStream out, Output
         }
     }
 
+    public static Value runScript(Map<String, String> options, String[] args, String source, OutputStream out, OutputStream err) {
+        try {
+            enterContext(options, args);
+            return context.eval(org.graalvm.polyglot.Source.create("python", source));
+        } finally {
+            flush(out, err);
+        }
+    }
+
     public static Value runScript(String[] args, String source, OutputStream out, OutputStream err) {
         try {
             enterContext(args);
@@ -352,8 +367,12 @@ public static Value runScript(String[] args, org.graalvm.polyglot.Source source,
     }
 
     public static Value runScript(String[] args, String source, OutputStream out, OutputStream err, Runnable cb) {
+        return runScript(Collections.emptyMap(), args, source, out, err, cb);
+    }
+
+    public static Value runScript(Map<String, String> options, String[] args, String source, OutputStream out, OutputStream err, Runnable cb) {
         try {
-            enterContext(args);
+            enterContext(options, args);
             return context.eval(org.graalvm.polyglot.Source.create("python", source));
         } finally {
             cb.run();

graalpython/com.oracle.graal.python.test/src/com/oracle/graal/python/test/module/ThreadPoolTests.java

@@ -0,0 +1,77 @@
+/*
+ * Copyright (c) 2020, Oracle and/or its affiliates. All rights reserved.
+ * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
+ *
+ * The Universal Permissive License (UPL), Version 1.0
+ *
+ * Subject to the condition set forth below, permission is hereby granted to any
+ * person obtaining a copy of this software, associated documentation and/or
+ * data (collectively the "Software"), free of charge and under any and all
+ * copyright rights in the Software, and any and all patent rights owned or
+ * freely licensable by each licensor hereunder covering either (i) the
+ * unmodified Software as contributed to or provided by such licensor, or (ii)
+ * the Larger Works (as defined below), to deal in both
+ *
+ * (a) the Software, and
+ *
+ * (b) any piece of software and/or hardware listed in the lrgrwrks.txt file if
+ * one is included with the Software each a "Larger Work" to which the Software
+ * is contributed by such licensors),
+ *
+ * without restriction, including without limitation the rights to copy, create
+ * derivative works of, display, perform, and distribute the Software and make,
+ * use, sell, offer for sale, import, export, have made, and have sold the
+ * Software and the Larger Work(s), and to sublicense the foregoing rights on
+ * either these or other terms.
+ *
+ * This license is subject to the following condition:
+ *
+ * The above copyright notice and either this complete permission notice or at a
+ * minimum a reference to the UPL must be included in all copies or substantial
+ * portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+ * SOFTWARE.
+ */
+package com.oracle.graal.python.test.module;
+
+import static org.junit.Assert.assertEquals;
+
+import java.io.ByteArrayOutputStream;
+import java.io.PrintStream;
+import java.util.Collections;
+
+import org.junit.Test;
+
+import com.oracle.graal.python.test.PythonTests;
+
+public class ThreadPoolTests {
+
+    @Test
+    public void threadPool() {
+        String source = "import _sysconfig\n" +
+                        "assert _sysconfig.get_config_vars().get('WITH_THREAD'), 'context was not started for threading'\n" +
+                        "from multiprocessing.pool import ThreadPool\n" +
+                        "\n" +
+                        "def fun(item):\n" +
+                        "    return item != None\n" +
+                        "\n" +
+                        "items = list(range(0, 10))\n" +
+                        "\n" +
+                        "pool = ThreadPool(2)\n" +
+                        "res = list(pool.imap(fun, items))\n" +
+                        "pool.close()\n" +
+                        "\n" +
+                        "print(res)\n";
+        final ByteArrayOutputStream byteArray = new ByteArrayOutputStream();
+        final PrintStream printStream = new PrintStream(byteArray);
+        PythonTests.runScript(Collections.singletonMap("python.WithThread", "true"), new String[0], source, printStream, System.err, () -> PythonTests.closeContext());
+        String result = byteArray.toString().replaceAll("\r\n", "\n");
+        assertEquals("[True, True, True, True, True, True, True, True, True, True]\n", result);
+    }
+}