| # 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 | |
| ) |