Ned Nguyen | e5d0485 | 2019-02-19 16:33:15 -0700 | [diff] [blame] | 1 | # -*- coding: utf-8 -*- |
| 2 | # Copyright 2019 The Chromium OS Authors. All rights reserved. |
| 3 | # Use of this source code is governed by a BSD-style license that can be |
| 4 | # found in the LICENSE file. |
| 5 | """Board build dependency service tests.""" |
| 6 | |
| 7 | from __future__ import print_function |
| 8 | |
Navil Perez | a782326 | 2020-08-07 17:56:57 +0000 | [diff] [blame] | 9 | import os |
Mike Frysinger | ef94e4c | 2020-02-10 23:59:54 -0500 | [diff] [blame] | 10 | import sys |
| 11 | |
Michael Mortensen | af10cca | 2019-11-18 20:21:32 -0700 | [diff] [blame] | 12 | from chromite.api import api_config |
LaMont Jones | 00b9cff | 2019-12-12 12:14:07 -0700 | [diff] [blame] | 13 | from chromite.api.controller import controller_util |
Alex Klein | ec869c1 | 2019-07-09 15:22:16 -0600 | [diff] [blame] | 14 | from chromite.api.controller import dependency |
Ned Nguyen | 8e144bd | 2019-05-02 09:09:41 -0600 | [diff] [blame] | 15 | from chromite.api.gen.chromite.api import depgraph_pb2 |
Navil Perez | a782326 | 2020-08-07 17:56:57 +0000 | [diff] [blame] | 16 | from chromite.api.gen.chromite.api import sysroot_pb2 |
LaMont Jones | 00b9cff | 2019-12-12 12:14:07 -0700 | [diff] [blame] | 17 | from chromite.api.gen.chromiumos import common_pb2 |
Navil Perez | a782326 | 2020-08-07 17:56:57 +0000 | [diff] [blame] | 18 | from chromite.lib import cros_build_lib |
Ned Nguyen | e5d0485 | 2019-02-19 16:33:15 -0700 | [diff] [blame] | 19 | from chromite.lib import cros_test_lib |
Navil Perez | a782326 | 2020-08-07 17:56:57 +0000 | [diff] [blame] | 20 | from chromite.lib import osutils |
Michael Mortensen | af10cca | 2019-11-18 20:21:32 -0700 | [diff] [blame] | 21 | from chromite.service import dependency as dependency_service |
Ned Nguyen | e5d0485 | 2019-02-19 16:33:15 -0700 | [diff] [blame] | 22 | |
Greg Edelston | a4c9b3b | 2020-01-07 17:51:13 -0700 | [diff] [blame] | 23 | pytestmark = cros_test_lib.pytestmark_inside_only |
| 24 | |
Ned Nguyen | e5d0485 | 2019-02-19 16:33:15 -0700 | [diff] [blame] | 25 | |
Mike Frysinger | ef94e4c | 2020-02-10 23:59:54 -0500 | [diff] [blame] | 26 | assert sys.version_info >= (3, 6), 'This module requires Python 3.6+' |
| 27 | |
| 28 | |
Michael Mortensen | 3232ab3 | 2020-01-05 19:14:36 -0700 | [diff] [blame] | 29 | class BoardBuildDependencyTest(cros_test_lib.MockTestCase, |
| 30 | api_config.ApiConfigMixin): |
Ned Nguyen | e5d0485 | 2019-02-19 16:33:15 -0700 | [diff] [blame] | 31 | """Unittests for board_build_dependency.""" |
| 32 | |
Michael Mortensen | af10cca | 2019-11-18 20:21:32 -0700 | [diff] [blame] | 33 | def setUp(self): |
| 34 | self.response = depgraph_pb2.GetBuildDependencyGraphResponse() |
| 35 | self.json_deps = { |
Ned Nguyen | e5d0485 | 2019-02-19 16:33:15 -0700 | [diff] [blame] | 36 | 'target_board': 'deathstar', |
LaMont Jones | 4cbecba | 2020-05-12 11:54:27 -0600 | [diff] [blame] | 37 | 'sysroot_path': '/build/deathstar', |
Ned Nguyen | e5d0485 | 2019-02-19 16:33:15 -0700 | [diff] [blame] | 38 | 'package_deps': { |
| 39 | 'commander/darthvader-1.49.3.3': { |
| 40 | 'action': 'merge', |
| 41 | 'category': 'commander', |
| 42 | 'cpes': [], |
| 43 | 'deps': ['troop/clone', 'troop/robot'], |
| 44 | 'rev_deps': [], |
| 45 | 'full_name': 'commander/darthvader-1.49.3.3', |
| 46 | 'name': 'darthvader', |
| 47 | 'version': '1.49.3.3' |
| 48 | }, |
| 49 | 'troop/clone-1.2.3': { |
| 50 | 'action': 'merge', |
| 51 | 'category': 'troop', |
| 52 | 'cpes': [], |
Navil Perez | a782326 | 2020-08-07 17:56:57 +0000 | [diff] [blame] | 53 | 'deps': ['equipment/jetpack'], |
Ned Nguyen | e5d0485 | 2019-02-19 16:33:15 -0700 | [diff] [blame] | 54 | 'rev_deps': ['commander/darthvader'], |
| 55 | 'full_name': 'troop/clone-1.2.3', |
| 56 | 'name': 'clone', |
| 57 | 'version': '1.2.3' |
| 58 | }, |
| 59 | 'troop/robot-2.3.4': { |
| 60 | 'action': 'merge', |
| 61 | 'category': 'troop', |
| 62 | 'cpes': [], |
| 63 | 'deps': [], |
| 64 | 'rev_deps': ['commander/darthvader'], |
| 65 | 'full_name': 'troop/robot-2.3.4', |
Ned Nguyen | 8be1005 | 2019-02-26 08:58:45 -0700 | [diff] [blame] | 66 | 'name': 'robot', |
Ned Nguyen | e5d0485 | 2019-02-19 16:33:15 -0700 | [diff] [blame] | 67 | 'version': '2.3.4' |
| 68 | }, |
Navil Perez | a782326 | 2020-08-07 17:56:57 +0000 | [diff] [blame] | 69 | 'equipment/jetpack-3.4.5': { |
| 70 | 'action': 'merge', |
| 71 | 'category': 'equipment', |
| 72 | 'cpes': [], |
| 73 | 'deps': [], |
| 74 | 'rev_deps': ['commander/darthvader'], |
| 75 | 'full_name': 'equipment/jetpack-3.4.5', |
| 76 | 'name': 'jetpack', |
| 77 | 'version': '3.4.5' |
| 78 | }, |
Ned Nguyen | e5d0485 | 2019-02-19 16:33:15 -0700 | [diff] [blame] | 79 | }, |
| 80 | 'source_path_mapping': { |
Ned Nguyen | 8be1005 | 2019-02-26 08:58:45 -0700 | [diff] [blame] | 81 | 'commander/darthvader-1.49.3.3': ['/control/room'], |
| 82 | 'troop/clone-1.2.3': ['/bunker'], |
| 83 | 'troop/robot-2.3.4': ['/factory'], |
Navil Perez | a782326 | 2020-08-07 17:56:57 +0000 | [diff] [blame] | 84 | 'equipment/jetpack-3.4.5': ['/factory'], |
Ned Nguyen | e5d0485 | 2019-02-19 16:33:15 -0700 | [diff] [blame] | 85 | }, |
| 86 | } |
| 87 | |
Michael Mortensen | af10cca | 2019-11-18 20:21:32 -0700 | [diff] [blame] | 88 | def testCreateDepGraphProtoFromJsonMap(self): |
| 89 | """Test creating DepGraph protobuf from json map.""" |
Ned Nguyen | 8e144bd | 2019-05-02 09:09:41 -0600 | [diff] [blame] | 90 | depgraph_proto = depgraph_pb2.DepGraph() |
Alex Klein | ec869c1 | 2019-07-09 15:22:16 -0600 | [diff] [blame] | 91 | dependency.AugmentDepGraphProtoFromJsonMap( |
Michael Mortensen | af10cca | 2019-11-18 20:21:32 -0700 | [diff] [blame] | 92 | self.json_deps, depgraph_proto) |
Ned Nguyen | e5d0485 | 2019-02-19 16:33:15 -0700 | [diff] [blame] | 93 | self.assertEqual(depgraph_proto.build_target.name, 'deathstar') |
| 94 | darthvader_dep = None |
| 95 | for package_dep_info in depgraph_proto.package_deps: |
| 96 | if package_dep_info.package_info.package_name == 'darthvader': |
| 97 | darthvader_dep = package_dep_info |
| 98 | self.assertTrue(darthvader_dep) |
Mike Frysinger | 2d589a1 | 2019-08-25 14:15:12 -0400 | [diff] [blame] | 99 | self.assertEqual(darthvader_dep.dependency_packages[0].category, |
| 100 | 'troop') |
| 101 | self.assertEqual(darthvader_dep.dependency_packages[0].package_name, |
| 102 | 'clone') |
| 103 | self.assertEqual(darthvader_dep.dependency_source_paths[0].path, |
| 104 | '/control/room') |
Michael Mortensen | af10cca | 2019-11-18 20:21:32 -0700 | [diff] [blame] | 105 | |
| 106 | def testGetBuildDependencyGraph(self): |
| 107 | """GetBuildDependencyGraph calls helper method with correct args.""" |
Chris McDonald | a22b74f | 2019-11-22 13:55:06 -0700 | [diff] [blame] | 108 | patch = self.PatchObject( |
| 109 | dependency_service, |
| 110 | 'GetBuildDependency', |
| 111 | return_value=(self.json_deps, self.json_deps)) |
Michael Mortensen | af10cca | 2019-11-18 20:21:32 -0700 | [diff] [blame] | 112 | input_proto = depgraph_pb2.GetBuildDependencyGraphRequest() |
| 113 | input_proto.build_target.name = 'target' |
| 114 | dependency.GetBuildDependencyGraph(input_proto, self.response, |
| 115 | self.api_config) |
| 116 | self.assertEqual(self.response.dep_graph.build_target.name, 'deathstar') |
| 117 | patch.assert_called_once() |
| 118 | |
LaMont Jones | 00b9cff | 2019-12-12 12:14:07 -0700 | [diff] [blame] | 119 | def testGetBuildDependencyGraphForPackages(self): |
| 120 | """GetBuildDependencyGraph calls helper method with correct args.""" |
| 121 | get_dep = self.PatchObject( |
| 122 | dependency_service, |
| 123 | 'GetBuildDependency', |
| 124 | return_value=(self.json_deps, self.json_deps)) |
| 125 | pkg_mock = 'package-CPV' |
| 126 | pkg_to_cpv = self.PatchObject( |
| 127 | controller_util, |
| 128 | 'PackageInfoToCPV', return_value=pkg_mock) |
| 129 | package = common_pb2.PackageInfo( |
| 130 | package_name='chromeos-chrome', category='chromeos-base') |
| 131 | input_proto = depgraph_pb2.GetBuildDependencyGraphRequest( |
| 132 | build_target=common_pb2.BuildTarget(name='target'), |
| 133 | packages=[package] |
| 134 | ) |
| 135 | dependency.GetBuildDependencyGraph(input_proto, self.response, |
| 136 | self.api_config) |
| 137 | self.assertEqual(self.response.dep_graph.build_target.name, 'deathstar') |
| 138 | pkg_to_cpv.assert_called_once_with(package) |
Alex Klein | d8cd4c6 | 2020-09-14 13:37:47 -0600 | [diff] [blame^] | 139 | get_dep.assert_called_once_with('/build/target', 'target', (pkg_mock,)) |
LaMont Jones | 00b9cff | 2019-12-12 12:14:07 -0700 | [diff] [blame] | 140 | |
Michael Mortensen | af10cca | 2019-11-18 20:21:32 -0700 | [diff] [blame] | 141 | def testValidateOnly(self): |
| 142 | """Sanity check that a validate only call does not execute any logic.""" |
| 143 | patch = self.PatchObject(dependency_service, 'GetBuildDependency') |
| 144 | input_proto = depgraph_pb2.GetBuildDependencyGraphRequest() |
| 145 | input_proto.build_target.name = 'target' |
| 146 | dependency.GetBuildDependencyGraph(input_proto, self.response, |
| 147 | self.validate_only_config) |
| 148 | patch.assert_not_called() |
| 149 | |
| 150 | def testMockCall(self): |
| 151 | """Test that a mock call does not execute logic, returns mocked value.""" |
| 152 | patch = self.PatchObject(dependency_service, 'GetBuildDependency') |
| 153 | input_proto = depgraph_pb2.GetBuildDependencyGraphRequest() |
| 154 | input_proto.build_target.name = 'target' |
| 155 | dependency.GetBuildDependencyGraph(input_proto, self.response, |
| 156 | self.mock_call_config) |
| 157 | patch.assert_not_called() |
| 158 | self.assertEqual(self.response.dep_graph.build_target.name, 'target_board') |
Navil Perez | a782326 | 2020-08-07 17:56:57 +0000 | [diff] [blame] | 159 | |
| 160 | |
| 161 | class ListTest(cros_test_lib.MockTempDirTestCase, api_config.ApiConfigMixin): |
| 162 | """Unittests for the List endpoint.""" |
| 163 | |
| 164 | def setUp(self): |
| 165 | self.response = depgraph_pb2.ListResponse() |
| 166 | self.build_target = common_pb2.BuildTarget(name='target') |
| 167 | self.sysroot = os.path.join(self.tempdir, 'target') |
| 168 | osutils.SafeMakedirs(self.sysroot) |
| 169 | |
| 170 | def testValidateOnly(self): |
| 171 | """Sanity check that a validate only call does not execute any logic.""" |
| 172 | sysroot = sysroot_pb2.Sysroot( |
| 173 | path=self.sysroot, build_target=self.build_target) |
| 174 | input_proto = depgraph_pb2.ListRequest(sysroot=sysroot) |
| 175 | dependency.List(input_proto, self.response, self.validate_only_config) |
| 176 | |
| 177 | def testArgumentValidationMissingSysrootPath(self): |
| 178 | """Test missing sysroot path.""" |
| 179 | sysroot = sysroot_pb2.Sysroot(build_target=self.build_target) |
| 180 | input_proto = depgraph_pb2.ListRequest(sysroot=sysroot) |
| 181 | with self.assertRaises(cros_build_lib.DieSystemExit): |
| 182 | dependency.List(input_proto, self.response, self.api_config) |
| 183 | |
| 184 | def testArgumentValidationMissingBuildTarget(self): |
| 185 | """Test missing build target name.""" |
| 186 | sysroot = sysroot_pb2.Sysroot( |
| 187 | path=self.sysroot, build_target=common_pb2.BuildTarget()) |
| 188 | input_proto = depgraph_pb2.ListRequest(sysroot=sysroot) |
| 189 | with self.assertRaises(cros_build_lib.DieSystemExit): |
| 190 | dependency.List(input_proto, self.response, self.api_config) |
| 191 | |
| 192 | def testListResponse(self): |
| 193 | """Test calls helper method with correct args.""" |
| 194 | mock_get_deps = self.PatchObject( |
| 195 | dependency_service, 'GetDependencies', return_value=['foo/bar-1.2.3']) |
| 196 | sysroot = sysroot_pb2.Sysroot( |
| 197 | path=self.sysroot, build_target=self.build_target) |
| 198 | path = '/path' |
| 199 | package = common_pb2.PackageInfo(category='foo', package_name='bar') |
| 200 | input_proto = depgraph_pb2.ListRequest( |
| 201 | sysroot=sysroot, |
| 202 | src_paths=[ |
| 203 | depgraph_pb2.SourcePath(path='/path'), |
| 204 | ], |
| 205 | packages=[package]) |
| 206 | dependency.List(input_proto, self.response, self.api_config) |
| 207 | mock_get_deps.assert_called_once_with( |
| 208 | self.sysroot, |
| 209 | build_target=controller_util.ParseBuildTarget(self.build_target), |
| 210 | src_paths=[path], |
| 211 | packages=[controller_util.PackageInfoToCPV(package)]) |
| 212 | expected_deps = [ |
| 213 | common_pb2.PackageInfo( |
| 214 | category='foo', package_name='bar', version='1.2.3') |
| 215 | ] |
| 216 | self.assertCountEqual(expected_deps, self.response.package_deps) |