mirror of
https://github.com/safishamsi/graphify.git
synced 2026-07-12 10:27:11 +00:00
Danil Tarasov
52 lines
1.8 KiB
Python
52 lines
1.8 KiB
Python
"""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
|