| Index: build/android/pylib/host_driven/test_info_collection.py
|
| diff --git a/build/android/pylib/host_driven/test_info_collection.py b/build/android/pylib/host_driven/test_info_collection.py
|
| deleted file mode 100644
|
| index 2e2db251190c66f3f3d33369a8d38aa74d80672f..0000000000000000000000000000000000000000
|
| --- a/build/android/pylib/host_driven/test_info_collection.py
|
| +++ /dev/null
|
| @@ -1,139 +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.
|
| -
|
| -"""Module containing information about the host-driven tests."""
|
| -
|
| -import logging
|
| -import os
|
| -
|
| -from pylib.constants import host_paths
|
| -from pylib.host_driven import tests_annotations
|
| -
|
| -with host_paths.SysPath(host_paths.BUILD_COMMON_PATH):
|
| - import unittest_util # pylint: disable=import-error
|
| -
|
| -class TestInfo(object):
|
| - """An object containing and representing a test function, plus metadata."""
|
| -
|
| - def __init__(self, runnable, set_up=None, tear_down=None):
|
| - # The actual test function/method.
|
| - self.runnable = runnable
|
| - # Qualified name of test function/method (e.g. FooModule.testBar).
|
| - self.qualified_name = self._GetQualifiedName(runnable)
|
| - # setUp and teardown functions, if any.
|
| - self.set_up = set_up
|
| - self.tear_down = tear_down
|
| -
|
| - @staticmethod
|
| - def _GetQualifiedName(runnable):
|
| - """Helper method to infer a runnable's name and module name.
|
| -
|
| - Many filters and lists presuppose a format of module_name.testMethodName.
|
| - To make this easy on everyone, we use some reflection magic to infer this
|
| - name automatically.
|
| -
|
| - Args:
|
| - runnable: the test method to get the qualified name for
|
| -
|
| - Returns:
|
| - qualified name for this runnable, incl. module name and method name.
|
| - """
|
| - runnable_name = runnable.__name__
|
| - # See also tests_annotations.
|
| - module_name = os.path.splitext(
|
| - os.path.basename(runnable.__globals__['__file__']))[0]
|
| - return '.'.join([module_name, runnable_name])
|
| -
|
| - def __str__(self):
|
| - return self.qualified_name
|
| -
|
| -
|
| -class TestInfoCollection(object):
|
| - """A collection of TestInfo objects which facilitates filtering."""
|
| -
|
| - def __init__(self):
|
| - """Initialize a new TestInfoCollection."""
|
| - # Master list of all valid tests.
|
| - self.all_tests = []
|
| -
|
| - def AddTests(self, test_infos):
|
| - """Adds a set of tests to this collection.
|
| -
|
| - The user may then retrieve them, optionally according to criteria, via
|
| - GetAvailableTests().
|
| -
|
| - Args:
|
| - test_infos: a list of TestInfos representing test functions/methods.
|
| - """
|
| - self.all_tests = test_infos
|
| -
|
| - def GetAvailableTests(self, annotations, exclude_annotations, name_filter):
|
| - """Get a collection of TestInfos which match the supplied criteria.
|
| -
|
| - Args:
|
| - annotations: List of annotations. Each test in the returned list is
|
| - annotated with atleast one of these annotations.
|
| - exclude_annotations: List of annotations. The tests in the returned
|
| - list are not annotated with any of these annotations.
|
| - name_filter: name filter which tests must match, if any
|
| -
|
| - Returns:
|
| - List of available tests.
|
| - """
|
| - available_tests = self.all_tests
|
| -
|
| - # Filter out tests which match neither the requested annotation, nor the
|
| - # requested name filter, if any.
|
| - available_tests = [t for t in available_tests if
|
| - self._AnnotationIncludesTest(t, annotations)]
|
| - if annotations and len(annotations) == 1 and annotations[0] == 'SmallTest':
|
| - tests_without_annotation = [
|
| - t for t in self.all_tests if
|
| - not tests_annotations.AnnotatedFunctions.GetTestAnnotations(
|
| - t.qualified_name)]
|
| - test_names = [t.qualified_name for t in tests_without_annotation]
|
| - logging.warning('The following tests do not contain any annotation. '
|
| - 'Assuming "SmallTest":\n%s',
|
| - '\n'.join(test_names))
|
| - available_tests += tests_without_annotation
|
| - if exclude_annotations:
|
| - excluded_tests = [t for t in available_tests if
|
| - self._AnnotationIncludesTest(t, exclude_annotations)]
|
| - available_tests = list(set(available_tests) - set(excluded_tests))
|
| -
|
| - if name_filter:
|
| - available_test_names = unittest_util.FilterTestNames(
|
| - [t.qualified_name for t in available_tests], name_filter)
|
| - available_tests = [
|
| - t for t in available_tests if
|
| - t.qualified_name in available_test_names]
|
| - return available_tests
|
| -
|
| - @staticmethod
|
| - def _AnnotationIncludesTest(test_info, annotation_filter_list):
|
| - """Checks whether a given test represented by test_info matches annotation.
|
| -
|
| - Args:
|
| - test_info: TestInfo object representing the test
|
| - annotation_filter_list: list of annotation filters to match (e.g. Smoke)
|
| -
|
| - Returns:
|
| - True if no annotation was supplied or the test matches; false otherwise.
|
| - """
|
| - if not annotation_filter_list:
|
| - return True
|
| - for annotation_filter in annotation_filter_list:
|
| - filters = annotation_filter.split('=')
|
| - if len(filters) == 2:
|
| - key = filters[0]
|
| - value_list = filters[1].split(',')
|
| - for value in value_list:
|
| - if tests_annotations.AnnotatedFunctions.IsAnnotated(
|
| - key + ':' + value, test_info.qualified_name):
|
| - return True
|
| - elif tests_annotations.AnnotatedFunctions.IsAnnotated(
|
| - annotation_filter, test_info.qualified_name):
|
| - return True
|
| - return False
|
| -
|
|
|