service/dependency_unittest.py

# 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."""

from chromite.lib import cros_test_lib
from chromite.lib import depgraph
from chromite.lib import dependency_graph
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)