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 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)
@@ skipping 14 matching lines @@
 
 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
     self.retries = 1
     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(None, self.build_type).KillHostForwarder()

[felipeg, 2012/10/30 22:37:30]

make it static

 
   def OnTestsCompleted(self, test_runners, test_results):
     """Notifies that we completed the tests."""
-    pass
+    Forwarder(None, self.build_type).KillHostForwarder()
 
   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 36 matching lines @@
         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