Index: build/android/pylib/instrumentation/test_jar.py |
diff --git a/build/android/pylib/instrumentation/test_jar.py b/build/android/pylib/instrumentation/test_jar.py |
deleted file mode 100644 |
index 7ad8997755cc2c49763a51a946adb05604e21b19..0000000000000000000000000000000000000000 |
--- a/build/android/pylib/instrumentation/test_jar.py |
+++ /dev/null |
@@ -1,230 +0,0 @@ |
-# Copyright (c) 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. |
- |
-"""Helper class for instrumenation test jar.""" |
-# pylint: disable=W0702 |
- |
-import logging |
-import os |
-import pickle |
-import re |
-import sys |
- |
-from pylib import cmd_helper |
-from pylib import constants |
-from pylib.device import device_utils |
-from pylib.utils import md5sum |
-from pylib.utils import proguard |
- |
-sys.path.insert(0, |
- os.path.join(constants.DIR_SOURCE_ROOT, |
- 'build', 'util', 'lib', 'common')) |
- |
-import unittest_util # pylint: disable=F0401 |
- |
-# If you change the cached output of proguard, increment this number |
-PICKLE_FORMAT_VERSION = 4 |
- |
- |
-class TestJar(object): |
- _ANNOTATIONS = frozenset( |
- ['Smoke', 'SmallTest', 'MediumTest', 'LargeTest', 'EnormousTest', |
- 'FlakyTest', 'DisabledTest', 'Manual', 'PerfTest', 'HostDrivenTest', |
- 'IntegrationTest']) |
- _DEFAULT_ANNOTATION = 'SmallTest' |
- _PROGUARD_CLASS_RE = re.compile(r'\s*?- Program class:\s*([\S]+)$') |
- _PROGUARD_SUPERCLASS_RE = re.compile(r'\s*? Superclass:\s*([\S]+)$') |
- _PROGUARD_METHOD_RE = re.compile(r'\s*?- Method:\s*(\S*)[(].*$') |
- _PROGUARD_ANNOTATION_RE = re.compile(r'\s*?- Annotation \[L(\S*);\]:$') |
- _PROGUARD_ANNOTATION_CONST_RE = ( |
- re.compile(r'\s*?- Constant element value.*$')) |
- _PROGUARD_ANNOTATION_VALUE_RE = re.compile(r'\s*?- \S+? \[(.*)\]$') |
- |
- def __init__(self, jar_path): |
- if not os.path.exists(jar_path): |
- raise Exception('%s not found, please build it' % jar_path) |
- |
- self._PROGUARD_PATH = os.path.join(constants.ANDROID_SDK_ROOT, |
- 'tools/proguard/lib/proguard.jar') |
- if not os.path.exists(self._PROGUARD_PATH): |
- self._PROGUARD_PATH = os.path.join(os.environ['ANDROID_BUILD_TOP'], |
- 'external/proguard/lib/proguard.jar') |
- self._jar_path = jar_path |
- self._pickled_proguard_name = self._jar_path + '-proguard.pickle' |
- self._test_methods = {} |
- if not self._GetCachedProguardData(): |
- self._GetProguardData() |
- |
- def _GetCachedProguardData(self): |
- if (os.path.exists(self._pickled_proguard_name) and |
- (os.path.getmtime(self._pickled_proguard_name) > |
- os.path.getmtime(self._jar_path))): |
- logging.info('Loading cached proguard output from %s', |
- self._pickled_proguard_name) |
- try: |
- with open(self._pickled_proguard_name, 'r') as r: |
- d = pickle.loads(r.read()) |
- jar_md5 = md5sum.CalculateHostMd5Sums( |
- self._jar_path)[os.path.realpath(self._jar_path)] |
- if (d['JAR_MD5SUM'] == jar_md5 and |
- d['VERSION'] == PICKLE_FORMAT_VERSION): |
- self._test_methods = d['TEST_METHODS'] |
- return True |
- except: |
- logging.warning('PICKLE_FORMAT_VERSION has changed, ignoring cache') |
- return False |
- |
- def _GetProguardData(self): |
- logging.info('Retrieving test methods via proguard.') |
- |
- p = proguard.Dump(self._jar_path) |
- |
- class_lookup = dict((c['class'], c) for c in p['classes']) |
- def recursive_get_annotations(c): |
- s = c['superclass'] |
- if s in class_lookup: |
- a = recursive_get_annotations(class_lookup[s]) |
- else: |
- a = {} |
- a.update(c['annotations']) |
- return a |
- |
- test_classes = (c for c in p['classes'] |
- if c['class'].endswith('Test')) |
- for c in test_classes: |
- class_annotations = recursive_get_annotations(c) |
- test_methods = (m for m in c['methods'] |
- if m['method'].startswith('test')) |
- for m in test_methods: |
- qualified_method = '%s#%s' % (c['class'], m['method']) |
- annotations = dict(class_annotations) |
- annotations.update(m['annotations']) |
- self._test_methods[qualified_method] = m |
- self._test_methods[qualified_method]['annotations'] = annotations |
- |
- logging.info('Storing proguard output to %s', self._pickled_proguard_name) |
- d = {'VERSION': PICKLE_FORMAT_VERSION, |
- 'TEST_METHODS': self._test_methods, |
- 'JAR_MD5SUM': |
- md5sum.CalculateHostMd5Sums( |
- self._jar_path)[os.path.realpath(self._jar_path)]} |
- with open(self._pickled_proguard_name, 'w') as f: |
- f.write(pickle.dumps(d)) |
- |
- @staticmethod |
- def _IsTestMethod(test): |
- class_name, method = test.split('#') |
- return class_name.endswith('Test') and method.startswith('test') |
- |
- def GetTestAnnotations(self, test): |
- """Returns a list of all annotations for the given |test|. May be empty.""" |
- if not self._IsTestMethod(test) or not test in self._test_methods: |
- return [] |
- return self._test_methods[test]['annotations'] |
- |
- @staticmethod |
- def _AnnotationsMatchFilters(annotation_filter_list, annotations): |
- """Checks if annotations match any of the filters.""" |
- 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 key in annotations and value == annotations[key]: |
- return True |
- elif annotation_filter in annotations: |
- return True |
- return False |
- |
- def GetAnnotatedTests(self, annotation_filter_list): |
- """Returns a list of all tests that match the given annotation filters.""" |
- return [test for test in self.GetTestMethods() |
- if self._IsTestMethod(test) and self._AnnotationsMatchFilters( |
- annotation_filter_list, self.GetTestAnnotations(test))] |
- |
- def GetTestMethods(self): |
- """Returns a dict of all test methods and relevant attributes. |
- |
- Test methods are retrieved as Class#testMethod. |
- """ |
- return self._test_methods |
- |
- def _GetTestsMissingAnnotation(self): |
- """Get a list of test methods with no known annotations.""" |
- tests_missing_annotations = [] |
- for test_method in self.GetTestMethods().iterkeys(): |
- annotations_ = frozenset(self.GetTestAnnotations(test_method).iterkeys()) |
- if (annotations_.isdisjoint(self._ANNOTATIONS) and |
- not self.IsHostDrivenTest(test_method)): |
- tests_missing_annotations.append(test_method) |
- return sorted(tests_missing_annotations) |
- |
- def _IsTestValidForSdkRange(self, test_name, attached_min_sdk_level): |
- required_min_sdk_level = int( |
- self.GetTestAnnotations(test_name).get('MinAndroidSdkLevel', 0)) |
- return (required_min_sdk_level is None or |
- attached_min_sdk_level >= required_min_sdk_level) |
- |
- def GetAllMatchingTests(self, annotation_filter_list, |
- exclude_annotation_list, test_filter): |
- """Get a list of tests matching any of the annotations and the filter. |
- |
- Args: |
- annotation_filter_list: List of test annotations. A test must have at |
- least one of these annotations. A test without any annotations is |
- considered to be SmallTest. |
- exclude_annotation_list: List of test annotations. A test must not have |
- any of these annotations. |
- test_filter: Filter used for partial matching on the test method names. |
- |
- Returns: |
- List of all matching tests. |
- """ |
- if annotation_filter_list: |
- available_tests = self.GetAnnotatedTests(annotation_filter_list) |
- # Include un-annotated tests in SmallTest. |
- if annotation_filter_list.count(self._DEFAULT_ANNOTATION) > 0: |
- for test in self._GetTestsMissingAnnotation(): |
- logging.warning( |
- '%s has no annotations. Assuming "%s".', test, |
- self._DEFAULT_ANNOTATION) |
- available_tests.append(test) |
- else: |
- available_tests = [m for m in self.GetTestMethods() |
- if not self.IsHostDrivenTest(m)] |
- |
- if exclude_annotation_list: |
- excluded_tests = self.GetAnnotatedTests(exclude_annotation_list) |
- available_tests = list(set(available_tests) - set(excluded_tests)) |
- |
- tests = [] |
- if test_filter: |
- # |available_tests| are in adb instrument format: package.path.class#test. |
- |
- # Maps a 'class.test' name to each 'package.path.class#test' name. |
- sanitized_test_names = dict([ |
- (t.split('.')[-1].replace('#', '.'), t) for t in available_tests]) |
- # Filters 'class.test' names and populates |tests| with the corresponding |
- # 'package.path.class#test' names. |
- tests = [ |
- sanitized_test_names[t] for t in unittest_util.FilterTestNames( |
- sanitized_test_names.keys(), test_filter.replace('#', '.'))] |
- else: |
- tests = available_tests |
- |
- # Filter out any tests with SDK level requirements that don't match the set |
- # of attached devices. |
- devices = device_utils.DeviceUtils.parallel() |
- min_sdk_version = min(devices.build_version_sdk.pGet(None)) |
- tests = [t for t in tests |
- if self._IsTestValidForSdkRange(t, min_sdk_version)] |
- |
- return tests |
- |
- @staticmethod |
- def IsHostDrivenTest(test): |
- return 'pythonDrivenTests' in test |