Files
skills/tests/test_skill_cleanup.py
T

120 lines
4.4 KiB
Python
Raw Normal View History

"""Exercise legacy.py: renames chained through history, deletions, project and unknown skills."""
from __future__ import annotations
import json
import shutil
import subprocess
import sys
import tempfile
import unittest
from pathlib import Path
REPO_ROOT = Path(__file__).resolve().parents[1]
SCRIPT = REPO_ROOT / "skills" / "kvb-skill-cleanup" / "scripts" / "legacy.py"
@unittest.skipUnless(shutil.which("git"), "git not installed")
class TestLegacy(unittest.TestCase):
def setUp(self):
self.tmp = tempfile.TemporaryDirectory()
self.addCleanup(self.tmp.cleanup)
base = Path(self.tmp.name)
self.env = {
"PATH": "/usr/bin:/bin",
"HOME": str(base),
"GIT_AUTHOR_NAME": "t",
"GIT_AUTHOR_EMAIL": "t@t",
"GIT_COMMITTER_NAME": "t",
"GIT_COMMITTER_EMAIL": "t@t",
}
self.repo = base / "skills-repo"
self._skill(self.repo / "skills" / "old")
(self.repo / "skills-lock.json").write_text(
json.dumps({"skills": {"caveman": {}}}), encoding="utf-8"
)
self._git(self.repo, "init", "-q", "-b", "main")
self._commit(self.repo, "add old")
self._git(self.repo, "mv", "skills/old", "skills/mid")
self._commit(self.repo, "rename old to mid")
self._git(self.repo, "mv", "skills/mid", "skills/kvb-x-new")
self._commit(self.repo, "rename mid to kvb-x-new")
self._skill(self.repo / "skills" / "gone")
self._commit(self.repo, "add gone")
self._git(self.repo, "rm", "-q", "-r", "skills/gone")
self._commit(self.repo, "remove gone")
self.home = base / "home" / ".claude" / "skills"
for name in ("old", "mid", "gone", "kvb-x-new", "caveman", "mine"):
self._skill(self.home / name)
(self.home / "notaskill").mkdir()
self.project = base / "project"
self._skill(self.project / "skills" / "p-own")
self._git(self.project, "init", "-q", "-b", "main")
self._commit(self.project, "own skill")
self.installed = self.project / ".claude" / "skills"
for name in ("p-own", "old"):
self._skill(self.installed / name)
def _skill(self, path: Path) -> None:
path.mkdir(parents=True, exist_ok=True)
(path / "SKILL.md").write_text(
f"---\nname: {path.name}\ndescription: probe\n---\n\nBody of the probe skill.\n",
encoding="utf-8",
)
def _git(self, cwd: Path, *args: str) -> None:
subprocess.run(["git", *args], check=True, capture_output=True, env=self.env, cwd=cwd)
def _commit(self, cwd: Path, message: str) -> None:
self._git(cwd, "add", "-A")
self._git(cwd, "commit", "-q", "-m", message)
def _run(self, *args: str) -> subprocess.CompletedProcess:
return subprocess.run(
[sys.executable, str(SCRIPT), *args],
check=False,
capture_output=True,
text=True,
env=self.env,
)
def test_every_installed_skill_gets_its_verdict(self):
result = self._run(
"--skills-repo", str(self.repo),
"--root", str(self.home),
"--root", str(self.installed),
"--json",
)
self.assertEqual(result.returncode, 0, result.stderr)
home, project = str(self.home), str(self.installed)
rows = sorted(
(str(Path(row["path"]).parent), Path(row["path"]).name, row["verdict"], row["successor"])
for row in json.loads(result.stdout)
)
self.assertEqual(
rows,
sorted(
[
(home, "caveman", "current", None),
(home, "gone", "removed", None),
(home, "kvb-x-new", "current", None),
(home, "mid", "legacy", "kvb-x-new"),
(home, "mine", "unknown", None),
(home, "old", "legacy", "kvb-x-new"),
(project, "old", "legacy", "kvb-x-new"),
(project, "p-own", "project", None),
]
),
)
def test_invalid_input_exits_2(self):
missing = self._run("--skills-repo", str(self.repo), "--root", str(self.home / "nope"))
self.assertEqual(missing.returncode, 2)
no_repo = self._run("--skills-repo", str(self.home), "--root", str(self.home))
self.assertEqual(no_repo.returncode, 2)
if __name__ == "__main__":
unittest.main()