Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(1009)

Unified Diff: build/android/pylib/base_test_runner.py

Issue 11879031: [Android] Move base tests classes to pylib/base (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: Fixed nits Created 7 years, 11 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « build/android/pylib/base/test_result.py ('k') | build/android/pylib/base_test_sharder.py » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: build/android/pylib/base_test_runner.py
diff --git a/build/android/pylib/base_test_runner.py b/build/android/pylib/base_test_runner.py
deleted file mode 100644
index 5355633282658f15820e5890e704b89b67419f42..0000000000000000000000000000000000000000
--- a/build/android/pylib/base_test_runner.py
+++ /dev/null
@@ -1,210 +0,0 @@
-# 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 contextlib
-import httplib
-import logging
-import os
-import tempfile
-import time
-
-import android_commands
-import constants
-from chrome_test_server_spawner import SpawningServer
-import constants
-from flag_changer import FlagChanger
-from forwarder import Forwarder
-import lighttpd_server
-import ports
-from valgrind_tools import CreateTool
-
-
-# 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.
-
- 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, tool, shard_index, build_type):
- """
- 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.
- build_type: 'Release' or 'Debug'.
- """
- self.device = device
- self.adb = android_commands.AndroidCommands(device=device)
- self.tool = CreateTool(tool, self.adb)
- self._http_server = None
- self._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
- self.flags.AddFlags(['--disable-fre'])
- self._spawning_server = None
- self._spawner_forwarder = None
- # 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
- self.build_type = build_type
-
- def _PushTestServerPortInfoToDevice(self):
- """Pushes the latest port information to device."""
- self.adb.SetFileContents(self.adb.GetExternalStorage() + '/' +
- NET_TEST_SERVER_PORT_INFO_FILE,
- '%d:%d' % (self.test_server_spawner_port,
- self.test_server_port))
-
- def Run(self):
- """Calls subclass functions to set up tests, run them and tear them down.
-
- Returns:
- Test results returned from RunTests().
- """
- if not self.HasTests():
- return True
- self.SetUp()
- try:
- return self.RunTests()
- finally:
- self.TearDown()
-
- def SetUp(self):
- """Called before tests run."""
- Forwarder.KillDevice(self.adb)
-
- def HasTests(self):
- """Whether the test suite has tests to run."""
- return True
-
- 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
- |CHROME_DIR|.
- dest_dir: Absolute path to copy to on the device.
- """
- for p in test_data_paths:
- self.adb.PushIfNeeded(
- os.path.join(constants.CHROME_DIR, p),
- os.path.join(dest_dir, p))
-
- 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.StartForwarderForHttpServer()
- return (self._forwarder_device_port, self._http_server.port)
-
- def _CreateAndRunForwarder(
- self, adb, port_pairs, tool, host_name, build_type):
- """Creates and run a forwarder."""
- forwarder = Forwarder(adb, build_type)
- forwarder.Run(port_pairs, tool, host_name)
- return forwarder
-
- 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.
- """
- if self._forwarder:
- self._forwarder.Close()
- self._forwarder = self._CreateAndRunForwarder(
- self.adb, port_pairs, self.tool, '127.0.0.1', self.build_type)
-
- def StartForwarderForHttpServer(self):
- """Starts a forwarder for the HTTP server.
-
- The forwarder forwards HTTP requests and responses between host and device.
- """
- self.StartForwarder([(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.adb,
- self._forwarder_device_port):
- self.StartForwarderForHttpServer()
-
- def ShutdownHelperToolsForTestSuite(self):
- """Shuts down the server and the forwarder."""
- # Forwarders should be killed before the actual servers they're forwarding
- # to as they are clients potentially with open connections and to allow for
- # proper hand-shake/shutdown.
- Forwarder.KillDevice(self.adb)
- if self._http_server:
- self._http_server.ShutdownHttpServer()
- if self._spawning_server:
- self._spawning_server.Stop()
- self.flags.Restore()
-
- def LaunchChromeTestServerSpawner(self):
- """Launches test server spawner."""
- server_ready = False
- error_msgs = []
- # Try 3 times to launch test spawner server.
- for i in xrange(0, 3):
- # Do not allocate port for test server here. We will allocate
- # different port for individual test in TestServerThread.
- self.test_server_spawner_port = ports.AllocateTestServerPort()
- self._spawning_server = SpawningServer(self.test_server_spawner_port,
- self.adb,
- self.tool,
- self.build_type)
- self._spawning_server.Start()
- server_ready, error_msg = ports.IsHttpServerConnectable(
- '127.0.0.1', self.test_server_spawner_port, path='/ping',
- expected_read='ready')
- if server_ready:
- break
- else:
- error_msgs.append(error_msg)
- self._spawning_server.Stop()
- # Wait for 2 seconds then restart.
- time.sleep(2)
- if not server_ready:
- logging.error(';'.join(error_msgs))
- raise Exception('Can not start the test spawner server.')
- self._PushTestServerPortInfoToDevice()
- self._spawner_forwarder = self._CreateAndRunForwarder(
- self.adb,
- [(self.test_server_spawner_port, self.test_server_spawner_port)],
- self.tool, '127.0.0.1', self.build_type)
« no previous file with comments | « build/android/pylib/base/test_result.py ('k') | build/android/pylib/base_test_sharder.py » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698