Index: build/android/pylib/utils/parallelizer_test.py |
diff --git a/build/android/pylib/utils/parallelizer_test.py b/build/android/pylib/utils/parallelizer_test.py |
new file mode 100644 |
index 0000000000000000000000000000000000000000..6e0c7e79a5873cfd0ea4c585480454b32fc13161 |
--- /dev/null |
+++ b/build/android/pylib/utils/parallelizer_test.py |
@@ -0,0 +1,166 @@ |
+# Copyright 2014 The Chromium Authors. All rights reserved. |
+# Use of this source code is governed by a BSD-style license that can be |
+# found in the LICENSE file. |
+ |
+"""Unit tests for the contents of parallelizer.py.""" |
+ |
+# pylint: disable=W0212 |
+# pylint: disable=W0613 |
+ |
+import os |
+import tempfile |
+import time |
+import unittest |
+ |
+from pylib.utils import parallelizer |
+ |
+ |
+class ParallelizerTestObject(object): |
+ """Class used to test parallelizer.Parallelizer.""" |
+ |
+ parallel = parallelizer.Parallelizer |
+ |
+ def __init__(self, thing, completion_file_name=None): |
+ self._thing = thing |
+ self._completion_file_name = completion_file_name |
+ self.helper = ParallelizerTestObjectHelper(thing) |
+ |
+ @staticmethod |
+ def doReturn(what): |
+ return what |
+ |
+ @classmethod |
+ def doRaise(cls, what): |
+ raise what |
+ |
+ def doSetTheThing(self, new_thing): |
+ self._thing = new_thing |
+ |
+ def doReturnTheThing(self): |
+ return self._thing |
+ |
+ def doRaiseTheThing(self): |
+ raise self._thing |
+ |
+ def doRaiseIfExceptionElseSleepFor(self, sleep_duration): |
+ if isinstance(self._thing, Exception): |
+ raise self._thing |
+ time.sleep(sleep_duration) |
+ self._write_completion_file() |
+ return self._thing |
+ |
+ def _write_completion_file(self): |
+ if self._completion_file_name and len(self._completion_file_name): |
+ with open(self._completion_file_name, 'w+b') as completion_file: |
+ completion_file.write('complete') |
+ |
+ def __getitem__(self, index): |
+ return self._thing[index] |
+ |
+ def __str__(self): |
+ return type(self).__name__ |
+ |
+ |
+class ParallelizerTestObjectHelper(object): |
+ |
+ def __init__(self, thing): |
+ self._thing = thing |
+ |
+ def doReturnStringThing(self): |
+ return str(self._thing) |
+ |
+ |
+class ParallelizerTest(unittest.TestCase): |
+ |
+ def testInitWithNone(self): |
+ with self.assertRaises(AssertionError): |
+ parallelizer.Parallelizer(None) |
+ |
+ def testInitEmptyList(self): |
+ with self.assertRaises(AssertionError): |
+ parallelizer.Parallelizer([]) |
+ |
+ def testMethodCall(self): |
+ test_data = ['abc_foo', 'def_foo', 'ghi_foo'] |
+ expected = ['abc_bar', 'def_bar', 'ghi_bar'] |
+ r = parallelizer.Parallelizer(test_data).replace('_foo', '_bar').pGet(0.1) |
+ self.assertEquals(expected, r) |
+ |
+ def testMutate(self): |
+ devices = [ParallelizerTestObject(True) for _ in xrange(0, 10)] |
+ self.assertTrue(all(d.doReturnTheThing() for d in devices)) |
+ ParallelizerTestObject.parallel(devices).doSetTheThing(False).pFinish(1) |
+ self.assertTrue(not any(d.doReturnTheThing() for d in devices)) |
+ |
+ def testAllReturn(self): |
+ devices = [ParallelizerTestObject(True) for _ in xrange(0, 10)] |
+ results = ParallelizerTestObject.parallel( |
+ devices).doReturnTheThing().pGet(1) |
+ self.assertTrue(isinstance(results, list)) |
+ self.assertEquals(10, len(results)) |
+ self.assertTrue(all(results)) |
+ |
+ def testAllRaise(self): |
+ devices = [ParallelizerTestObject(Exception('thing %d' % i)) |
+ for i in xrange(0, 10)] |
+ p = ParallelizerTestObject.parallel(devices).doRaiseTheThing() |
+ with self.assertRaises(Exception): |
+ p.pGet(1) |
+ |
+ def testOneFailOthersComplete(self): |
+ parallel_device_count = 10 |
+ exception_index = 7 |
+ exception_msg = 'thing %d' % exception_index |
+ |
+ try: |
+ completion_files = [tempfile.NamedTemporaryFile(delete=False) |
+ for _ in xrange(0, parallel_device_count)] |
+ devices = [ |
+ ParallelizerTestObject( |
+ i if i != exception_index else Exception(exception_msg), |
+ completion_files[i].name) |
+ for i in xrange(0, parallel_device_count)] |
+ for f in completion_files: |
+ f.close() |
+ p = ParallelizerTestObject.parallel(devices) |
+ with self.assertRaises(Exception) as e: |
+ p.doRaiseIfExceptionElseSleepFor(2).pGet(3) |
+ self.assertTrue(exception_msg in str(e.exception)) |
+ for i in xrange(0, parallel_device_count): |
+ with open(completion_files[i].name) as f: |
+ if i == exception_index: |
+ self.assertEquals('', f.read()) |
+ else: |
+ self.assertEquals('complete', f.read()) |
+ finally: |
+ for f in completion_files: |
+ os.remove(f.name) |
+ |
+ def testReusable(self): |
+ devices = [ParallelizerTestObject(True) for _ in xrange(0, 10)] |
+ p = ParallelizerTestObject.parallel(devices) |
+ results = p.doReturn(True).pGet(1) |
+ self.assertTrue(all(results)) |
+ results = p.doReturn(True).pGet(1) |
+ self.assertTrue(all(results)) |
+ with self.assertRaises(Exception): |
+ results = p.doRaise(Exception('reusableTest')).pGet(1) |
+ |
+ def testContained(self): |
+ devices = [ParallelizerTestObject(i) for i in xrange(0, 10)] |
+ results = (ParallelizerTestObject.parallel(devices).helper |
+ .doReturnStringThing().pGet(1)) |
+ self.assertTrue(isinstance(results, list)) |
+ self.assertEquals(10, len(results)) |
+ for i in xrange(0, 10): |
+ self.assertEquals(str(i), results[i]) |
+ |
+ def testGetItem(self): |
+ devices = [ParallelizerTestObject(range(i, i+10)) for i in xrange(0, 10)] |
+ results = ParallelizerTestObject.parallel(devices)[9].pGet(1) |
+ self.assertEquals(range(9, 19), results) |
+ |
+ |
+if __name__ == '__main__': |
+ unittest.main(verbosity=2) |
+ |