# Copyright Spack Project Developers. See COPYRIGHT file for details.
#
# SPDX-License-Identifier: (Apache-2.0 OR MIT)

import re

import pytest

import spack.store
from spack.llnl.util.tty.color import color_when
from spack.main import SpackCommand

dependents = SpackCommand("dependents")


def test_immediate_dependents(mock_packages):
    out = dependents("libelf")
    actual = set(re.split(r"\s+", out.strip()))
    assert actual == set(
        [
            "dyninst",
            "libdwarf",
            "patch-a-dependency",
            "patch-several-dependencies",
            "quantum-espresso",
            "conditionally-patch-dependency",
        ]
    )


def test_transitive_dependents(mock_packages):
    out = dependents("--transitive", "libelf")
    actual = set(re.split(r"\s+", out.strip()))
    assert actual == {
        "callpath",
        "dyninst",
        "libdwarf",
        "mixing-parent",
        "mpileaks",
        "multivalue-variant",
        "singlevalue-variant-dependent",
        "trilinos",
        "patch-a-dependency",
        "patch-several-dependencies",
        "quantum-espresso",
        "conditionally-patch-dependency",
    }


@pytest.mark.db
def test_immediate_installed_dependents(mock_packages, database):
    with color_when(False):
        out = dependents("--installed", "libelf")

    lines = [li for li in out.strip().split("\n") if not li.startswith("--")]
    hashes = set([re.split(r"\s+", li)[0] for li in lines if li])

    expected = set(
        [spack.store.STORE.db.query_one(s).dag_hash(7) for s in ["dyninst", "libdwarf"]]
    )

    libelf = spack.store.STORE.db.query_one("libelf")
    expected = set([d.dag_hash(7) for d in libelf.dependents()])

    assert expected == hashes


@pytest.mark.db
def test_transitive_installed_dependents(mock_packages, database):
    with color_when(False):
        out = dependents("--installed", "--transitive", "fake")

    lines = [li for li in out.strip().split("\n") if li and not li.startswith("--")]
    hashes = set([re.split(r"\s+", li)[0] for li in lines])

    expected = set(
        [
            spack.store.STORE.db.query_one(s).dag_hash(7)
            for s in ["zmpi", "callpath^zmpi", "mpileaks^zmpi"]
        ]
    )

    assert expected == hashes
