| Index: build/android/pylib/host_driven/setup.py
|
| diff --git a/build/android/pylib/host_driven/setup.py b/build/android/pylib/host_driven/setup.py
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..b2ed348aeb0bbb1bc124dcf4aba13b516fbbb21e
|
| --- /dev/null
|
| +++ b/build/android/pylib/host_driven/setup.py
|
| @@ -0,0 +1,200 @@
|
| +# 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.
|
| +
|
| +"""Setup for instrumentation host-driven tests."""
|
| +
|
| +import logging
|
| +import os
|
| +import sys
|
| +import types
|
| +
|
| +from pylib.host_driven import test_case
|
| +from pylib.host_driven import test_info_collection
|
| +from pylib.host_driven import test_runner
|
| +
|
| +
|
| +def _GetPythonFiles(root, files):
|
| + """Returns all files from |files| that end in 'Test.py'.
|
| +
|
| + Args:
|
| + root: A directory name with python files.
|
| + files: A list of file names.
|
| +
|
| + Returns:
|
| + 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')]
|
| +
|
| +
|
| +def _InferImportNameFromFile(python_file):
|
| + """Given a file, infer the import name for that file.
|
| +
|
| + Example: /usr/foo/bar/baz.py -> baz.
|
| +
|
| + Args:
|
| + python_file: Path to the Python file, ostensibly to import later.
|
| +
|
| + Returns:
|
| + The module name for the given file.
|
| + """
|
| + return os.path.splitext(os.path.basename(python_file))[0]
|
| +
|
| +
|
| +def _GetTestModules(host_driven_test_root, is_official_build):
|
| + """Retrieve a list of python modules that match the testing naming scheme.
|
| +
|
| + Walks the location of host-driven tests, imports them, and provides the list
|
| + of imported modules to the caller.
|
| +
|
| + Args:
|
| + 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 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 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') 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 host_driven_test_file_list]
|
| + return test_module_list
|
| +
|
| +
|
| +def _GetModuleFromFile(python_file):
|
| + """Gets the python module associated with a file by importing it.
|
| +
|
| + Args:
|
| + python_file: File to import.
|
| +
|
| + Returns:
|
| + The module object.
|
| + """
|
| + sys.path.append(os.path.dirname(python_file))
|
| + import_name = _InferImportNameFromFile(python_file)
|
| + return __import__(import_name)
|
| +
|
| +
|
| +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_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 case objects, each initialized for a particular test method.
|
| + """
|
| + 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|.
|
| + """
|
| +
|
| + tests = []
|
| + for name in dir(test_module):
|
| + attr = getattr(test_module, name)
|
| + if _IsTestCaseClass(attr):
|
| + tests.extend(_GetTestsFromClass(attr, **kwargs))
|
| + return tests
|
| +
|
| +
|
| +def _IsTestCaseClass(test_class):
|
| + return (type(test_class) is types.TypeType and
|
| + 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.
|
| +
|
| + Returns:
|
| + True if test_case_class.'attrname' is callable and it starts with 'test';
|
| + False otherwise.
|
| + """
|
| + attr = getattr(test_case_class, attrname)
|
| + return callable(attr) and attrname.startswith('test')
|
| +
|
| +
|
| +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 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, 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(_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)
|
| +
|
| + return (TestRunnerFactory, available_tests)
|
|
|