server/cros/tradefed/cts_expected_failure_parser_unittest.py

# Lint as: python2, python3
# Copyright 2021 The ChromiumOS Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
import glob
import os
import unittest

import common

from autotest_lib.server.cros.tradefed import cts_expected_failure_parser


def glob_add_files(expected_fail_dir):
    """Return a list of files based on a directory path."""

    expected_fail_files = []
    expected_fail_dir_path = os.path.join(
            os.path.dirname(os.path.realpath(__file__)), expected_fail_dir)
    if os.path.exists(expected_fail_dir_path):
        expected_fail_files += glob.glob(expected_fail_dir_path + '/*.yaml')
    return expected_fail_files


class CtsExpectedFailureParserTest(unittest.TestCase):
    """Unittest for cts_expected_failure_parser."""

    def test_binarytranslated_tag(self):
        expected_fail_files = glob_add_files(
                'cts_expected_failure_parser_unittest_data')

        waivers = cts_expected_failure_parser.ParseKnownCTSFailures(
                expected_fail_files)
        # params: arch, board, model, bundle_abi, sdk_ver, first_api_level, host
        found_waivers = waivers.find_waivers('x86', 'hatch', 'kohaku', 'arm',
                                             '30', '30')

        self.assertTrue('GtsOnlyPrimaryAbiTestCases' in found_waivers)

        # params: arch, board, model, bundle_abi, sdk_ver, first_api_level, host
        found_waivers = waivers.find_waivers('x86', 'hatch', 'kohaku', 'x86',
                                             '30', '30')

        self.assertFalse('GtsOnlyPrimaryAbiTestCases' in found_waivers)


if __name__ == '__main__':
    unittest.main()