| # 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. |
| |
| """Unittests for the dependency.py module.""" |
| |
| import pytest |
| |
| from chromite.lib import cros_test_lib |
| from chromite.lib import depgraph |
| from chromite.lib import dependency_graph |
| from chromite.lib import dependency_lib |
| from chromite.lib.parser import package_info |
| from chromite.service import dependency |
| |
| pytestmark = cros_test_lib.pytestmark_inside_only |
| |
| |
| class DependencyTests(cros_test_lib.MockTestCase): |
| """General unittests for dependency module.""" |
| |
| def _build_sysroot_depgraph(self): |
| """Build a DependencyGraph to test against.""" |
| sysroot = '/build/target' |
| |
| virtual = package_info.parse('virtual/target-foo-1.2.3') |
| dep1 = package_info.parse('cat/dep-1.0.0-r1') |
| dep2 = package_info.parse('cat/dep-2.0.0-r1') |
| depdep = package_info.parse('cat/depdep-2.0.1-r5') |
| |
| virtual_node = dependency_graph.PackageNode( |
| virtual, sysroot, src_paths=['/virtual/foo']) |
| dep1_node = dependency_graph.PackageNode( |
| dep1, sysroot, src_paths=['/cat/dep/one']) |
| dep2_node = dependency_graph.PackageNode( |
| dep2, sysroot, src_paths=['/cat/dep/two']) |
| depdep_node = dependency_graph.PackageNode( |
| depdep, sysroot, src_paths=['/cat/depdep', '/other/depdep']) |
| |
| virtual_node.add_dependency(dep1_node) |
| virtual_node.add_dependency(dep2_node) |
| dep1_node.add_dependency(depdep_node) |
| |
| nodes = (virtual_node, dep1_node, dep2_node, depdep_node) |
| |
| return dependency_graph.DependencyGraph(nodes, sysroot, [virtual]) |
| |
| def setUp(self): |
| 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 testDeterminePackageRelevanceNotRelevant(self): |
| """Test determine_package_relevance with no matching paths.""" |
| src_paths = ['foo/bar/baz', 'foo/bar/b', 'foo/bar', 'bar/foo'] |
| dep_src_paths = ['foo/bar/ba'] |
| self.assertFalse( |
| dependency.determine_package_relevance(dep_src_paths, src_paths)) |
| |
| def testDeterminePackageRelevanceExactMatch(self): |
| """Test determine_package_relevance given an exact match.""" |
| src_paths = ['foo/bar/baz'] |
| dep_src_paths = ['foo/bar/baz'] |
| self.assertTrue( |
| dependency.determine_package_relevance(dep_src_paths, src_paths)) |
| |
| def testDeterminePackageRelevanceDirectoryMatch(self): |
| """Test determine_package_relevance given a directory match.""" |
| src_paths = ['foo/bar/baz'] |
| dep_src_paths = ['foo/bar'] |
| self.assertTrue( |
| dependency.determine_package_relevance(dep_src_paths, src_paths)) |
| |
| def testGetDependenciesWithDefaultArgs(self): |
| """Test GetDependencies using the default args.""" |
| self.PatchObject( |
| depgraph, |
| 'get_sysroot_dependency_graph', |
| return_value=(self._build_sysroot_depgraph())) |
| sysroot_path = '/build/target' |
| actual_deps = dependency.GetDependencies(sysroot_path) |
| dep1 = package_info.parse('cat/dep-1.0.0-r1') |
| dep2 = package_info.parse('cat/dep-2.0.0-r1') |
| virtual = package_info.parse('virtual/target-foo-1.2.3') |
| depdep = package_info.parse('cat/depdep-2.0.1-r5') |
| |
| expected_deps = [dep1, dep2, virtual, depdep] |
| self.assertCountEqual(expected_deps, actual_deps) |
| |
| def testGetDependenciesWithSrcPaths(self): |
| """Test GetDependencies given a list of paths.""" |
| self.PatchObject( |
| depgraph, |
| 'get_sysroot_dependency_graph', |
| return_value=(self._build_sysroot_depgraph())) |
| sysroot_path = '/build/target' |
| src_paths = ['/cat/dep/one'] |
| actual_deps = dependency.GetDependencies(sysroot_path, src_paths) |
| dep = package_info.parse('cat/dep-1.0.0-r1') |
| self.assertCountEqual([dep], actual_deps) |
| |
| def testGetDependenciesWithSrcPathsAndReverseDeps(self): |
| """Test GetDependencies given a list of paths.""" |
| self.PatchObject( |
| depgraph, |
| 'get_sysroot_dependency_graph', |
| return_value=(self._build_sysroot_depgraph())) |
| sysroot_path = '/build/target' |
| src_paths = ['/cat/dep/one'] |
| actual_deps = dependency.GetDependencies( |
| sysroot_path, src_paths, include_rev_dependencies=True) |
| |
| revdep = package_info.parse('virtual/target-foo-1.2.3') |
| dep = package_info.parse('cat/dep-1.0.0-r1') |
| self.assertCountEqual([dep, revdep], actual_deps) |
| |
| |
| def test_generate_source_path_mapping_sdk(monkeypatch): |
| """Test GenerateSourcePathMapping sdk argument.""" |
| |
| def gspm_patch(_packages, sysroot_path, board, *_args, **_kwargs): |
| assert sysroot_path == '/' |
| assert board is None |
| |
| monkeypatch.setattr(dependency_lib, 'get_source_path_mapping', gspm_patch) |
| dependency.GenerateSourcePathMapping(['cat/pkg'], sdk=True) |
| |
| |
| def test_generate_source_path_mapping_sdk_only(): |
| """Test GenerateSourcePathMapping raises errors when sdk not only argument.""" |
| # Board. |
| with pytest.raises(AssertionError): |
| dependency.GenerateSourcePathMapping(['cat/pkg'], |
| sdk=True, |
| board='board') |
| # Sysroot. |
| with pytest.raises(AssertionError): |
| dependency.GenerateSourcePathMapping(['cat/pkg'], |
| sdk=True, |
| sysroot_path='/build/board') |
| # Board and sysroot. |
| with pytest.raises(AssertionError): |
| dependency.GenerateSourcePathMapping(['cat/pkg'], |
| sdk=True, |
| board='board', |
| sysroot_path='/build/board') |
| |
| |
| def test_generate_source_path_mapping_sdk_sysroot(monkeypatch): |
| """Test GenerateSourcePathMapping with the sdk's sysroot.""" |
| |
| def gspm_patch(_packages, sysroot_path, board, *_args, **_kwargs): |
| assert sysroot_path == '/' |
| assert board is None |
| |
| monkeypatch.setattr(dependency_lib, 'get_source_path_mapping', gspm_patch) |
| dependency.GenerateSourcePathMapping(['cat/pkg'], sysroot_path='/') |
| |
| |
| def test_generate_source_path_mapping_board_sysroot(monkeypatch): |
| """Test GenerateSourcePathMapping with a board's sysroot.""" |
| |
| def gspm_patch(_packages, sysroot_path, board, *_args, **_kwargs): |
| assert sysroot_path == '/build/board' |
| assert board == 'board' |
| |
| monkeypatch.setattr(dependency_lib, 'get_source_path_mapping', gspm_patch) |
| dependency.GenerateSourcePathMapping(['cat/pkg'], sysroot_path='/build/board') |
| |
| |
| def test_generate_source_path_mapping_board(monkeypatch): |
| """Test GenerateSourcePathMapping with a board.""" |
| |
| def gspm_patch(_packages, sysroot_path, board, *_args, **_kwargs): |
| assert sysroot_path == '/build/board' |
| assert board == 'board' |
| |
| monkeypatch.setattr(dependency_lib, 'get_source_path_mapping', gspm_patch) |
| dependency.GenerateSourcePathMapping(['cat/pkg'], board='board') |
| |
| |
| def test_generate_source_path_mapping_board_and_sysroot(monkeypatch): |
| """Test GenerateSourcePathMapping with a board and custom sysroot.""" |
| |
| def gspm_patch(_packages, sysroot_path, board, *_args, **_kwargs): |
| assert sysroot_path == '/some/sysroot' |
| assert board == 'board' |
| |
| monkeypatch.setattr(dependency_lib, 'get_source_path_mapping', gspm_patch) |
| dependency.GenerateSourcePathMapping(['cat/pkg'], |
| board='board', |
| sysroot_path='/some/sysroot') |
