contrib/chromium_metadata/chromium_metadata_test.py

#!/usr/bin/env python3
# Copyright 2020 The ChromiumOS Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

# pylint: disable=missing-docstring

import pathlib
import unittest

from chromium_metadata import chromium_metadata
from chromium_metadata import create_metadata_files
from chromium_metadata import extract_owners
from chromium_metadata import find_owners_files
from mock import call
from mock import patch


GIT_REPO_DIR = pathlib.Path("/mnt/host/source/src/platform/dev")
TEST_DATA_DIR = pathlib.Path("./test_data")
OWNERS_FILE1 = TEST_DATA_DIR.joinpath("OWNERS")
OWNERS_FILE2 = TEST_DATA_DIR.joinpath("component/OWNERS")


class TestExtractOwners(unittest.TestCase):
    def test_extract_owners(self):
        self.assertEqual(
            extract_owners(TEST_DATA_DIR.joinpath("OWNERS")),
            ["tomhughes@chromium.org"],
        )


class TestFindOwnersFiles(unittest.TestCase):
    def test_find_owners_files(self):
        self.assertEqual(
            find_owners_files(TEST_DATA_DIR), [OWNERS_FILE1, OWNERS_FILE2]
        )


class TestCreateMetadataFiles(unittest.TestCase):
    @patch("chromium_metadata.create_cl")
    def test_create_metadata_files(self, create_cl_mock):
        create_metadata_files(TEST_DATA_DIR)
        expected = [
            call(
                sub_dir=pathlib.Path("."),
                git_repo=TEST_DATA_DIR,
                reviewers=["tomhughes@chromium.org"],
            ),
            call(
                sub_dir=pathlib.Path("component"),
                git_repo=TEST_DATA_DIR,
                reviewers=["adlr@chromium.org", "tomhughes@chromium.org"],
            ),
        ]
        create_cl_mock.assert_has_calls(expected)
        self.assertEqual(create_cl_mock.call_count, len(expected))


class TestChromiumMetadata(unittest.TestCase):
    @patch("chromium_metadata.create_metadata_files")
    def test_chromium_metadata(self, create_metadata_files_mock):
        chromium_metadata()
        create_metadata_files_mock.assert_called_with(GIT_REPO_DIR)


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