| # Copyright 2019, The Android Open Source Project |
| # |
| # Licensed under the Apache License, Version 2.0 (the "License"); |
| # you may not use this file except in compliance with the License. |
| # You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, software |
| # distributed under the License is distributed on an "AS IS" BASIS, |
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| # See the License for the specific language governing permissions and |
| # limitations under the License. |
| |
| """ |
| Cache Finder class. |
| """ |
| |
| import logging |
| |
| import atest_utils |
| import constants |
| |
| from test_finders import test_finder_base |
| from test_finders import test_info |
| |
| class CacheFinder(test_finder_base.TestFinderBase): |
| """Cache Finder class.""" |
| NAME = 'CACHE' |
| |
| def __init__(self, module_info=None): |
| super().__init__() |
| self.module_info = module_info |
| |
| def _is_latest_testinfos(self, test_infos): |
| """Check whether test_infos are up-to-date. |
| |
| Args: |
| test_infos: A list of TestInfo. |
| |
| Returns: |
| True if all keys in test_infos and TestInfo object are equal. |
| Otherwise, False. |
| """ |
| sorted_base_ti = sorted( |
| vars(test_info.TestInfo(None, None, None)).keys()) |
| for cached_test_info in test_infos: |
| sorted_cache_ti = sorted(vars(cached_test_info).keys()) |
| if not sorted_cache_ti == sorted_base_ti: |
| logging.debug('test_info is not up-to-date.') |
| return False |
| return True |
| |
| def find_test_by_cache(self, test_reference): |
| """Find the matched test_infos in saved caches. |
| |
| Args: |
| test_reference: A string of the path to the test's file or dir. |
| |
| Returns: |
| A list of TestInfo namedtuple if cache found and is in latest |
| TestInfo format, else None. |
| """ |
| test_infos = atest_utils.load_test_info_cache(test_reference) |
| if test_infos and self._is_test_infos_valid(test_infos): |
| return test_infos |
| return None |
| |
| def _is_test_infos_valid(self, test_infos): |
| """Check if the given test_infos are valid. |
| |
| Args: |
| test_infos: A list of TestInfo. |
| |
| Returns: |
| True if test_infos are all valid. Otherwise, False. |
| """ |
| if not self._is_latest_testinfos(test_infos): |
| return False |
| for t_info in test_infos: |
| if not self._is_test_path_valid(t_info): |
| return False |
| if not self._is_test_build_target_valid(t_info): |
| return False |
| if not self._is_test_filter_valid(t_info): |
| return False |
| return True |
| |
| def _is_test_path_valid(self, t_info): |
| """Check if test path is valid. |
| |
| Args: |
| t_info: TestInfo that has been filled out by a find method. |
| |
| Returns: |
| True if test path is valid. Otherwise, False. |
| """ |
| # For RoboTest it won't have 'MODULES-IN-' as build target. Treat test |
| # path is valid if cached_test_paths is None. |
| cached_test_paths = t_info.get_test_paths() |
| if cached_test_paths is None: |
| return True |
| current_test_paths = self.module_info.get_paths(t_info.test_name) |
| if not current_test_paths: |
| return False |
| if sorted(cached_test_paths) != sorted(current_test_paths): |
| logging.debug('Not a valid test path.') |
| return False |
| return True |
| |
| def _is_test_build_target_valid(self, t_info): |
| """Check if test build targets are valid. |
| |
| Args: |
| t_info: TestInfo that has been filled out by a find method. |
| |
| Returns: |
| True if test's build target is valid. Otherwise, False. |
| """ |
| # If the cached build target can be found in current module-info, then |
| # it is a valid build targets of the test. |
| for build_target in t_info.build_targets: |
| if str(build_target).startswith(constants.MODULES_IN): |
| continue |
| if not self.module_info.is_module(build_target): |
| logging.debug('%s is not a valid build target.', build_target) |
| return False |
| return True |
| |
| def _is_test_filter_valid(self, t_info): |
| """Check if test filter is valid. |
| |
| Args: |
| t_info: TestInfo that has been filled out by a find method. |
| |
| Returns: |
| True if test filter is valid. Otherwise, False. |
| """ |
| test_filters = t_info.data.get(constants.TI_FILTER, []) |
| if not test_filters: |
| return True |
| for test_filter in test_filters: |
| # Check if the class filter is under current module. |
| # TODO: (b/172260100) The test_name may not be inevitably equal to |
| # the module_name. |
| if self._is_java_filter_in_module(t_info.test_name , |
| test_filter.class_name): |
| return True |
| # TODO: (b/172260100) Also check for CC. |
| logging.debug('Not a valid test filter.') |
| return False |
| |
| def _is_java_filter_in_module(self, module_name, filter_class): |
| """Check if input class is part of input module. |
| |
| Args: |
| module_name: A string of the module name of the test. |
| filter_class: A string of the class name field of TI_FILTER. |
| |
| Returns: |
| True if input filter_class is in the input module. Otherwise, False. |
| """ |
| mod_info = self.module_info.get_module_info(module_name) |
| if not mod_info: |
| return False |
| module_srcs = mod_info.get(constants.MODULE_SRCS, []) |
| # If module didn't have src information treat the cached filter still |
| # valid. Remove this after all java srcs could be found in module-info. |
| if not module_srcs: |
| return True |
| ref_end = filter_class.rsplit('.', 1)[-1] |
| if '.' in filter_class: |
| file_path = str(filter_class).replace('.', '/') |
| # A Java class file always starts with a capital letter. |
| if ref_end[0].isupper(): |
| file_path = file_path + '.' |
| for src_path in module_srcs: |
| # If java class, check if class file in module's src. |
| if src_path.find(file_path) >= 0: |
| return True |
| return False |