from __future__ import annotations

from textwrap import dedent
from typing import TYPE_CHECKING

from pipdeptree._models.dag import PackageDAG
from pipdeptree._render.json import render_json

if TYPE_CHECKING:
    from collections.abc import Callable, Iterator
    from unittest.mock import Mock

    import pytest

    from tests.our_types import MockGraph


def test_render_json(mock_pkgs: Callable[[MockGraph], Iterator[Mock]], capsys: pytest.CaptureFixture[str]) -> None:
    graph: MockGraph = {
        ("a", "1.2.3"): [("b", [(">=", "4.0.0")])],
        ("b", "4.5.6"): [],
    }
    expected_output = dedent("""\
     [
         {
             "package": {
                 "key": "a",
                 "package_name": "a",
                 "installed_version": "1.2.3"
             },
             "dependencies": [
                 {
                     "key": "b",
                     "package_name": "b",
                     "installed_version": "4.5.6",
                     "required_version": ">=4.0.0"
                 }
             ]
         },
         {
             "package": {
                 "key": "b",
                 "package_name": "b",
                 "installed_version": "4.5.6"
             },
             "dependencies": []
         }
     ]
    """)
    package_dag = PackageDAG.from_pkgs(list(mock_pkgs(graph)))

    render_json(package_dag)

    output = capsys.readouterr()
    assert output.out == expected_output