# Copyright 2017 The ChromiumOS Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""Unit tests for cts_helper module."""

import os

from chromite.lib import cros_test_lib
from chromite.lib import cts_helper
from chromite.lib import osutils


class CtsHelperTestCase(cros_test_lib.MockTestCase):
    """Tests for functions that do not interact with the file system."""

    def testIsCtsTest(self) -> None:
        self.assertTrue(cts_helper.isCtsTest("cheets_CTS_N.arm.all"))
        self.assertTrue(cts_helper.isCtsTest("cheets_CTS_N.x86.all"))
        self.assertTrue(cts_helper.isCtsTest("cheets_GTS_N.all"))
        self.assertTrue(cts_helper.isCtsTest("cheets_GTS.all"))
        self.assertFalse(cts_helper.isCtsTest("cheets_GTS"))
        self.assertFalse(cts_helper.isCtsTest("cheets_CTS"))

    def testGetXMLPattern(self) -> None:
        self.assertEqual(
            "test_result.xml", cts_helper.getXMLPattern("cheets_CTS_P.arm.all")
        )
        self.assertEqual(
            "test_result.xml", cts_helper.getXMLPattern("cheets_GTS.arm.all")
        )


class UnmockedTests(cros_test_lib.TempDirTestCase):
    """Tests for functions which interact with the file system."""

    def testGetApfeFiles(self) -> None:
        results_path = os.path.join(self.tempdir, "tmp")
        os.makedirs(results_path)

        test_folder = os.path.join(results_path, "test_folder")
        file1 = os.path.join(
            test_folder,
            "cheets_CTS_N",
            "results",
            "android-cts",
            "2017.10.03_10.43.10.zip",
        )
        file2 = os.path.join(
            test_folder,
            "cheets_CTS_N",
            "results",
            "android-cts",
            "2017.10.03_10.43.22.zip",
        )
        osutils.WriteFile(file1, "", makedirs=True)
        osutils.WriteFile(file2, "", makedirs=True)

        self.assertEqual(
            set(cts_helper.getApfeFiles("cheets_CTS_N.Audio", test_folder)),
            set([file1, file2]),
        )

    def testGetXMLGZFiles1(self) -> None:
        results_path = os.path.join(self.tempdir, "tmp")
        os.makedirs(results_path)

        test_folder = os.path.join(results_path, "test_folder")
        file1 = os.path.join(
            test_folder,
            "cheets_CTS_N",
            "results",
            "android-cts",
            "2017.10.03_10.43.22",
            "test_result.xml",
        )
        file2 = os.path.join(
            test_folder,
            "cheets_CTS_N",
            "results",
            "android-cts",
            "2017.10.03_10.43.24",
            "test_result.xml",
        )
        osutils.WriteFile(file1, "", makedirs=True)
        osutils.WriteFile(file2, "", makedirs=True)

        self.assertEqual(
            set(cts_helper.getXMLGZFiles("cheets_CTS_N", test_folder)),
            set([file1 + ".gz", file2 + ".gz"]),
        )