mirror of
https://github.com/safishamsi/graphify.git
synced 2026-07-12 10:27:11 +00:00
balloon72
f0d29a1c6d
* fix(codex): keep graph-first guidance with dirty graph output * fix(codex): include dirty graph guidance in agents install --------- Co-authored-by: hanmo1 <hanmo1@lenovo.com>
384 lines
14 KiB
Python
384 lines
14 KiB
Python
"""Tests for graphify install --platform routing."""
|
|
import os
|
|
from pathlib import Path
|
|
import sys
|
|
from unittest.mock import patch
|
|
import pytest
|
|
|
|
|
|
PLATFORMS = {
|
|
"claude": (".claude/skills/graphify/SKILL.md",),
|
|
"codex": (".agents/skills/graphify/SKILL.md",),
|
|
"opencode": (".config/opencode/skills/graphify/SKILL.md",),
|
|
"claw": (".openclaw/skills/graphify/SKILL.md",),
|
|
"droid": (".factory/skills/graphify/SKILL.md",),
|
|
"trae": (".trae/skills/graphify/SKILL.md",),
|
|
"trae-cn": (".trae-cn/skills/graphify/SKILL.md",),
|
|
"windows": (".claude/skills/graphify/SKILL.md",),
|
|
}
|
|
|
|
|
|
def _install(tmp_path, platform):
|
|
from graphify.__main__ import install
|
|
old_cwd = Path.cwd()
|
|
try:
|
|
os.chdir(tmp_path)
|
|
with patch("graphify.__main__.Path.home", return_value=tmp_path):
|
|
install(platform=platform)
|
|
finally:
|
|
os.chdir(old_cwd)
|
|
|
|
|
|
def test_install_default_claude(tmp_path):
|
|
_install(tmp_path, "claude")
|
|
assert (tmp_path / ".claude" / "skills" / "graphify" / "SKILL.md").exists()
|
|
|
|
|
|
def test_install_codex(tmp_path):
|
|
_install(tmp_path, "codex")
|
|
assert (tmp_path / ".agents" / "skills" / "graphify" / "SKILL.md").exists()
|
|
|
|
|
|
def test_install_opencode(tmp_path):
|
|
_install(tmp_path, "opencode")
|
|
assert (tmp_path / ".config" / "opencode" / "skills" / "graphify" / "SKILL.md").exists()
|
|
|
|
|
|
def test_install_positional_platform_opencode(tmp_path, monkeypatch):
|
|
from graphify.__main__ import main
|
|
monkeypatch.chdir(tmp_path)
|
|
monkeypatch.setattr(sys, "argv", ["graphify", "install", "opencode"])
|
|
with patch("graphify.__main__.Path.home", return_value=tmp_path):
|
|
main()
|
|
assert (tmp_path / ".config" / "opencode" / "skills" / "graphify" / "SKILL.md").exists()
|
|
assert not (tmp_path / ".claude" / "skills" / "graphify" / "SKILL.md").exists()
|
|
|
|
|
|
def test_install_help_does_not_install_default(tmp_path, monkeypatch, capsys):
|
|
from graphify.__main__ import main
|
|
monkeypatch.chdir(tmp_path)
|
|
monkeypatch.setattr(sys, "argv", ["graphify", "install", "opencode", "--help"])
|
|
with patch("graphify.__main__.Path.home", return_value=tmp_path):
|
|
main()
|
|
out = capsys.readouterr().out
|
|
assert "Usage: graphify install" in out
|
|
assert "opencode" in out
|
|
assert not (tmp_path / ".claude").exists()
|
|
assert not (tmp_path / ".config").exists()
|
|
|
|
|
|
def test_install_claw(tmp_path):
|
|
_install(tmp_path, "claw")
|
|
assert (tmp_path / ".openclaw" / "skills" / "graphify" / "SKILL.md").exists()
|
|
|
|
|
|
def test_install_droid(tmp_path):
|
|
_install(tmp_path, "droid")
|
|
assert (tmp_path / ".factory" / "skills" / "graphify" / "SKILL.md").exists()
|
|
|
|
|
|
def test_install_trae(tmp_path):
|
|
_install(tmp_path, "trae")
|
|
assert (tmp_path / ".trae" / "skills" / "graphify" / "SKILL.md").exists()
|
|
|
|
|
|
def test_install_trae_cn(tmp_path):
|
|
_install(tmp_path, "trae-cn")
|
|
assert (tmp_path / ".trae-cn" / "skills" / "graphify" / "SKILL.md").exists()
|
|
|
|
|
|
def test_install_windows(tmp_path):
|
|
_install(tmp_path, "windows")
|
|
assert (tmp_path / ".claude" / "skills" / "graphify" / "SKILL.md").exists()
|
|
|
|
|
|
def test_install_unknown_platform_exits(tmp_path):
|
|
with pytest.raises(SystemExit):
|
|
_install(tmp_path, "unknown")
|
|
|
|
|
|
def test_codex_skill_contains_spawn_agent():
|
|
"""Codex skill file must reference spawn_agent."""
|
|
import graphify
|
|
skill = (Path(graphify.__file__).parent / "skill-codex.md").read_text()
|
|
assert "spawn_agent" in skill
|
|
|
|
|
|
def test_codex_skill_uses_graphify_with_dirty_graph_output():
|
|
"""Codex skill must keep graph-first orientation even when graph output is dirty."""
|
|
import graphify
|
|
skill = (Path(graphify.__file__).parent / "skill-codex.md").read_text()
|
|
assert "Dirty `graphify-out/` artifacts are expected" in skill
|
|
assert "not a reason to skip Graphify" in skill
|
|
assert "graphify query" in skill
|
|
assert "graphify explain" in skill
|
|
assert "graphify path" in skill
|
|
|
|
|
|
def test_codex_agents_install_mentions_dirty_graph_output(tmp_path):
|
|
_agents_install(tmp_path, "codex")
|
|
content = (tmp_path / "AGENTS.md").read_text()
|
|
assert "Dirty graphify-out/ files are expected" in content
|
|
assert "not a reason to skip graphify" in content
|
|
|
|
|
|
def test_opencode_skill_contains_mention():
|
|
"""OpenCode skill file must reference @mention."""
|
|
import graphify
|
|
skill = (Path(graphify.__file__).parent / "skill-opencode.md").read_text()
|
|
assert "@mention" in skill
|
|
|
|
|
|
def test_opencode_skill_uses_opencode_agent_guidance():
|
|
"""OpenCode skill must not reference Codex/Claude agent type names."""
|
|
import graphify
|
|
skill = (Path(graphify.__file__).parent / "skill-opencode.md").read_text()
|
|
assert "general-purpose" not in skill
|
|
assert 'subagent_type="general-purpose"' not in skill
|
|
assert "@agent" in skill
|
|
assert "serial fallback" in skill
|
|
assert "reduce semantic chunks to 10-12 files each" in skill
|
|
assert "10-12 files each if the smaller-chunk large-corpus policy was applied" in skill
|
|
assert "process chunks one at a time" in skill
|
|
assert "Wait for the user's answer before proceeding" not in skill
|
|
|
|
|
|
def test_claw_skill_is_sequential():
|
|
"""OpenClaw skill file must describe sequential extraction."""
|
|
import graphify
|
|
skill = (Path(graphify.__file__).parent / "skill-claw.md").read_text()
|
|
assert "sequential" in skill.lower()
|
|
assert "spawn_agent" not in skill
|
|
assert "@mention" not in skill
|
|
|
|
|
|
def test_all_skill_files_exist_in_package():
|
|
"""All installable platform skill files must be present in the installed package."""
|
|
import graphify
|
|
pkg = Path(graphify.__file__).parent
|
|
for name in ("skill.md", "skill-codex.md", "skill-opencode.md", "skill-claw.md", "skill-windows.md", "skill-droid.md", "skill-trae.md"):
|
|
assert (pkg / name).exists(), f"Missing: {name}"
|
|
|
|
|
|
def test_claude_install_registers_claude_md(tmp_path):
|
|
"""Claude platform install writes CLAUDE.md; others do not."""
|
|
_install(tmp_path, "claude")
|
|
assert (tmp_path / ".claude" / "CLAUDE.md").exists()
|
|
|
|
|
|
def test_codex_install_does_not_write_claude_md(tmp_path):
|
|
_install(tmp_path, "codex")
|
|
assert not (tmp_path / ".claude" / "CLAUDE.md").exists()
|
|
|
|
|
|
# --- always-on AGENTS.md install/uninstall tests ---
|
|
|
|
def _agents_install(tmp_path, platform):
|
|
from graphify.__main__ import _agents_install as _install_fn
|
|
_install_fn(tmp_path, platform)
|
|
|
|
|
|
def _agents_uninstall(tmp_path, platform=""):
|
|
from graphify.__main__ import _agents_uninstall as _uninstall_fn
|
|
_uninstall_fn(tmp_path, platform=platform)
|
|
|
|
|
|
def test_codex_agents_install_writes_agents_md(tmp_path):
|
|
_agents_install(tmp_path, "codex")
|
|
agents_md = tmp_path / "AGENTS.md"
|
|
assert agents_md.exists()
|
|
assert "graphify" in agents_md.read_text()
|
|
assert "GRAPH_REPORT.md" in agents_md.read_text()
|
|
|
|
|
|
def test_opencode_agents_install_writes_agents_md(tmp_path):
|
|
_agents_install(tmp_path, "opencode")
|
|
assert (tmp_path / "AGENTS.md").exists()
|
|
|
|
|
|
def test_claw_agents_install_writes_agents_md(tmp_path):
|
|
_agents_install(tmp_path, "claw")
|
|
assert (tmp_path / "AGENTS.md").exists()
|
|
|
|
|
|
def test_agents_install_idempotent(tmp_path):
|
|
"""Installing twice does not duplicate the section."""
|
|
_agents_install(tmp_path, "codex")
|
|
_agents_install(tmp_path, "codex")
|
|
content = (tmp_path / "AGENTS.md").read_text()
|
|
assert content.count("## graphify") == 1
|
|
|
|
|
|
def test_agents_install_appends_to_existing(tmp_path):
|
|
"""Installs into an existing AGENTS.md without overwriting other content."""
|
|
agents_md = tmp_path / "AGENTS.md"
|
|
agents_md.write_text("# Existing rules\n\nDo not break things.\n")
|
|
_agents_install(tmp_path, "codex")
|
|
content = agents_md.read_text()
|
|
assert "Do not break things." in content
|
|
assert "## graphify" in content
|
|
|
|
|
|
def test_agents_uninstall_removes_section(tmp_path):
|
|
_agents_install(tmp_path, "codex")
|
|
_agents_uninstall(tmp_path)
|
|
agents_md = tmp_path / "AGENTS.md"
|
|
# File deleted when it only contained graphify section
|
|
assert not agents_md.exists()
|
|
|
|
|
|
def test_agents_uninstall_preserves_other_content(tmp_path):
|
|
"""Uninstall keeps pre-existing content."""
|
|
agents_md = tmp_path / "AGENTS.md"
|
|
agents_md.write_text("# Existing rules\n\nDo not break things.\n")
|
|
_agents_install(tmp_path, "codex")
|
|
_agents_uninstall(tmp_path)
|
|
assert agents_md.exists()
|
|
content = agents_md.read_text()
|
|
assert "Do not break things." in content
|
|
assert "## graphify" not in content
|
|
|
|
|
|
def test_agents_uninstall_no_op_when_not_installed(tmp_path, capsys):
|
|
_agents_uninstall(tmp_path)
|
|
out = capsys.readouterr().out
|
|
assert "nothing to do" in out
|
|
|
|
|
|
# --- OpenCode plugin tests ---
|
|
|
|
def test_opencode_agents_install_writes_plugin(tmp_path):
|
|
"""opencode install writes .opencode/plugins/graphify.js."""
|
|
_agents_install(tmp_path, "opencode")
|
|
plugin = tmp_path / ".opencode" / "plugins" / "graphify.js"
|
|
assert plugin.exists()
|
|
assert "tool.execute.before" in plugin.read_text()
|
|
|
|
|
|
def test_opencode_agents_install_registers_plugin_in_config(tmp_path):
|
|
"""opencode install registers the plugin in .opencode/opencode.json."""
|
|
_agents_install(tmp_path, "opencode")
|
|
config_file = tmp_path / ".opencode" / "opencode.json"
|
|
assert config_file.exists()
|
|
import json as _json
|
|
config = _json.loads(config_file.read_text())
|
|
assert any("graphify.js" in p for p in config.get("plugin", []))
|
|
|
|
|
|
def test_opencode_agents_install_merges_existing_config(tmp_path):
|
|
"""opencode install preserves existing .opencode/opencode.json keys."""
|
|
import json as _json
|
|
config_file = tmp_path / ".opencode" / "opencode.json"
|
|
config_file.parent.mkdir(parents=True, exist_ok=True)
|
|
config_file.write_text(_json.dumps({"model": "claude-opus-4-5", "plugin": []}))
|
|
_agents_install(tmp_path, "opencode")
|
|
config = _json.loads(config_file.read_text())
|
|
assert config["model"] == "claude-opus-4-5"
|
|
assert any("graphify.js" in p for p in config["plugin"])
|
|
|
|
|
|
def test_opencode_agents_uninstall_removes_plugin(tmp_path):
|
|
"""opencode uninstall removes the plugin file and deregisters from opencode.json."""
|
|
import json as _json
|
|
_agents_install(tmp_path, "opencode")
|
|
_agents_uninstall(tmp_path, platform="opencode")
|
|
plugin = tmp_path / ".opencode" / "plugins" / "graphify.js"
|
|
assert not plugin.exists()
|
|
config_file = tmp_path / ".opencode" / "opencode.json"
|
|
if config_file.exists():
|
|
config = _json.loads(config_file.read_text())
|
|
assert not any("graphify.js" in p for p in config.get("plugin", []))
|
|
|
|
|
|
# ── Cursor ────────────────────────────────────────────────────────────────────
|
|
|
|
def test_cursor_install_writes_rule(tmp_path):
|
|
"""cursor install writes .cursor/rules/graphify.mdc."""
|
|
from graphify.__main__ import _cursor_install
|
|
_cursor_install(tmp_path)
|
|
rule = tmp_path / ".cursor" / "rules" / "graphify.mdc"
|
|
assert rule.exists()
|
|
content = rule.read_text()
|
|
assert "alwaysApply: true" in content
|
|
assert "graphify-out/GRAPH_REPORT.md" in content
|
|
|
|
|
|
def test_cursor_install_idempotent(tmp_path):
|
|
"""cursor install does not overwrite an existing rule file."""
|
|
from graphify.__main__ import _cursor_install
|
|
_cursor_install(tmp_path)
|
|
rule = tmp_path / ".cursor" / "rules" / "graphify.mdc"
|
|
original = rule.read_text()
|
|
_cursor_install(tmp_path)
|
|
assert rule.read_text() == original
|
|
|
|
|
|
def test_cursor_uninstall_removes_rule(tmp_path):
|
|
"""cursor uninstall removes the rule file."""
|
|
from graphify.__main__ import _cursor_install, _cursor_uninstall
|
|
_cursor_install(tmp_path)
|
|
_cursor_uninstall(tmp_path)
|
|
rule = tmp_path / ".cursor" / "rules" / "graphify.mdc"
|
|
assert not rule.exists()
|
|
|
|
|
|
def test_cursor_uninstall_noop_if_not_installed(tmp_path):
|
|
"""cursor uninstall does nothing if rule was never written."""
|
|
from graphify.__main__ import _cursor_uninstall
|
|
_cursor_uninstall(tmp_path) # should not raise
|
|
|
|
|
|
# ── Gemini CLI ────────────────────────────────────────────────────────────────
|
|
|
|
def test_gemini_install_writes_gemini_md(tmp_path):
|
|
from graphify.__main__ import gemini_install
|
|
gemini_install(tmp_path)
|
|
md = tmp_path / "GEMINI.md"
|
|
assert md.exists()
|
|
assert "graphify-out/GRAPH_REPORT.md" in md.read_text()
|
|
|
|
def test_gemini_install_writes_hook(tmp_path):
|
|
import json as _json
|
|
from graphify.__main__ import gemini_install
|
|
gemini_install(tmp_path)
|
|
settings = _json.loads((tmp_path / ".gemini" / "settings.json").read_text())
|
|
hooks = settings["hooks"]["BeforeTool"]
|
|
assert any("graphify" in str(h) for h in hooks)
|
|
|
|
def test_gemini_install_idempotent(tmp_path):
|
|
from graphify.__main__ import gemini_install
|
|
gemini_install(tmp_path)
|
|
gemini_install(tmp_path)
|
|
md = tmp_path / "GEMINI.md"
|
|
assert md.read_text().count("## graphify") == 1
|
|
|
|
def test_gemini_install_merges_existing_gemini_md(tmp_path):
|
|
from graphify.__main__ import gemini_install
|
|
(tmp_path / "GEMINI.md").write_text("# My project rules\n")
|
|
gemini_install(tmp_path)
|
|
content = (tmp_path / "GEMINI.md").read_text()
|
|
assert "# My project rules" in content
|
|
assert "graphify-out/GRAPH_REPORT.md" in content
|
|
|
|
def test_gemini_uninstall_removes_section(tmp_path):
|
|
from graphify.__main__ import gemini_install, gemini_uninstall
|
|
gemini_install(tmp_path)
|
|
gemini_uninstall(tmp_path)
|
|
md = tmp_path / "GEMINI.md"
|
|
assert not md.exists()
|
|
|
|
def test_gemini_uninstall_removes_hook(tmp_path):
|
|
import json as _json
|
|
from graphify.__main__ import gemini_install, gemini_uninstall
|
|
gemini_install(tmp_path)
|
|
gemini_uninstall(tmp_path)
|
|
settings_path = tmp_path / ".gemini" / "settings.json"
|
|
if settings_path.exists():
|
|
settings = _json.loads(settings_path.read_text())
|
|
hooks = settings.get("hooks", {}).get("BeforeTool", [])
|
|
assert not any("graphify" in str(h) for h in hooks)
|
|
|
|
def test_gemini_uninstall_noop_if_not_installed(tmp_path):
|
|
from graphify.__main__ import gemini_uninstall
|
|
gemini_uninstall(tmp_path) # should not raise
|