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

"""Board build dependency service tests."""

import os

from chromite.api import api_config
from chromite.api.controller import dependency
from chromite.api.gen.chromite.api import depgraph_pb2
from chromite.api.gen.chromite.api import sysroot_pb2
from chromite.api.gen.chromiumos import common_pb2
from chromite.lib import cros_build_lib
from chromite.lib import cros_test_lib
from chromite.lib import osutils
from chromite.lib.parser import package_info
from chromite.service import dependency as dependency_service


pytestmark = cros_test_lib.pytestmark_inside_only


class BoardBuildDependencyTest(
    cros_test_lib.MockTestCase, api_config.ApiConfigMixin
):
    """Unittests for board_build_dependency."""

    def setUp(self):
        self.response = depgraph_pb2.GetBuildDependencyGraphResponse()
        self.json_deps = {
            "target_board": "deathstar",
            "sysroot_path": "/build/deathstar",
            "package_deps": {
                "commander/darthvader-1.49.3.3": {
                    "action": "merge",
                    "category": "commander",
                    "cpes": [],
                    "deps": ["troop/clone", "troop/robot"],
                    "rev_deps": [],
                    "full_name": "commander/darthvader-1.49.3.3",
                    "name": "darthvader",
                    "version": "1.49.3.3",
                },
                "troop/clone-1.2.3": {
                    "action": "merge",
                    "category": "troop",
                    "cpes": [],
                    "deps": ["equipment/jetpack"],
                    "rev_deps": ["commander/darthvader"],
                    "full_name": "troop/clone-1.2.3",
                    "name": "clone",
                    "version": "1.2.3",
                },
                "troop/robot-2.3.4": {
                    "action": "merge",
                    "category": "troop",
                    "cpes": [],
                    "deps": [],
                    "rev_deps": ["commander/darthvader"],
                    "full_name": "troop/robot-2.3.4",
                    "name": "robot",
                    "version": "2.3.4",
                },
                "equipment/jetpack-3.4.5": {
                    "action": "merge",
                    "category": "equipment",
                    "cpes": [],
                    "deps": [],
                    "rev_deps": ["commander/darthvader"],
                    "full_name": "equipment/jetpack-3.4.5",
                    "name": "jetpack",
                    "version": "3.4.5",
                },
            },
            "source_path_mapping": {
                "commander/darthvader-1.49.3.3": ["/control/room"],
                "troop/clone-1.2.3": ["/bunker"],
                "troop/robot-2.3.4": ["/factory"],
                "equipment/jetpack-3.4.5": ["/factory"],
            },
        }

    def testCreateDepGraphProtoFromJsonMap(self):
        """Test creating DepGraph protobuf from json map."""
        depgraph_proto = depgraph_pb2.DepGraph()
        dependency.AugmentDepGraphProtoFromJsonMap(
            self.json_deps, depgraph_proto
        )
        self.assertEqual(depgraph_proto.build_target.name, "deathstar")
        darthvader_dep = None
        for package_dep_info in depgraph_proto.package_deps:
            if package_dep_info.package_info.package_name == "darthvader":
                darthvader_dep = package_dep_info
        self.assertTrue(darthvader_dep)
        self.assertEqual(
            darthvader_dep.dependency_packages[0].category, "troop"
        )
        self.assertEqual(
            darthvader_dep.dependency_packages[0].package_name, "clone"
        )
        self.assertEqual(
            darthvader_dep.dependency_source_paths[0].path, "/control/room"
        )

    def testGetBuildDependencyGraph(self):
        """GetBuildDependencyGraph calls helper method with correct args."""
        patch = self.PatchObject(
            dependency_service,
            "GetBuildDependency",
            return_value=(self.json_deps, self.json_deps),
        )
        input_proto = depgraph_pb2.GetBuildDependencyGraphRequest()
        input_proto.build_target.name = "target"
        dependency.GetBuildDependencyGraph(
            input_proto, self.response, self.api_config
        )
        self.assertEqual(self.response.dep_graph.build_target.name, "deathstar")
        patch.assert_called_once()

    def testGetBuildDependencyGraphForPackages(self):
        """GetBuildDependencyGraph calls helper method with correct args."""
        get_dep = self.PatchObject(
            dependency_service,
            "GetBuildDependency",
            return_value=(self.json_deps, self.json_deps),
        )
        pkg_atom = "chromeos-base/chromeos-chrome"
        package = common_pb2.PackageInfo(
            package_name="chromeos-chrome", category="chromeos-base"
        )
        input_proto = depgraph_pb2.GetBuildDependencyGraphRequest(
            build_target=common_pb2.BuildTarget(name="target"),
            packages=[package],
        )
        dependency.GetBuildDependencyGraph(
            input_proto, self.response, self.api_config
        )
        self.assertEqual(self.response.dep_graph.build_target.name, "deathstar")
        get_dep.assert_called_once_with("/build/target", "target", (pkg_atom,))

    def testValidateOnly(self):
        """Test that a validate only call does not execute any logic."""
        patch = self.PatchObject(dependency_service, "GetBuildDependency")
        input_proto = depgraph_pb2.GetBuildDependencyGraphRequest()
        input_proto.build_target.name = "target"
        dependency.GetBuildDependencyGraph(
            input_proto, self.response, self.validate_only_config
        )
        patch.assert_not_called()

    def testMockCall(self):
        """Test a mock call does not execute logic, returns mocked value."""
        patch = self.PatchObject(dependency_service, "GetBuildDependency")
        input_proto = depgraph_pb2.GetBuildDependencyGraphRequest()
        input_proto.build_target.name = "target"
        dependency.GetBuildDependencyGraph(
            input_proto, self.response, self.mock_call_config
        )
        patch.assert_not_called()
        self.assertEqual(
            self.response.dep_graph.build_target.name, "target_board"
        )


class ListTest(cros_test_lib.MockTempDirTestCase, api_config.ApiConfigMixin):
    """Unittests for the List endpoint."""

    def setUp(self):
        self.response = depgraph_pb2.ListResponse()
        self.build_target = common_pb2.BuildTarget(name="target")
        self.sysroot = os.path.join(self.tempdir, "target")
        osutils.SafeMakedirs(self.sysroot)

    def testValidateOnly(self):
        """Test that a validate only call does not execute any logic."""
        sysroot = sysroot_pb2.Sysroot(
            path=self.sysroot, build_target=self.build_target
        )
        input_proto = depgraph_pb2.ListRequest(sysroot=sysroot)
        dependency.List(input_proto, self.response, self.validate_only_config)

    def testArgumentValidationMissingSysrootPath(self):
        """Test missing sysroot path."""
        sysroot = sysroot_pb2.Sysroot(build_target=self.build_target)
        input_proto = depgraph_pb2.ListRequest(sysroot=sysroot)
        with self.assertRaises(cros_build_lib.DieSystemExit):
            dependency.List(input_proto, self.response, self.api_config)

    def testArgumentValidationMissingBuildTarget(self):
        """Test missing build target name."""
        sysroot = sysroot_pb2.Sysroot(
            path=self.sysroot, build_target=common_pb2.BuildTarget()
        )
        input_proto = depgraph_pb2.ListRequest(sysroot=sysroot)
        with self.assertRaises(cros_build_lib.DieSystemExit):
            dependency.List(input_proto, self.response, self.api_config)

    def testDefaultArguments(self):
        """Test with default arguments."""
        sysroot = sysroot_pb2.Sysroot(
            path=self.sysroot, build_target=self.build_target
        )
        input_proto = depgraph_pb2.ListRequest(sysroot=sysroot)
        mock_get_deps = self.PatchObject(dependency_service, "GetDependencies")
        dependency.List(input_proto, self.response, self.api_config)
        mock_get_deps.assert_called_once_with(
            self.sysroot,
            src_paths=[],
            packages=[],
            include_affected_pkgs=False,
        )

    def testListResponse(self):
        """Test calls helper method with correct args."""
        sysroot = sysroot_pb2.Sysroot(
            path=self.sysroot, build_target=self.build_target
        )
        path = "/path"
        return_package_info = package_info.parse("foo/bar-1.2.3")
        return_package_info_proto = common_pb2.PackageInfo(
            category="foo", package_name="bar", version="1.2.3"
        )
        mock_get_deps = self.PatchObject(
            dependency_service,
            "GetDependencies",
            return_value=[return_package_info],
        )

        input_package_info_proto = common_pb2.PackageInfo(
            category="foo", package_name="bar"
        )
        input_package_info = package_info.parse("foo/bar")
        input_proto = depgraph_pb2.ListRequest(
            sysroot=sysroot,
            src_paths=[
                depgraph_pb2.SourcePath(path=path),
            ],
            packages=[input_package_info_proto],
            include_rev_deps=True,
        )
        dependency.List(input_proto, self.response, self.api_config)
        mock_get_deps.assert_called_once_with(
            self.sysroot,
            src_paths=[path],
            packages=[input_package_info],
            include_affected_pkgs=True,
        )
        self.assertCountEqual(
            [return_package_info_proto], self.response.package_deps
        )