Index: build/android/pylib/base/test_dispatcher_unittest.py |
diff --git a/build/android/pylib/base/test_dispatcher_unittest.py b/build/android/pylib/base/test_dispatcher_unittest.py |
index 253826c03f2e2da0ba20a19da06416c222e23422..88dfda69f01939a1253b8b7739a3207cf3dadcc2 100644 |
--- a/build/android/pylib/base/test_dispatcher_unittest.py |
+++ b/build/android/pylib/base/test_dispatcher_unittest.py |
@@ -167,7 +167,7 @@ class TestShard(unittest.TestCase): |
@staticmethod |
def _RunShard(runner_factory): |
return test_dispatcher.RunTests( |
- ['a', 'b', 'c'], runner_factory, False, None, shard=True) |
+ ['a', 'b', 'c'], runner_factory, ['0', '1'], shard=True) |
def testShard(self): |
results, exit_code = TestShard._RunShard(MockRunner) |
@@ -182,7 +182,7 @@ class TestShard(unittest.TestCase): |
def testNoTests(self): |
results, exit_code = test_dispatcher.RunTests( |
- [], MockRunner, False, None, shard=True) |
+ [], MockRunner, ['0', '1'], shard=True) |
self.assertEqual(len(results.GetAll()), 0) |
self.assertEqual(exit_code, constants.ERROR_EXIT_CODE) |
@@ -192,7 +192,7 @@ class TestReplicate(unittest.TestCase): |
@staticmethod |
def _RunReplicate(runner_factory): |
return test_dispatcher.RunTests( |
- ['a', 'b', 'c'], runner_factory, False, None, shard=False) |
+ ['a', 'b', 'c'], runner_factory, ['0', '1'], shard=False) |
def testReplicate(self): |
results, exit_code = TestReplicate._RunReplicate(MockRunner) |
@@ -208,7 +208,7 @@ class TestReplicate(unittest.TestCase): |
def testNoTests(self): |
results, exit_code = test_dispatcher.RunTests( |
- [], MockRunner, False, None, shard=False) |
+ [], MockRunner, ['0', '1'], shard=False) |
self.assertEqual(len(results.GetAll()), 0) |
self.assertEqual(exit_code, constants.ERROR_EXIT_CODE) |