| Index: build/android/pylib/host_driven/test_server.py
|
| diff --git a/build/android/pylib/host_driven/test_server.py b/build/android/pylib/host_driven/test_server.py
|
| deleted file mode 100644
|
| index 02560c91293845fa181332298fe333e9e86baf94..0000000000000000000000000000000000000000
|
| --- a/build/android/pylib/host_driven/test_server.py
|
| +++ /dev/null
|
| @@ -1,131 +0,0 @@
|
| -# Copyright 2013 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.
|
| -
|
| -"""Host driven test server controller.
|
| -
|
| -This class controls the startup and shutdown of a python driven test server that
|
| -runs in a separate process.
|
| -
|
| -The server starts up automatically when the object is created.
|
| -
|
| -After it starts up, it is possible to retreive the hostname it started on
|
| -through accessing the member field |host| and the port name through |port|.
|
| -
|
| -For shutting down the server, call TearDown().
|
| -"""
|
| -
|
| -import logging
|
| -import subprocess
|
| -import os
|
| -import os.path
|
| -import time
|
| -import urllib2
|
| -
|
| -from pylib import constants
|
| -from pylib.constants import host_paths
|
| -
|
| -# NOTE: when adding or modifying these lines, omit any leading slashes!
|
| -# Otherwise os.path.join() will (correctly) treat them as absolute paths
|
| -# instead of relative paths, and will do nothing.
|
| -_PYTHONPATH_DIRS = [
|
| - 'net/tools/testserver/',
|
| - 'third_party/',
|
| - 'third_party/pyftpdlib/src/',
|
| - 'third_party/pywebsocket/src',
|
| - 'third_party/tlslite/',
|
| -]
|
| -
|
| -# Python files in these directories are generated as part of the build.
|
| -# These dirs are located in out/(Debug|Release) directory.
|
| -# The correct path is determined based on the build type. E.g. out/Debug for
|
| -# debug builds and out/Release for release builds.
|
| -_GENERATED_PYTHONPATH_DIRS = [
|
| - 'pyproto/policy/proto/',
|
| - 'pyproto/sync/protocol/',
|
| - 'pyproto/'
|
| -]
|
| -
|
| -_TEST_SERVER_HOST = '127.0.0.1'
|
| -# Paths for supported test server executables.
|
| -TEST_NET_SERVER_PATH = 'net/tools/testserver/testserver.py'
|
| -TEST_SYNC_SERVER_PATH = 'sync/tools/testserver/sync_testserver.py'
|
| -TEST_POLICY_SERVER_PATH = 'chrome/browser/policy/test/policy_testserver.py'
|
| -# Parameters to check that the server is up and running.
|
| -TEST_SERVER_CHECK_PARAMS = {
|
| - TEST_NET_SERVER_PATH: {
|
| - 'url_path': '/',
|
| - 'response': 'Default response given for path'
|
| - },
|
| - TEST_SYNC_SERVER_PATH: {
|
| - 'url_path': 'chromiumsync/time',
|
| - 'response': '0123456789'
|
| - },
|
| - TEST_POLICY_SERVER_PATH: {
|
| - 'url_path': 'test/ping',
|
| - 'response': 'Policy server is up.'
|
| - },
|
| -}
|
| -
|
| -class TestServer(object):
|
| - """Sets up a host driven test server on the host machine.
|
| -
|
| - For shutting down the server, call TearDown().
|
| - """
|
| -
|
| - def __init__(self, shard_index, test_server_port, test_server_path,
|
| - test_server_flags=None):
|
| - """Sets up a Python driven test server on the host machine.
|
| -
|
| - Args:
|
| - shard_index: Index of the current shard.
|
| - test_server_port: Port to run the test server on. This is multiplexed with
|
| - the shard index. To retrieve the real port access the
|
| - member variable |port|.
|
| - test_server_path: The path (relative to the root src dir) of the server
|
| - test_server_flags: Optional list of additional flags to the test server
|
| - """
|
| - self.host = _TEST_SERVER_HOST
|
| - self.port = test_server_port + shard_index
|
| -
|
| - src_dir = host_paths.DIR_SOURCE_ROOT
|
| - # Make dirs into a list of absolute paths.
|
| - abs_dirs = [os.path.join(src_dir, d) for d in _PYTHONPATH_DIRS]
|
| - # Add the generated python files to the path
|
| - abs_dirs.extend([os.path.join(src_dir, constants.GetOutDirectory(), d)
|
| - for d in _GENERATED_PYTHONPATH_DIRS])
|
| - current_python_path = os.environ.get('PYTHONPATH')
|
| - extra_python_path = ':'.join(abs_dirs)
|
| - if current_python_path:
|
| - python_path = current_python_path + ':' + extra_python_path
|
| - else:
|
| - python_path = extra_python_path
|
| -
|
| - # NOTE: A separate python process is used to simplify getting the right
|
| - # system path for finding includes.
|
| - test_server_flags = test_server_flags or []
|
| - cmd = ['python', os.path.join(src_dir, test_server_path),
|
| - '--log-to-console',
|
| - ('--host=%s' % self.host),
|
| - ('--port=%d' % self.port),
|
| - '--on-remote-server'] + test_server_flags
|
| - self._test_server_process = subprocess.Popen(
|
| - cmd, env={'PYTHONPATH': python_path})
|
| - test_url = 'http://%s:%d/%s' % (self.host, self.port,
|
| - TEST_SERVER_CHECK_PARAMS[test_server_path]['url_path'])
|
| - expected_response = TEST_SERVER_CHECK_PARAMS[test_server_path]['response']
|
| - retries = 0
|
| - while retries < 5:
|
| - try:
|
| - d = urllib2.urlopen(test_url).read()
|
| - logging.info('URL %s GOT: %s', test_url, d)
|
| - if d.startswith(expected_response):
|
| - break
|
| - except Exception as e: # pylint: disable=broad-except
|
| - logging.info('URL %s GOT: %s', test_url, e)
|
| - time.sleep(retries * 0.1)
|
| - retries += 1
|
| -
|
| - def TearDown(self):
|
| - self._test_server_process.kill()
|
| - self._test_server_process.wait()
|
|
|