Upstream test sharder.

build/android/base_test_runner.py

 # Copyright (c) 2011 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 os
 
 import android_commands
 from chrome_test_server_spawner import SpawningServer
 from flag_changer import FlagChanger
 import lighttpd_server
 import run_tests_helper
 
 FORWARDER_PATH = '/data/local/tmp/forwarder'
 # These ports must match up with the constants in net/test/test_server.cc
 TEST_SERVER_SPAWNER_PORT = 8001
 TEST_SERVER_PORT = 8002
 TEST_SYNC_SERVER_PORT = 8003
 
 
 class BaseTestRunner(object):
-  """Base class for running tests on a single device."""
+  """Base class for running tests on a single device.
 
-  def __init__(self, device):
+  A subclass should implement RunTests() with no parameter, so that calling
+  the Run() method will set up tests, run them and tear them down.
+  """
+
+  def __init__(self, device, shard_index):
     """
       Args:
         device: Tests will run on the device of this ID.
+        shard_index: Index number of the shard on which the test suite will run.
     """
     self.device = device
     self.adb = android_commands.AndroidCommands(device=device)
     # Synchronize date/time between host and device. Otherwise same file on
     # host and device may have different timestamp which may cause
     # AndroidCommands.PushIfNeeded failed, or a test which may compare timestamp
     # got from http head and local time could be failed.
     self.adb.SynchronizeDateTime()
     self._http_server = None
     self._forwarder = None
     self._spawning_server = None
     self._spawner_forwarder = None
     self._forwarder_device_port = 8000
     self.forwarder_base_url = ('http://localhost:%d' %
         self._forwarder_device_port)
     self.flags = FlagChanger(self.adb)
+    self.shard_index = shard_index
 
-  def RunTests(self):
-    # TODO(bulach): this should actually do SetUp / RunTestsInternal / TearDown.
-    # Refactor the various subclasses to expose a RunTestsInternal without
-    # any params.
-    raise NotImplementedError
+  def Run(self):
+    """Calls subclass functions to set up tests, run them and tear them down.
+
+    Returns:
+      Test results returned from RunTests().
+    """
+    self.SetUp()
+    try:
+      return self.RunTests()
+    finally:
+      self.TearDown()
 
   def SetUp(self):
     """Called before tests run."""
     pass
 
+  def RunTests(self):
+    """Runs the tests. Need to be overridden."""
+    raise NotImplementedError
+
   def TearDown(self):
     """Called when tests finish running."""
     self.ShutdownHelperToolsForTestSuite()
 
   def CopyTestData(self, test_data_paths, dest_dir):
     """Copies |test_data_paths| list of files/directories to |dest_dir|.
 
     Args:
       test_data_paths: A list of files or directories relative to |dest_dir|
           which should be copied to the device. The paths must exist in
@@ skipping 72 matching lines @@
     """Launches test server spawner."""
     self._spawning_server = SpawningServer(TEST_SERVER_SPAWNER_PORT,
                                           TEST_SERVER_PORT)
     self._spawning_server.Start()
     # TODO(yfriedman): Ideally we'll only try to start up a port forwarder if
     # there isn't one already running but for now we just get an error message
     # and the existing forwarder still works.
     self._spawner_forwarder = run_tests_helper.ForwardDevicePorts(
         self.adb, [(TEST_SERVER_SPAWNER_PORT, TEST_SERVER_SPAWNER_PORT),
                    (TEST_SERVER_PORT, TEST_SERVER_PORT)])