[Andoid] Threaded TestRunner creation and SetUp and TearDown calls.

build/android/pylib/base/shard_unittest.py

 # 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.
 
 """Unittests for shard.py."""
 
 import os
 import sys
 import unittest
 
 sys.path.append(os.path.join(os.path.dirname(os.path.realpath(__file__)),
                 os.pardir, os.pardir))
 from pylib import android_commands
 
 import shard
 import test_result
 
 
 class TestException(Exception):
   pass
 
 
 class MockRunner(object):
   """A mock TestRunner."""
   def __init__(self, device='0'):
     self.device = device
+    self.setups = 0
+    self.teardowns = 0
 
-  def Run(self, test):
+  def RunTest(self, test):
     return (test_result.TestResults.FromRun(
                 ok=[test_result.BaseTestResult(test, '')]),
             None)
 
+  def SetUp(self):
+    self.setups += 1
+
+  def TearDown(self):
+    self.teardowns += 1
+
 
 class MockRunnerRetry(MockRunner):
-  def Run(self, test):
+  def RunTest(self, test):
     return (test_result.TestResults.FromRun(
                 failed=[test_result.BaseTestResult(test, '')]),
             test)
 
 
 class MockRunnerException(MockRunner):
-  def Run(self, test):
+  def RunTest(self, test):
     raise TestException
 
 
-class TestWorker(unittest.TestCase):
-  """Tests for shard._Worker."""
+class TestFunctions(unittest.TestCase):
+  """Tests for shard._RunTestsFromQueue."""
   @staticmethod
-  def _RunRunner(mock_runner, tests):
+  def _RunTests(mock_runner, tests):
     results = []
     retry = []
-    worker = shard._Worker(mock_runner, tests, results, retry)
-    worker.start()
-    worker.join()
-    worker.ReraiseIfException()
+    shard._RunTestsFromQueue(mock_runner, tests, results, retry)
     return (results, retry)
 
-  def testRun(self):
-    results, retry = TestWorker._RunRunner(MockRunner(), ['a', 'b'])
+  def testRunTestsFromQueue(self):
+    results, retry = TestFunctions._RunTests(MockRunner(), ['a', 'b'])
     self.assertEqual(len(results), 2)
     self.assertEqual(len(retry), 0)
 
-  def testRetry(self):
-    results, retry = TestWorker._RunRunner(MockRunnerRetry(), ['a', 'b'])
+  def testRunTestsFromQueueRetry(self):
+    results, retry = TestFunctions._RunTests(MockRunnerRetry(), ['a', 'b'])
     self.assertEqual(len(results), 2)
     self.assertEqual(len(retry), 2)
 
-  def testReraise(self):
-    with self.assertRaises(TestException):
-      TestWorker._RunRunner(MockRunnerException(), ['a', 'b'])
+  def testSetUp(self):
+    runners = []
+    shard._SetUp(MockRunner, '0', runners)
+    self.assertEqual(len(runners), 1)
+    self.assertEqual(runners[0].setups, 1)
 
 
-class TestRunAllTests(unittest.TestCase):
+class TestThreadGroupFunctions(unittest.TestCase):
   """Tests for shard._RunAllTests and shard._CreateRunners."""
   def setUp(self):
     self.tests = ['a', 'b', 'c', 'd', 'e', 'f', 'g']
 
+  def testCreate(self):
+    runners = shard._CreateRunners(MockRunner, ['0', '1'])
+    for runner in runners:
+      self.assertEqual(runner.setups, 1)
+
   def testRun(self):
-    runners = shard._CreateRunners(MockRunner, ['0', '1'])
+    runners = [MockRunner('0'), MockRunner('1')]
     results, retry = shard._RunAllTests(runners, self.tests)
     self.assertEqual(len(results), len(self.tests))
     self.assertEqual(len(retry), 0)
 
+  def testTearDown(self):
+    runners = [MockRunner('0'), MockRunner('1')]
+    shard._TearDownRunners(runners)
+    for runner in runners:
+      self.assertEqual(runner.teardowns, 1)
+
   def testRetry(self):
     runners = shard._CreateRunners(MockRunnerRetry, ['0', '1'])
     results, retry = shard._RunAllTests(runners, self.tests)
     self.assertEqual(len(results), len(self.tests))
     self.assertEqual(len(retry), len(self.tests))
 
   def testReraise(self):
     runners = shard._CreateRunners(MockRunnerException, ['0', '1'])
     with self.assertRaises(TestException):
       shard._RunAllTests(runners, self.tests)
@@ skipping 13 matching lines @@
     self.assertEqual(len(results.ok), 3)
 
   def testFailing(self):
     results = TestShard._RunShard(MockRunnerRetry)
     self.assertEqual(len(results.ok), 0)
     self.assertEqual(len(results.failed), 3)
 
 
 if __name__ == '__main__':
   unittest.main()