From 8511fb42b6b1b13d7f18db045ae4ddaf8042a147 Mon Sep 17 00:00:00 2001
From: Kevin F <beardog@mailbox.org>
Date: Thu, 12 Jan 2023 21:51:03 -0600
Subject: [PATCH] Added test for multiprocess wrapper

---
 tests/test_multiproc.py | 61 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 61 insertions(+)
 create mode 100644 tests/test_multiproc.py

diff --git a/tests/test_multiproc.py b/tests/test_multiproc.py
new file mode 100644
index 00000000..12d8bd81
--- /dev/null
+++ b/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()