| # -*- coding: utf-8 -*- |
| # Copyright 2019 The Chromium OS Authors. All rights reserved. |
| # 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.""" |
| |
| from __future__ import print_function |
| |
| import sys |
| |
| from chromite.api import api_config |
| from chromite.api.controller import controller_util |
| from chromite.api.controller import dependency |
| from chromite.api.gen.chromite.api import depgraph_pb2 |
| from chromite.api.gen.chromiumos import common_pb2 |
| from chromite.lib import cros_test_lib |
| from chromite.service import dependency as dependency_service |
| |
| pytestmark = cros_test_lib.pytestmark_inside_only |
| |
| |
| assert sys.version_info >= (3, 6), 'This module requires Python 3.6+' |
| |
| |
| 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': [], |
| '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' |
| }, |
| }, |
| 'source_path_mapping': { |
| 'commander/darthvader-1.49.3.3': ['/control/room'], |
| 'troop/clone-1.2.3': ['/bunker'], |
| 'troop/robot-2.3.4': ['/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_mock = 'package-CPV' |
| pkg_to_cpv = self.PatchObject( |
| controller_util, |
| 'PackageInfoToCPV', return_value=pkg_mock) |
| 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') |
| pkg_to_cpv.assert_called_once_with(package) |
| get_dep.assert_called_once_with('/build/target', 'target', [pkg_mock]) |
| |
| def testValidateOnly(self): |
| """Sanity check 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 that 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') |