[Android] Converts host driven tests to common test_dispatcher

build/android/pylib/host_driven/setup.py

Index: build/android/pylib/host_driven/setup.py
diff --git a/build/android/pylib/host_driven/run_python_tests.py b/build/android/pylib/host_driven/setup.py
similarity index 29%
rename from build/android/pylib/host_driven/run_python_tests.py
rename to build/android/pylib/host_driven/setup.py
index 09f2aa10c45eaca6d8afe9d50d6608f30c8e93fa..f460d2d235eaa60d99418e54ba7f4def1ae34c19 100644
--- a/build/android/pylib/host_driven/run_python_tests.py
+++ b/build/android/pylib/host_driven/setup.py
@@ -2,23 +2,16 @@
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
-"""Runs the Python tests (relies on using the Java test runner)."""
+"""Setup for instrumentation host-driven tests."""
 
 import logging
 import os
 import sys
 import types
 
-from pylib import android_commands
-from pylib.base import base_test_result
-from pylib.instrumentation import test_options
-from pylib.instrumentation import test_package
-from pylib.instrumentation import test_runner
-from pylib.utils import report_results
-
-import python_test_base
-from python_test_sharder import PythonTestSharder
-from test_info_collection import TestInfoCollection
+import test_case
+import test_info_collection
+import test_runner
 
 
 def _GetPythonFiles(root, files):
@@ -29,7 +22,7 @@ def _GetPythonFiles(root, files):
     files: A list of file names.
 
   Returns:
-    A list with all Python driven test file paths.
+    A list with all python files that match the testing naming scheme.
   """
   return [os.path.join(root, f) for f in files if f.endswith('Test.py')]
 
@@ -40,7 +33,7 @@ def _InferImportNameFromFile(python_file):
   Example: /usr/foo/bar/baz.py -> baz.
 
   Args:
-    python_file: path to the Python file, ostensibly to import later.
+    python_file: Path to the Python file, ostensibly to import later.
 
   Returns:
     The module name for the given file.
@@ -48,118 +41,43 @@ def _InferImportNameFromFile(python_file):
   return os.path.splitext(os.path.basename(python_file))[0]
 
 
-def DispatchPythonTests(options):
-  """Dispatches the Python tests. If there are multiple devices, use sharding.
-
-  Args:
-    options: command line options.
-
-  Returns:
-    A tuple of (base_test_result.TestRunResults object, exit code)
-
-  Raises:
-    Exception: If there are no attached devices.
-  """
-
-  attached_devices = android_commands.GetAttachedDevices()
-  if not attached_devices:
-    raise Exception('You have no devices attached or visible!')
-  if options.test_device:
-    attached_devices = [options.test_device]
-
-  test_collection = TestInfoCollection()
-  all_tests = _GetAllTests(options.python_test_root, options.official_build)
-  test_collection.AddTests(all_tests)
-  test_names = [t.qualified_name for t in all_tests]
-  logging.debug('All available tests: ' + str(test_names))
+def _GetTestModules(host_driven_test_root, is_official_build):
+  """Retrieve a list of python modules that match the testing naming scheme.
 
-  available_tests = test_collection.GetAvailableTests(
-      options.annotations, options.exclude_annotations, options.test_filter)
-
-  if not available_tests:
-    logging.warning('No Python tests to run with current args.')
-    return (base_test_result.TestRunResults(), 0)
-
-  test_names = [t.qualified_name for t in available_tests]
-  logging.debug('Final list of tests to run: ' + str(test_names))
-
-  # Copy files to each device before running any tests.
-  for device_id in attached_devices:
-    logging.debug('Pushing files to device %s', device_id)
-    test_pkg = test_package.TestPackage(options.test_apk_path,
-                                        options.test_apk_jar_path)
-    instrumentation_options = test_options.InstrumentationOptions(
-        options.build_type,
-        options.tool,
-        options.cleanup_test_files,
-        options.push_deps,
-        options.annotations,
-        options.exclude_annotations,
-        options.test_filter,
-        options.test_data,
-        options.save_perf_json,
-        options.screenshot_failures,
-        options.disable_assertions,
-        options.wait_for_debugger,
-        options.test_apk,
-        options.test_apk_path,
-        options.test_apk_jar_path)
-    test_files_copier = test_runner.TestRunner(instrumentation_options,
-                                               device_id, 0, test_pkg, [])
-    test_files_copier.InstallTestPackage()
-    if options.push_deps:
-      logging.info('Pushing data deps to device.')
-      test_files_copier.PushDataDeps()
-    else:
-      logging.warning('Skipping pushing data deps to device.')
-
-  # Actually run the tests.
-  if len(attached_devices) > 1 and options.wait_for_debugger:
-    logging.warning('Debugger can not be sharded, '
-                    'using first available device')
-    attached_devices = attached_devices[:1]
-  logging.debug('Running Python tests')
-  sharder = PythonTestSharder(attached_devices, available_tests, options)
-  test_results = sharder.RunShardedTests()
-
-  if not test_results.DidRunPass():
-    return (test_results, 1)
-
-  return (test_results, 0)
-
-
-def _GetTestModules(python_test_root, is_official_build):
-  """Retrieve a sorted list of pythonDrivenTests.
-
-  Walks the location of pythonDrivenTests, imports them, and provides the list
+  Walks the location of host-driven tests, imports them, and provides the list
   of imported modules to the caller.
 
   Args:
-    python_test_root: the path to walk, looking for pythonDrivenTests
-    is_official_build: whether to run only those tests marked 'official'
+    host_driven_test_root: The path to walk, looking for the
+        pythonDrivenTests or host_driven_tests directory
+    is_official_build: Whether to run only those tests marked 'official'
 
   Returns:
-    A list of Python modules which may have zero or more tests.
+    A list of python modules under |host_driven_test_root| which match the
+    testing naming scheme. Each module should define one or more classes that
+    derive from HostDrivenTestCase.
   """
-  # By default run all python tests under pythonDrivenTests.
-  python_test_file_list = []
-  for root, _, files in os.walk(python_test_root):
+  # By default run all host-driven tests under pythonDrivenTests or
+  # host_driven_tests.
+  host_driven_test_file_list = []
+  for root, _, files in os.walk(host_driven_test_root):
     if (root.endswith('host_driven_tests') or
         root.endswith('pythonDrivenTests') or
-        (is_official_build and root.endswith('pythonDrivenTests/official'))):
-      python_test_file_list += _GetPythonFiles(root, files)
-  python_test_file_list.sort()
+        (is_official_build and (root.endswith('pythonDrivenTests/official') or
+                                root.endswith('host_driven_tests/official')))):
+      host_driven_test_file_list += _GetPythonFiles(root, files)
+  host_driven_test_file_list.sort()
 
   test_module_list = [_GetModuleFromFile(test_file)
-                      for test_file in python_test_file_list]
+                      for test_file in host_driven_test_file_list]
   return test_module_list
 
 
 def _GetModuleFromFile(python_file):
-  """Gets the module associated with a file by importing it.
+  """Gets the python module associated with a file by importing it.
 
   Args:
-    python_file: file to import
+    python_file: File to import.
 
   Returns:
     The module object.
@@ -169,43 +87,56 @@ def _GetModuleFromFile(python_file):
   return __import__(import_name)
 
 
-def _GetTestsFromClass(test_class):
-  """Create a list of test objects for each test method on this class.
+def _GetTestsFromClass(test_case_class, **kwargs):
+  """Returns one test object for each test method in |test_case_class|.
 
   Test methods are methods on the class which begin with 'test'.
 
   Args:
-    test_class: class object which contains zero or more test methods.
+    test_case_class: Class derived from HostDrivenTestCase which contains zero
+        or more test methods.
+    kwargs: Keyword args to pass into the constructor of test cases.
 
   Returns:
-    A list of test objects, each of which is bound to one test.
+    A list of test case objects, each initialized for a particular test method.
   """
-  test_names = [m for m in dir(test_class)
-                if _IsTestMethod(m, test_class)]
-  return map(test_class, test_names)
+  test_names = [m for m in dir(test_case_class)
+                if _IsTestMethod(m, test_case_class)]
+  return [test_case_class(name, **kwargs) for name in test_names]
+
+
+def _GetTestsFromModule(test_module, **kwargs):
+  """Gets a list of test objects from |test_module|.
+
+  Args:
+    test_module: Module from which to get the set of test methods.
+    kwargs: Keyword args to pass into the constructor of test cases.
 
+  Returns:
+    A list of test case objects each initialized for a particular test method
+    defined in |test_module|.
+  """
 
-def _GetTestClassesFromModule(test_module):
   tests = []
   for name in dir(test_module):
     attr = getattr(test_module, name)
-    if _IsTestClass(attr):
-      tests.extend(_GetTestsFromClass(attr))
+    if _IsTestCaseClass(attr):
+      tests.extend(_GetTestsFromClass(attr, **kwargs))
   return tests
 
 
-def _IsTestClass(test_class):
+def _IsTestCaseClass(test_class):
   return (type(test_class) is types.TypeType and
-          issubclass(test_class, python_test_base.PythonTestBase) and
-          test_class is not python_test_base.PythonTestBase)
+          issubclass(test_class, test_case.HostDrivenTestCase) and
+          test_class is not test_case.HostDrivenTestCase)
 
 
 def _IsTestMethod(attrname, test_case_class):
   """Checks whether this is a valid test method.
 
   Args:
-    attrname: the method name.
-    test_case_class: the test case class.
+    attrname: The method name.
+    test_case_class: The test case class.
 
   Returns:
     True if test_case_class.'attrname' is callable and it starts with 'test';
@@ -215,20 +146,58 @@ def _IsTestMethod(attrname, test_case_class):
   return callable(attr) and attrname.startswith('test')
 
 
-def _GetAllTests(test_root, is_official_build):
-  """Retrieve a list of Python test modules and their respective methods.
+def _GetAllTests(test_root, is_official_build, **kwargs):
+  """Retrieve a list of host-driven tests defined under |test_root|.
 
   Args:
-    test_root: path which contains Python-driven test files
-    is_official_build: whether this is an official build
+    test_root: Path which contains host-driven test files.
+    is_official_build: Whether this is an official build.
+    kwargs: Keyword args to pass into the constructor of test cases.
 
   Returns:
-    List of test case objects for all available test methods.
+    List of test case objects, one for each available test method.
   """
   if not test_root:
     return []
   all_tests = []
   test_module_list = _GetTestModules(test_root, is_official_build)
   for module in test_module_list:
-    all_tests.extend(_GetTestClassesFromModule(module))
+    all_tests.extend(_GetTestsFromModule(module, **kwargs))
   return all_tests
+
+
+def InstrumentationSetup(host_driven_test_root, official_build,
+                         instrumentation_options):
+  """Creates a list of host-driven instrumentation tests and a runner factory.
+
+  Args:
+    host_driven_test_root: Directory where the host-driven tests are.
+    official_build: True if this is an official build.
+    instrumentation_options: An InstrumentationOptions object.
+
+  Returns:
+    A tuple of (TestRunnerFactory, tests).
+  """
+
+  test_collection = test_info_collection.TestInfoCollection()
+  all_tests = _GetAllTests(
+      host_driven_test_root, official_build,
+      instrumentation_options=instrumentation_options)
+  test_collection.AddTests(all_tests)
+
+  available_tests = test_collection.GetAvailableTests(
+      instrumentation_options.annotations,
+      instrumentation_options.exclude_annotations,
+      instrumentation_options.test_filter)
+  logging.debug('All available tests: ' + str(
+      [t.tagged_name for t in available_tests]))
+
+  def TestRunnerFactory(device, shard_index):
+    return test_runner.HostDrivenTestRunner(
+        device, shard_index,
+        instrumentation_options.tool,
+        instrumentation_options.build_type,
+        instrumentation_options.push_deps,
+        instrumentation_options.cleanup_test_files)
+
+  return (TestRunnerFactory, available_tests)