| Index: build/android/base_test_sharder.py
|
| diff --git a/build/android/base_test_sharder.py b/build/android/base_test_sharder.py
|
| deleted file mode 100644
|
| index 3be45ea81bb27396b5c6d76427ec28a1ef2e83a0..0000000000000000000000000000000000000000
|
| --- a/build/android/base_test_sharder.py
|
| +++ /dev/null
|
| @@ -1,107 +0,0 @@
|
| -# Copyright (c) 2012 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.
|
| -
|
| -
|
| -import logging
|
| -import multiprocessing
|
| -
|
| -from test_result import *
|
| -
|
| -
|
| -def _ShardedTestRunnable(test):
|
| - """Standalone function needed by multiprocessing.Pool."""
|
| - log_format = '[' + test.device + '] # %(asctime)-15s: %(message)s'
|
| - if logging.getLogger().handlers:
|
| - logging.getLogger().handlers[0].setFormatter(logging.Formatter(log_format))
|
| - else:
|
| - logging.basicConfig(format=log_format)
|
| - return test.Run()
|
| -
|
| -
|
| -def SetTestsContainer(tests_container):
|
| - """Sets tests container.
|
| -
|
| - multiprocessing.Queue can't be pickled across processes, so we need to set
|
| - this as a 'global', per process, via multiprocessing.Pool.
|
| - """
|
| - BaseTestSharder.tests_container = tests_container
|
| -
|
| -
|
| -class BaseTestSharder(object):
|
| - """Base class for sharding tests across multiple devices.
|
| -
|
| - Args:
|
| - attached_devices: A list of attached devices.
|
| - """
|
| - # See more in SetTestsContainer.
|
| - tests_container = None
|
| -
|
| - def __init__(self, attached_devices):
|
| - self.attached_devices = attached_devices
|
| - self.retries = 1
|
| - self.tests = []
|
| -
|
| - def CreateShardedTestRunner(self, device, index):
|
| - """Factory function to create a suite-specific test runner.
|
| -
|
| - Args:
|
| - device: Device serial where this shard will run
|
| - index: Index of this device in the pool.
|
| -
|
| - Returns:
|
| - An object of BaseTestRunner type (that can provide a "Run()" method).
|
| - """
|
| - pass
|
| -
|
| - def SetupSharding(self, tests):
|
| - """Called before starting the shards."""
|
| - pass
|
| -
|
| - def OnTestsCompleted(self, test_runners, test_results):
|
| - """Notifies that we completed the tests."""
|
| - pass
|
| -
|
| - def RunShardedTests(self):
|
| - """Runs the tests in all connected devices.
|
| -
|
| - Returns:
|
| - A TestResults object.
|
| - """
|
| - logging.warning('*' * 80)
|
| - logging.warning('Sharding in ' + str(len(self.attached_devices)) +
|
| - ' devices.')
|
| - logging.warning('Note that the output is not synchronized.')
|
| - logging.warning('Look for the "Final result" banner in the end.')
|
| - logging.warning('*' * 80)
|
| - final_results = TestResults()
|
| - for retry in xrange(self.retries):
|
| - logging.warning('Try %d of %d' % (retry + 1, self.retries))
|
| - self.SetupSharding(self.tests)
|
| - test_runners = []
|
| - for index, device in enumerate(self.attached_devices):
|
| - logging.warning('*' * 80)
|
| - logging.warning('Creating shard %d for %s' % (index, device))
|
| - logging.warning('*' * 80)
|
| - test_runner = self.CreateShardedTestRunner(device, index)
|
| - test_runners += [test_runner]
|
| - logging.warning('Starting...')
|
| - pool = multiprocessing.Pool(len(self.attached_devices),
|
| - SetTestsContainer,
|
| - [BaseTestSharder.tests_container])
|
| - results_lists = pool.map(_ShardedTestRunnable, test_runners)
|
| - test_results = TestResults.FromTestResults(results_lists)
|
| - if retry == self.retries - 1:
|
| - all_passed = final_results.ok + test_results.ok
|
| - final_results = test_results
|
| - final_results.ok = all_passed
|
| - break
|
| - else:
|
| - final_results.ok += test_results.ok
|
| - self.tests = []
|
| - for t in test_results.GetAllBroken():
|
| - self.tests += [t.name]
|
| - if not self.tests:
|
| - break
|
| - self.OnTestsCompleted(test_runners, final_results)
|
| - return final_results
|
|
|