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 |
deleted file mode 100755 |
index 186a0721d25022cae7596ec3d0eeb2e3861df6a3..0000000000000000000000000000000000000000 |
--- a/build/android/pylib/base/test_dispatcher_unittest.py |
+++ /dev/null |
@@ -1,241 +0,0 @@ |
-#!/usr/bin/env python |
-# Copyright 2013 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. |
- |
-"""Unittests for test_dispatcher.py.""" |
- |
-# pylint: disable=no-self-use |
-# pylint: disable=protected-access |
- |
-import unittest |
- |
-from pylib.base import base_test_result |
-from pylib.base import test_collection |
-from pylib.base import test_dispatcher |
-from pylib.constants import host_paths |
- |
-with host_paths.SysPath(host_paths.DEVIL_PATH): |
- from devil.android import device_utils |
- from devil.android.sdk import adb_wrapper |
- from devil.constants import exit_codes |
- from devil.utils import watchdog_timer |
- |
-with host_paths.SysPath(host_paths.PYMOCK_PATH): |
- import mock # pylint: disable=import-error |
- |
- |
-class TestException(Exception): |
- pass |
- |
- |
-def _MockDevice(serial): |
- d = mock.MagicMock(spec=device_utils.DeviceUtils) |
- d.__str__.return_value = serial |
- d.adb = mock.MagicMock(spec=adb_wrapper.AdbWrapper) |
- d.adb.GetDeviceSerial = mock.MagicMock(return_value=serial) |
- d.IsOnline = mock.MagicMock(return_value=True) |
- return d |
- |
- |
-class MockRunner(object): |
- """A mock TestRunner.""" |
- def __init__(self, device=None, shard_index=0): |
- self.device = device or _MockDevice('0') |
- self.device_serial = self.device.adb.GetDeviceSerial() |
- self.shard_index = shard_index |
- self.setups = 0 |
- self.teardowns = 0 |
- |
- def RunTest(self, test): |
- results = base_test_result.TestRunResults() |
- results.AddResult( |
- base_test_result.BaseTestResult(test, base_test_result.ResultType.PASS)) |
- return (results, None) |
- |
- def SetUp(self): |
- self.setups += 1 |
- |
- def TearDown(self): |
- self.teardowns += 1 |
- |
- |
-class MockRunnerFail(MockRunner): |
- def RunTest(self, test): |
- results = base_test_result.TestRunResults() |
- results.AddResult( |
- base_test_result.BaseTestResult(test, base_test_result.ResultType.FAIL)) |
- return (results, test) |
- |
- |
-class MockRunnerFailTwice(MockRunner): |
- def __init__(self, device=None, shard_index=0): |
- super(MockRunnerFailTwice, self).__init__(device, shard_index) |
- self._fails = 0 |
- |
- def RunTest(self, test): |
- self._fails += 1 |
- results = base_test_result.TestRunResults() |
- if self._fails <= 2: |
- results.AddResult(base_test_result.BaseTestResult( |
- test, base_test_result.ResultType.FAIL)) |
- return (results, test) |
- else: |
- results.AddResult(base_test_result.BaseTestResult( |
- test, base_test_result.ResultType.PASS)) |
- return (results, None) |
- |
- |
-class MockRunnerException(MockRunner): |
- def RunTest(self, test): |
- raise TestException |
- |
- |
-class TestFunctions(unittest.TestCase): |
- """Tests test_dispatcher._RunTestsFromQueue.""" |
- @staticmethod |
- def _RunTests(mock_runner, tests): |
- results = [] |
- tests = test_collection.TestCollection( |
- [test_dispatcher._Test(t) for t in tests]) |
- test_dispatcher._RunTestsFromQueue(mock_runner, tests, results, |
- watchdog_timer.WatchdogTimer(None), 2) |
- run_results = base_test_result.TestRunResults() |
- for r in results: |
- run_results.AddTestRunResults(r) |
- return run_results |
- |
- def testRunTestsFromQueue(self): |
- results = TestFunctions._RunTests(MockRunner(), ['a', 'b']) |
- self.assertEqual(len(results.GetPass()), 2) |
- self.assertEqual(len(results.GetNotPass()), 0) |
- |
- def testRunTestsFromQueueRetry(self): |
- results = TestFunctions._RunTests(MockRunnerFail(), ['a', 'b']) |
- self.assertEqual(len(results.GetPass()), 0) |
- self.assertEqual(len(results.GetFail()), 2) |
- |
- def testRunTestsFromQueueFailTwice(self): |
- results = TestFunctions._RunTests(MockRunnerFailTwice(), ['a', 'b']) |
- self.assertEqual(len(results.GetPass()), 2) |
- self.assertEqual(len(results.GetNotPass()), 0) |
- |
- def testSetUp(self): |
- runners = [] |
- counter = test_dispatcher._ThreadSafeCounter() |
- test_dispatcher._SetUp(MockRunner, _MockDevice('0'), runners, counter) |
- self.assertEqual(len(runners), 1) |
- self.assertEqual(runners[0].setups, 1) |
- |
- def testThreadSafeCounter(self): |
- counter = test_dispatcher._ThreadSafeCounter() |
- for i in xrange(5): |
- self.assertEqual(counter.GetAndIncrement(), i) |
- |
- def testApplyMaxPerRun(self): |
- self.assertEqual( |
- ['A:B', 'C:D', 'E', 'F:G', 'H:I'], |
- test_dispatcher.ApplyMaxPerRun(['A:B', 'C:D:E', 'F:G:H:I'], 2)) |
- |
- |
-class TestThreadGroupFunctions(unittest.TestCase): |
- """Tests test_dispatcher._RunAllTests and test_dispatcher._CreateRunners.""" |
- def setUp(self): |
- self.tests = ['a', 'b', 'c', 'd', 'e', 'f', 'g'] |
- shared_test_collection = test_collection.TestCollection( |
- [test_dispatcher._Test(t) for t in self.tests]) |
- self.test_collection_factory = lambda: shared_test_collection |
- |
- def testCreate(self): |
- runners = test_dispatcher._CreateRunners( |
- MockRunner, [_MockDevice('0'), _MockDevice('1')]) |
- for runner in runners: |
- self.assertEqual(runner.setups, 1) |
- self.assertEqual(set([r.device_serial for r in runners]), |
- set(['0', '1'])) |
- self.assertEqual(set([r.shard_index for r in runners]), |
- set([0, 1])) |
- |
- def testRun(self): |
- runners = [MockRunner(_MockDevice('0')), MockRunner(_MockDevice('1'))] |
- results, exit_code = test_dispatcher._RunAllTests( |
- runners, self.test_collection_factory, 0) |
- self.assertEqual(len(results.GetPass()), len(self.tests)) |
- self.assertEqual(exit_code, 0) |
- |
- def testTearDown(self): |
- runners = [MockRunner(_MockDevice('0')), MockRunner(_MockDevice('1'))] |
- test_dispatcher._TearDownRunners(runners) |
- for runner in runners: |
- self.assertEqual(runner.teardowns, 1) |
- |
- def testRetry(self): |
- runners = test_dispatcher._CreateRunners( |
- MockRunnerFail, [_MockDevice('0'), _MockDevice('1')]) |
- results, exit_code = test_dispatcher._RunAllTests( |
- runners, self.test_collection_factory, 0) |
- self.assertEqual(len(results.GetFail()), len(self.tests)) |
- self.assertEqual(exit_code, exit_codes.ERROR) |
- |
- def testReraise(self): |
- runners = test_dispatcher._CreateRunners( |
- MockRunnerException, [_MockDevice('0'), _MockDevice('1')]) |
- with self.assertRaises(TestException): |
- test_dispatcher._RunAllTests(runners, self.test_collection_factory, 0) |
- |
- |
-class TestShard(unittest.TestCase): |
- """Tests test_dispatcher.RunTests with sharding.""" |
- @staticmethod |
- def _RunShard(runner_factory): |
- return test_dispatcher.RunTests( |
- ['a', 'b', 'c'], runner_factory, [_MockDevice('0'), _MockDevice('1')], |
- shard=True) |
- |
- def testShard(self): |
- results, exit_code = TestShard._RunShard(MockRunner) |
- self.assertEqual(len(results.GetPass()), 3) |
- self.assertEqual(exit_code, 0) |
- |
- def testFailing(self): |
- results, exit_code = TestShard._RunShard(MockRunnerFail) |
- self.assertEqual(len(results.GetPass()), 0) |
- self.assertEqual(len(results.GetFail()), 3) |
- self.assertEqual(exit_code, exit_codes.ERROR) |
- |
- def testNoTests(self): |
- results, exit_code = test_dispatcher.RunTests( |
- [], MockRunner, [_MockDevice('0'), _MockDevice('1')], shard=True) |
- self.assertEqual(len(results.GetAll()), 0) |
- self.assertEqual(exit_code, exit_codes.ERROR) |
- |
- |
-class TestReplicate(unittest.TestCase): |
- """Tests test_dispatcher.RunTests with replication.""" |
- @staticmethod |
- def _RunReplicate(runner_factory): |
- return test_dispatcher.RunTests( |
- ['a', 'b', 'c'], runner_factory, [_MockDevice('0'), _MockDevice('1')], |
- shard=False) |
- |
- def testReplicate(self): |
- results, exit_code = TestReplicate._RunReplicate(MockRunner) |
- # We expect 6 results since each test should have been run on every device |
- self.assertEqual(len(results.GetPass()), 6) |
- self.assertEqual(exit_code, 0) |
- |
- def testFailing(self): |
- results, exit_code = TestReplicate._RunReplicate(MockRunnerFail) |
- self.assertEqual(len(results.GetPass()), 0) |
- self.assertEqual(len(results.GetFail()), 6) |
- self.assertEqual(exit_code, exit_codes.ERROR) |
- |
- def testNoTests(self): |
- results, exit_code = test_dispatcher.RunTests( |
- [], MockRunner, [_MockDevice('0'), _MockDevice('1')], shard=False) |
- self.assertEqual(len(results.GetAll()), 0) |
- self.assertEqual(exit_code, exit_codes.ERROR) |
- |
- |
-if __name__ == '__main__': |
- unittest.main() |