atest-py2/test_finders/cache_finder.py

# 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 atest_utils
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, **kwargs):
        super(CacheFinder, self).__init__()

    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:
                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_latest_testinfos(test_infos):
            return test_infos
        return None