| 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()
|
| +
|
|
|