# Copyright 2021 The ChromiumOS Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""Tests for depgraph."""

from chromite.lib import dependency_graph
from chromite.lib import depgraph


def _get_trees():
    """Get the hand-built deps trees for tests.

    Returns:
        tuple: (deps tree with bdeps,
            deps tree without bdeps,
            bdeps tree,
            sdk tree)
    """
    sysroot = "/build/target/"
    # Quick access to complete deps entries for each package.
    deps = {
        "virtual/sysroot-1": [
            {
                "action": "merge",
                "deptypes": [],
                "deps": {},
                "root": sysroot,
            }
        ],
        "virtual/sdk-1": [
            {
                "action": "merge",
                "deptypes": [],
                "deps": {},
                "root": "/",
            }
        ],
        "sysroot/both-1": [
            {
                "action": "merge",
                "deptypes": [],
                "deps": {},
                "root": "/",
            },
            {
                "action": "merge",
                "deptypes": [],
                "deps": {},
                "root": sysroot,
            },
        ],
        "sysroot/both-1-sysroot": [
            {
                "action": "merge",
                "deptypes": [],
                "deps": {},
                "root": sysroot,
            }
        ],
        "sysroot/dep-1": [
            {
                "action": "merge",
                "deptypes": [],
                "deps": {},
                "root": sysroot,
            },
            {
                "action": "uninstall",
                "deptypes": [],
                "deps": {},
                "root": sysroot,
            },
        ],
        "sysroot/uninstall-dep-1": [
            {
                "action": "uninstall",
                "deptypes": [],
                "deps": {},
                "root": sysroot,
            }
        ],
        "sysroot/bdep-1": [
            {
                "action": "merge",
                "deptypes": [],
                "deps": {},
                "root": "/",
            }
        ],
        "sdk/dep-1": [
            {
                "action": "merge",
                "deptypes": [],
                "deps": {},
                "root": "/",
            }
        ],
        "sdk/bdep-1": [
            {
                "action": "merge",
                "deptypes": [],
                "deps": {},
                "root": "/",
            }
        ],
    }
    # The build target tree, deps + bdeps.
    full_dep_tree = {
        "virtual/sysroot-1": [
            {
                "action": "merge",
                "deps": {
                    "sysroot/dep-1": deps["sysroot/dep-1"],
                    "sysroot/bdep-1": deps["sysroot/bdep-1"],
                    "sysroot/both-1": deps["sysroot/both-1"],
                },
                "root": sysroot,
            }
        ],
        "sysroot/dep-1": [
            {
                "action": "merge",
                "deps": {},
                "root": sysroot,
            },
            {
                "action": "uninstall",
                "deps": {
                    "sysroot/uninstall-dep-1": deps["sysroot/uninstall-dep-1"],
                },
                "root": sysroot,
            },
        ],
        "sysroot/uninstall-dep-1": [
            {
                "action": "uninstall",
                "deps": {},
                "root": sysroot,
            },
        ],
        "sysroot/bdep-1": [
            {
                "action": "merge",
                "deps": {},
                "root": "/",
            }
        ],
        "sysroot/both-1": [
            {
                "action": "merge",
                "deps": {},
                "root": "/",
            },
            {
                "action": "merge",
                "deps": {},
                "root": sysroot,
            },
        ],
    }
    # The sysroot tree, only deps.
    no_bdeps_dep_tree = {
        "virtual/sysroot-1": [
            {
                "action": "merge",
                "deps": {
                    "sysroot/dep-1": deps["sysroot/dep-1"],
                    "sysroot/both-1": deps["sysroot/both-1-sysroot"],
                },
                "root": sysroot,
            }
        ],
        "sysroot/both-1": [
            {
                "action": "merge",
                "deps": {},
                "root": sysroot,
            }
        ],
        "sysroot/dep-1": [
            {
                "action": "merge",
                "deps": {},
                "root": sysroot,
            }
        ],
    }
    # The bdeps tree for build target and sysroot trees.
    bdep_tree = {
        "sysroot/both-1": [
            {
                "action": "merge",
                "deps": {},
                "root": "/",
            }
        ],
        "sysroot/bdep-1": [
            {
                "action": "merge",
                "deps": {},
                "root": "/",
            }
        ],
    }
    # The SDK tree.
    sdk_tree = {
        "virtual/sdk-1": [
            {
                "action": "merge",
                "deps": {
                    "sdk/dep-1": deps["sdk/dep-1"],
                    "sdk/bdep-1": deps["sdk/bdep-1"],
                },
                "root": "/",
            }
        ],
        "sdk/dep-1": [
            {
                "action": "merge",
                "deps": {},
                "root": "/",
            }
        ],
        "sdk/bdep-1": [
            {
                "action": "merge",
                "deps": {},
                "root": "/",
            }
        ],
    }
    return full_dep_tree, no_bdeps_dep_tree, bdep_tree, sdk_tree


def _get_raw_sdk_depgraph(*_args, **_kwargs):
    """Build a DepgraphResult instance corresponding to the test SDK data."""
    _, _, _, sdk_tree = _get_trees()
    return depgraph.DepgraphResult(sdk_tree, {}, ["virtual/sdk-1"])


def _get_raw_sysroot_depgraph(*_args, **_kwargs):
    """Get a DepgraphResult instance corresponding to the test sysroot data."""
    _, deps_tree, bdeps_tree, _ = _get_trees()
    return depgraph.DepgraphResult(deps_tree, bdeps_tree, ["virtual/sysroot-1"])


def _get_raw_build_target_depgraph(*_args, **_kwargs):
    """Get a DepgraphResult instance corresponding to the full depgraph data."""
    deps_tree, _, bdeps_tree, _ = _get_trees()
    return depgraph.DepgraphResult(deps_tree, bdeps_tree, ["virtual/sysroot-1"])


def test_get_sdk_dependency_graph(monkeypatch) -> None:
    """Test the SDK depgraph is built correctly."""
    monkeypatch.setattr(
        depgraph, "_get_raw_sdk_depgraph", _get_raw_sdk_depgraph
    )
    graph = depgraph.get_sdk_dependency_graph()

    assert "virtual/sdk-1" in graph
    assert "sdk/dep-1" in graph
    assert "sdk/bdep-1" in graph
    assert "virtual/sysroot-1" not in graph
    assert "sysroot/both-1" not in graph
    assert "sysroot/dep-1" not in graph
    assert "sysroot/bdep-1" not in graph
    assert graph.sysroot_path == "/"


def test_get_sysroot_dependency_graph(monkeypatch) -> None:
    """Test the sysroot depgraph is built correctly."""
    monkeypatch.setattr(
        depgraph, "_get_raw_sysroot_depgraph", _get_raw_sysroot_depgraph
    )
    graph = depgraph.get_sysroot_dependency_graph("/build/target")

    assert "virtual/sysroot-1" in graph
    assert "sysroot/both-1" in graph
    assert "sysroot/dep-1" in graph
    assert "sysroot/bdep-1" not in graph
    assert "virtual/sdk-1" not in graph
    assert "sdk/dep-1" not in graph
    assert "sdk/bdep-1" not in graph
    assert graph.sysroot_path == "/build/target"
    assert not graph.sdk_root
    assert len(graph.get_nodes(["sysroot/both-1"])) == 1
    assert graph.get_nodes(
        ["sysroot/both-1"], dependency_graph.RootType.SYSROOT
    )
    assert not graph.get_nodes(
        ["sysroot/both-1"], dependency_graph.RootType.SDK
    )
    # We currently skip uninstalls.
    assert "sysroot/uninstall-dep-1" not in graph
    assert (
        "sysroot/uninstall-dep-1"
        not in graph.get_nodes(["sysroot/dep-1"])[0].dependencies
    )


def test_get_build_target_dependency_graph(monkeypatch) -> None:
    """Test the build target depgraph is built correctly."""
    monkeypatch.setattr(
        depgraph,
        "_get_raw_build_target_depgraph",
        _get_raw_build_target_depgraph,
    )
    graph = depgraph.get_build_target_dependency_graph("/build/target")

    assert "virtual/sysroot-1" in graph
    assert "sysroot/both-1" in graph
    assert "sysroot/dep-1" in graph
    assert "sysroot/bdep-1" in graph
    assert "virtual/sdk-1" not in graph
    assert "sdk/dep-1" not in graph
    assert "sdk/bdep-1" not in graph
    assert graph.sysroot_path == "/build/target"
    assert graph.sdk_root
    assert len(graph.get_nodes(["sysroot/both-1"])) == 2
    assert graph.get_nodes(
        ["sysroot/both-1"], dependency_graph.RootType.SYSROOT
    )
    assert graph.get_nodes(["sysroot/both-1"], dependency_graph.RootType.SDK)