|
@@ -1,5 +1,8 @@
|
|
|
+import sys
|
|
|
from unittest.case import TestCase
|
|
|
-from oe.utils import packages_filter_out_system, trim_version
|
|
|
+from contextlib import contextmanager
|
|
|
+from io import StringIO
|
|
|
+from oe.utils import packages_filter_out_system, trim_version, multiprocess_launch
|
|
|
|
|
|
class TestPackagesFilterOutSystem(TestCase):
|
|
|
def test_filter(self):
|
|
@@ -49,3 +52,44 @@ class TestTrimVersion(TestCase):
|
|
|
self.assertEqual(trim_version("1.2.3", 2), "1.2")
|
|
|
self.assertEqual(trim_version("1.2.3", 3), "1.2.3")
|
|
|
self.assertEqual(trim_version("1.2.3", 4), "1.2.3")
|
|
|
+
|
|
|
+
|
|
|
+class TestMultiprocessLaunch(TestCase):
|
|
|
+
|
|
|
+ def test_multiprocesslaunch(self):
|
|
|
+ import bb
|
|
|
+
|
|
|
+ def testfunction(item, d):
|
|
|
+ if item == "2" or item == "1":
|
|
|
+ raise KeyError("Invalid number %s" % item)
|
|
|
+ return "Found %s" % item
|
|
|
+
|
|
|
+ def dummyerror(msg):
|
|
|
+ print("ERROR: %s" % msg)
|
|
|
+
|
|
|
+ @contextmanager
|
|
|
+ def captured_output():
|
|
|
+ new_out, new_err = StringIO(), StringIO()
|
|
|
+ old_out, old_err = sys.stdout, sys.stderr
|
|
|
+ try:
|
|
|
+ sys.stdout, sys.stderr = new_out, new_err
|
|
|
+ yield sys.stdout, sys.stderr
|
|
|
+ finally:
|
|
|
+ sys.stdout, sys.stderr = old_out, old_err
|
|
|
+
|
|
|
+ d = bb.data_smart.DataSmart()
|
|
|
+ bb.error = dummyerror
|
|
|
+
|
|
|
+ # Assert the function returns the right results
|
|
|
+ result = multiprocess_launch(testfunction, ["3", "4", "5", "6"], d, extraargs=(d,))
|
|
|
+ self.assertIn("Found 3", result)
|
|
|
+ self.assertIn("Found 4", result)
|
|
|
+ self.assertIn("Found 5", result)
|
|
|
+ self.assertIn("Found 6", result)
|
|
|
+ self.assertEqual(len(result), 4)
|
|
|
+
|
|
|
+ # Assert the function prints exceptions
|
|
|
+ with captured_output() as (out, err):
|
|
|
+ self.assertRaises(bb.BBHandledException, multiprocess_launch, testfunction, ["1", "2", "3", "4", "5", "6"], d, extraargs=(d,))
|
|
|
+ self.assertIn("KeyError: 'Invalid number 1'", out.getvalue())
|
|
|
+ self.assertIn("KeyError: 'Invalid number 2'", out.getvalue())
|