"""Tests for graphify query CLI context filtering."""

from __future__ import annotations

import json

import networkx as nx

from networkx.readwrite import json_graph

import graphify.__main__ as mainmod

def _write_graph(tmp_path):

G = nx.Graph()

G.add_node("n1", label="extract", source_file="extract.py", source_location="L10", community=0)

G.add_node("n2", label="cluster", source_file="cluster.py", source_location="L5", community=0)

G.add_node("n3", label="build", source_file="build.py", source_location="L1", community=1)

G.add_edge("n1", "n2", relation="calls", confidence="EXTRACTED", context="call")

G.add_edge("n2", "n3", relation="imports", confidence="EXTRACTED", context="import")

graph_path = tmp_path / "graph.json"

graph_path.write_text(json.dumps(json_graph.node_link_data(G, edges="links")))

return graph_path

def test_query_cli_explicit_context_filter(monkeypatch, tmp_path, capsys):

graph_path = _write_graph(tmp_path)

monkeypatch.setattr(mainmod, "_check_skill_version", lambda _: None)

monkeypatch.setattr(

mainmod.sys,

"argv",

["graphify", "query", "extract", "--context", "call", "--graph", str(graph_path)],

)

mainmod.main()

out = capsys.readouterr().out

assert "Context: call (explicit)" in out

assert "cluster" in out

assert "build" not in out

def test_query_cli_heuristic_context_filter(monkeypatch, tmp_path, capsys):

graph_path = _write_graph(tmp_path)

monkeypatch.setattr(mainmod, "_check_skill_version", lambda _: None)

monkeypatch.setattr(

mainmod.sys,

"argv",

["graphify", "query", "who calls extract", "--graph", str(graph_path)],

)

mainmod.main()

out = capsys.readouterr().out

assert "Context: call (heuristic)" in out

assert "cluster" in out

assert "build" not in out

def test_query_cli_rejects_oversized_graph(monkeypatch, tmp_path, capsys):

"""#F4: query CLI must refuse to parse a graph.json that exceeds the cap."""

import pytest

graph_path = _write_graph(tmp_path)

monkeypatch.setattr(mainmod, "_check_skill_version", lambda _: None)

monkeypatch.setattr("graphify.security._MAX_GRAPH_FILE_BYTES", 16)

monkeypatch.setattr(

mainmod.sys,

"argv",

["graphify", "query", "extract", "--graph", str(graph_path)],

)

with pytest.raises(SystemExit):

mainmod.main()

err = capsys.readouterr().err

assert "exceeds" in err

assert "byte cap" in err