| Index: build/android/pylib/base/base_test_runner.py | 
| diff --git a/build/android/pylib/base/base_test_runner.py b/build/android/pylib/base/base_test_runner.py | 
| new file mode 100644 | 
| index 0000000000000000000000000000000000000000..2a7fdd3d6ed6fb5ba26cace9ef9f738c919dfbf3 | 
| --- /dev/null | 
| +++ b/build/android/pylib/base/base_test_runner.py | 
| @@ -0,0 +1,138 @@ | 
| +# 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. | 
| + | 
| +"""Base class for running tests on a single device.""" | 
| + | 
| +# TODO(jbudorick) Deprecate and remove this class and all subclasses after | 
| +# any relevant parts have been ported to the new environment + test instance | 
| +# model. | 
| + | 
| +import logging | 
| + | 
| +from pylib import ports | 
| +from pylib.device import device_utils | 
| +from pylib.forwarder import Forwarder | 
| +from pylib.valgrind_tools import CreateTool | 
| +# TODO(frankf): Move this to pylib/utils | 
| +import lighttpd_server | 
| + | 
| + | 
| +# A file on device to store ports of net test server. The format of the file is | 
| +# test-spawner-server-port:test-server-port | 
| +NET_TEST_SERVER_PORT_INFO_FILE = 'net-test-server-ports' | 
| + | 
| + | 
| +class BaseTestRunner(object): | 
| +  """Base class for running tests on a single device.""" | 
| + | 
| +  def __init__(self, device, tool): | 
| +    """ | 
| +      Args: | 
| +        device: An instance of DeviceUtils that the tests will run on. | 
| +        tool: Name of the Valgrind tool. | 
| +    """ | 
| +    assert isinstance(device, device_utils.DeviceUtils) | 
| +    self.device = device | 
| +    self.device_serial = self.device.adb.GetDeviceSerial() | 
| +    self.tool = CreateTool(tool, self.device) | 
| +    self._http_server = None | 
| +    self._forwarder_device_port = 8000 | 
| +    self.forwarder_base_url = ('http://localhost:%d' % | 
| +        self._forwarder_device_port) | 
| +    # We will allocate port for test server spawner when calling method | 
| +    # LaunchChromeTestServerSpawner and allocate port for test server when | 
| +    # starting it in TestServerThread. | 
| +    self.test_server_spawner_port = 0 | 
| +    self.test_server_port = 0 | 
| + | 
| +  def _PushTestServerPortInfoToDevice(self): | 
| +    """Pushes the latest port information to device.""" | 
| +    self.device.WriteFile( | 
| +        self.device.GetExternalStoragePath() + '/' + | 
| +            NET_TEST_SERVER_PORT_INFO_FILE, | 
| +        '%d:%d' % (self.test_server_spawner_port, self.test_server_port)) | 
| + | 
| +  def RunTest(self, test): | 
| +    """Runs a test. Needs to be overridden. | 
| + | 
| +    Args: | 
| +      test: A test to run. | 
| + | 
| +    Returns: | 
| +      Tuple containing: | 
| +        (base_test_result.TestRunResults, tests to rerun or None) | 
| +    """ | 
| +    raise NotImplementedError | 
| + | 
| +  def InstallTestPackage(self): | 
| +    """Installs the test package once before all tests are run.""" | 
| +    pass | 
| + | 
| +  def SetUp(self): | 
| +    """Run once before all tests are run.""" | 
| +    self.InstallTestPackage() | 
| + | 
| +  def TearDown(self): | 
| +    """Run once after all tests are run.""" | 
| +    self.ShutdownHelperToolsForTestSuite() | 
| + | 
| +  def LaunchTestHttpServer(self, document_root, port=None, | 
| +                           extra_config_contents=None): | 
| +    """Launches an HTTP server to serve HTTP tests. | 
| + | 
| +    Args: | 
| +      document_root: Document root of the HTTP server. | 
| +      port: port on which we want to the http server bind. | 
| +      extra_config_contents: Extra config contents for the HTTP server. | 
| +    """ | 
| +    self._http_server = lighttpd_server.LighttpdServer( | 
| +        document_root, port=port, extra_config_contents=extra_config_contents) | 
| +    if self._http_server.StartupHttpServer(): | 
| +      logging.info('http server started: http://localhost:%s', | 
| +                   self._http_server.port) | 
| +    else: | 
| +      logging.critical('Failed to start http server') | 
| +    self._ForwardPortsForHttpServer() | 
| +    return (self._forwarder_device_port, self._http_server.port) | 
| + | 
| +  def _ForwardPorts(self, port_pairs): | 
| +    """Forwards a port.""" | 
| +    Forwarder.Map(port_pairs, self.device, self.tool) | 
| + | 
| +  def _UnmapPorts(self, port_pairs): | 
| +    """Unmap previously forwarded ports.""" | 
| +    for (device_port, _) in port_pairs: | 
| +      Forwarder.UnmapDevicePort(device_port, self.device) | 
| + | 
| +  # Deprecated: Use ForwardPorts instead. | 
| +  def StartForwarder(self, port_pairs): | 
| +    """Starts TCP traffic forwarding for the given |port_pairs|. | 
| + | 
| +    Args: | 
| +      host_port_pairs: A list of (device_port, local_port) tuples to forward. | 
| +    """ | 
| +    self._ForwardPorts(port_pairs) | 
| + | 
| +  def _ForwardPortsForHttpServer(self): | 
| +    """Starts a forwarder for the HTTP server. | 
| + | 
| +    The forwarder forwards HTTP requests and responses between host and device. | 
| +    """ | 
| +    self._ForwardPorts([(self._forwarder_device_port, self._http_server.port)]) | 
| + | 
| +  def _RestartHttpServerForwarderIfNecessary(self): | 
| +    """Restarts the forwarder if it's not open.""" | 
| +    # Checks to see if the http server port is being used.  If not forwards the | 
| +    # request. | 
| +    # TODO(dtrainor): This is not always reliable because sometimes the port | 
| +    # will be left open even after the forwarder has been killed. | 
| +    if not ports.IsDevicePortUsed(self.device, self._forwarder_device_port): | 
| +      self._ForwardPortsForHttpServer() | 
| + | 
| +  def ShutdownHelperToolsForTestSuite(self): | 
| +    """Shuts down the server and the forwarder.""" | 
| +    if self._http_server: | 
| +      self._UnmapPorts([(self._forwarder_device_port, self._http_server.port)]) | 
| +      self._http_server.ShutdownHttpServer() | 
| + | 
|  |