Support HTTP test-server based net unit tests on Android.

build/android/pylib/base_test_sharder.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.
 
 
 import android_commands
 import logging
 import multiprocessing
 
 from android_commands import errors
+from forwarder import Forwarder
 from test_result import TestResults
 
 
 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)
   # Handle SystemExit here since python has a bug to exit current process
   try:
     return test.Run()
   except SystemExit:
     return TestResults()
 
+
 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):
+  def __init__(self, attached_devices, build_type='Debug'):
     self.attached_devices = attached_devices
     # Worst case scenario: a device will drop offline per run, so we need
     # to retry until we're out of devices.
     self.retries = len(self.attached_devices)
     self.tests = []
+    self.build_type = build_type
 
   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
+    Forwarder.KillHost(self.build_type)
 
   def OnTestsCompleted(self, test_runners, test_results):
     """Notifies that we completed the tests."""
-    pass
+    Forwarder.KillHost(self.build_type)
 
   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.')
@@ skipping 52 matching lines @@
         for t in test_results.GetAllBroken():
           self.tests += [t.name]
         if not self.tests:
           break
     else:
       # We ran out retries, possibly out of healthy devices.
       # There's no recovery at this point.
       raise Exception('Unrecoverable error while retrying test runs.')
     self.OnTestsCompleted(test_runners, final_results)
     return final_results