Added test for multiprocess wrapper

tests/test_multiproc.py

@@ -0,0 +1,61 @@
+#!/usr/bin/env python3
+import sys, os
+import time
+sys.path.append(".")
+sys.path.append("src/")
+import uuid
+TEST_DIR = 'testdata/%s-%s' % (uuid.uuid4(), os.path.basename(__file__)) + '/'
+print("Test directory:", TEST_DIR)
+os.environ["ONIONR_HOME"] = TEST_DIR
+import unittest, json
+
+from utils import identifyhome, createdirs
+from onionrsetup import setup_config
+from utils import multiproc
+createdirs.create_dirs()
+setup_config()
+
+class TestMultiProc(unittest.TestCase):
+    def test_list_args(self):
+        answer = multiproc.subprocess_compute(sum, 10, [1, 3])
+        self.assertEqual(answer, 4)
+    
+    def test_two_args(self):
+        def _add(a, b):
+            return a + b
+        answer = multiproc.subprocess_compute(_add, 10, 1, 3)
+        self.assertEqual(answer, 4)
+    
+    def test_kwargs(self):
+        def _add(a=0, b=0):
+            return a + b
+        answer = multiproc.subprocess_compute(_add, 10, a=1, b=3)
+        self.assertEqual(answer, 4)
+    
+    def test_exception(self):
+        def _fail():
+            raise Exception("This always fails")
+        with self.assertRaises(ChildProcessError):
+            multiproc.subprocess_compute(_fail, 10)
+    
+    def test_delayed_exception(self):
+        def _fail():
+            time.sleep(3)
+            raise Exception("This always fails")
+        with self.assertRaises(ChildProcessError):
+            multiproc.subprocess_compute(_fail, 10)
+    
+    def test_timeout(self):
+        def _sleep():
+            time.sleep(3)
+        with self.assertRaises(TimeoutError):
+            multiproc.subprocess_compute(_sleep, 1)
+    
+    def test_timeout_disabled(self):
+        def _sleep():
+            time.sleep(3)
+        self.assertIsNone(multiproc.subprocess_compute(_sleep, -1))
+        self.assertIsNone(multiproc.subprocess_compute(_sleep, 0))
+
+
+unittest.main()