62 lines
1.8 KiB
Python
62 lines
1.8 KiB
Python
|
#!/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()
|