import datetime import importlib.util import tempfile import unittest from pathlib import Path ROOT = Path(__file__).resolve().parents[1] def load_script(name: str): path = ROOT / "scripts" / f"{name}.py" spec = importlib.util.spec_from_file_location(f"test_{name}", path) module = importlib.util.module_from_spec(spec) spec.loader.exec_module(module) return module SKILL_TEMPLATE = """--- name: {name} description: {description} --- # 标题 最后复核：{reviewed} 证据来源：测试夹具。 """ class ContextBudgetTests(unittest.TestCase): @classmethod def setUpClass(cls): cls.checker = load_script("check_context_budget") def test_repository_context_files_pass(self): problems, reviewed = self.checker.check_repo(ROOT) self.assertEqual(problems, [], "\n".join(problems)) self.assertTrue(reviewed, "仓库应至少有一个带最后复核日期的技能") def test_current_budgets_stay_under_ceiling(self): self.assertEqual(self.checker.check_budget_history(), []) for name, budget in self.checker.LINE_BUDGET.items(): self.assertLessEqual(budget, self.checker.MAX_EVER[name], name) self.assertLessEqual( self.checker.SKILL_MAX_LINES, self.checker.MAX_EVER["SKILL.md"] ) self.assertLessEqual( self.checker.DESCRIPTION_MAX_CHARS, self.checker.MAX_EVER["description"] ) for name, budget in self.checker.BYTE_BUDGET.items(): self.assertLessEqual(budget, self.checker.MAX_EVER_BYTES[name], name) self.assertLessEqual( self.checker.SKILL_MAX_BYTES, self.checker.MAX_EVER_BYTES["SKILL.md"] ) def test_raising_a_budget_without_a_reason_is_reported(self): original = dict(self.checker.BUDGET_CHANGES) try: self.checker.BUDGET_CHANGES = { "AGENTS.md": [(70, "2026-08-17", "初始值"), (80, "2026-08-18", "初始值")] } problems = self.checker.check_budget_history() finally: self.checker.BUDGET_CHANGES = original self.assertTrue(any("是上调" in problem for problem in problems), problems) def test_budget_above_ceiling_is_reported(self): original = dict(self.checker.BUDGET_CHANGES) try: self.checker.BUDGET_CHANGES = { "AGENTS.md": [(70, "2026-08-17", "初始值"), (300, "2026-08-18", "内容实在下沉不了")] } problems = self.checker.check_budget_history() finally: self.checker.BUDGET_CHANGES = original self.assertTrue(any("越过天花板" in problem for problem in problems), problems) def test_over_budget_entry_is_reported(self): with tempfile.TemporaryDirectory() as tmp: root = Path(tmp) budget = self.checker.LINE_BUDGET["AGENTS.md"] (root / "AGENTS.md").write_text("行\n" * (budget + 1), encoding="utf-8") problems = self.checker.check_line_budgets(root) self.assertTrue(any("超出预算" in problem for problem in problems), problems) def test_raising_a_byte_budget_without_a_reason_is_reported(self): original = dict(self.checker.BYTE_BUDGET_CHANGES) try: self.checker.BYTE_BUDGET_CHANGES = { "AGENTS.md": [(8 * 1024, "2026-09-03", "初始值"), (12 * 1024, "2026-09-04", "")] } problems = self.checker.check_budget_history() finally: self.checker.BYTE_BUDGET_CHANGES = original self.assertTrue(any("字节预算" in problem for problem in problems), problems) def test_byte_budget_above_ceiling_is_reported(self): original = dict(self.checker.BYTE_BUDGET_CHANGES) try: self.checker.BYTE_BUDGET_CHANGES = { "AGENTS.md": [(64 * 1024, "2026-09-03", "初始值")] } problems = self.checker.check_budget_history() finally: self.checker.BYTE_BUDGET_CHANGES = original self.assertTrue(any("越过天花板" in problem for problem in problems), problems) def test_a_file_within_the_line_budget_can_still_be_over_the_byte_budget(self): """行数合规不代表体积合规：这正是 STATUS.md 155 KB 那次没被拦住的原因。""" with tempfile.TemporaryDirectory() as tmp: root = Path(tmp) budget = self.checker.BYTE_BUDGET["AGENTS.md"] (root / "AGENTS.md").write_text("字" * budget, encoding="utf-8") self.assertEqual( self.checker.check_line_budgets(root), ["CLAUDE.md: 文件缺失"], ) problems = self.checker.check_byte_budgets(root) self.assertTrue( any("AGENTS.md" in problem and "字节" in problem for problem in problems), problems, ) def test_over_long_line_is_reported(self): with tempfile.TemporaryDirectory() as tmp: root = Path(tmp) (root / "AGENTS.md").write_text( "短行\n" + "长" * (self.checker.MAX_LINE_CHARS + 1) + "\n", encoding="utf-8", ) problems = self.checker.check_line_lengths(root) self.assertEqual(len(problems), 1, problems) self.assertIn("第 2 行", problems[0]) def test_skill_frontmatter_and_review_line_are_required(self): with tempfile.TemporaryDirectory() as tmp: root = Path(tmp) skill_dir = root / ".claude" / "skills" / "broken" skill_dir.mkdir(parents=True) (skill_dir / "SKILL.md").write_text("# 没有 frontmatter\n", encoding="utf-8") named = root / ".claude" / "skills" / "mismatch" named.mkdir(parents=True) (named / "SKILL.md").write_text( SKILL_TEMPLATE.format( name="other", description="描述", reviewed="2026-01-01" ), encoding="utf-8", ) problems, reviewed = self.checker.check_skills(root) self.assertTrue(any("frontmatter" in problem for problem in problems), problems) self.assertTrue(any("不一致" in problem for problem in problems), problems) self.assertEqual([rel for rel, _ in reviewed], [".claude/skills/mismatch/SKILL.md"]) def test_long_description_is_reported(self): with tempfile.TemporaryDirectory() as tmp: root = Path(tmp) skill_dir = root / ".claude" / "skills" / "verbose" skill_dir.mkdir(parents=True) (skill_dir / "SKILL.md").write_text( SKILL_TEMPLATE.format( name="verbose", description="描" * (self.checker.DESCRIPTION_MAX_CHARS + 1), reviewed="2026-01-01", ), encoding="utf-8", ) problems, _ = self.checker.check_skills(root) self.assertTrue(any("description" in problem for problem in problems), problems) def test_skill_utf8_bytes_are_checked_within_the_line_limit(self): with tempfile.TemporaryDirectory() as tmp: root = Path(tmp).resolve() skill_dir = root / ".claude" / "skills" / "sized" skill_dir.mkdir(parents=True) header = SKILL_TEMPLATE.format( name="sized", description="描述", reviewed="2026-10-26" ) limit = self.checker.SKILL_MAX_BYTES row = "字" * 100 + "\n" rows = (limit - len(header.encode("utf-8"))) // len(row.encode("utf-8")) within = header + row * rows path = skill_dir / "SKILL.md" path.write_text(within, encoding="utf-8", newline="\n") self.assertLess(len(within.splitlines()), self.checker.SKILL_MAX_LINES) problems, _ = self.checker.check_skills(root) self.assertEqual(problems, []) path.write_text(within + row, encoding="utf-8", newline="\n") problems, _ = self.checker.check_skills(root) self.assertEqual(len(problems), 1, problems) self.assertIn("字节", problems[0]) self.assertEqual(self.checker.check_line_lengths(root), []) def test_skill_line_length_is_checked_independently_of_total_bytes(self): with tempfile.TemporaryDirectory() as tmp: root = Path(tmp).resolve() skill_dir = root / ".claude" / "skills" / "line" skill_dir.mkdir(parents=True) text = SKILL_TEMPLATE.format( name="line", description="描述", reviewed="2026-10-26" ) path = skill_dir / "SKILL.md" path.write_text(text + "字" * self.checker.MAX_LINE_CHARS, encoding="utf-8") self.assertEqual(self.checker.check_line_lengths(root), []) path.write_text(text + "字" * (self.checker.MAX_LINE_CHARS + 1), encoding="utf-8") self.assertLess(path.stat().st_size, self.checker.SKILL_MAX_BYTES) problems = self.checker.check_line_lengths(root) self.assertEqual(len(problems), 1, problems) self.assertIn(".claude/skills/line/SKILL.md", problems[0]) def test_index_and_files_must_match(self): with tempfile.TemporaryDirectory() as tmp: root = Path(tmp) (root / "AGENTS.md").write_text( "| 触发 | `.claude/skills/missing/SKILL.md` |\n", encoding="utf-8" ) skill_dir = root / ".claude" / "skills" / "present" skill_dir.mkdir(parents=True) (skill_dir / "SKILL.md").write_text( SKILL_TEMPLATE.format( name="present", description="描述", reviewed="2026-01-01" ), encoding="utf-8", ) problems = self.checker.check_skill_index(root) self.assertIn("AGENTS.md 索引指向不存在的技能：.claude/skills/missing/SKILL.md", problems) self.assertIn("技能未登记进 AGENTS.md 索引：.claude/skills/present/SKILL.md", problems) def test_stale_review_uses_fixed_today(self): reviewed = [ ("fresh", datetime.date(2026, 8, 1)), ("old", datetime.date(2025, 1, 1)), ] stale = self.checker.stale_reviews(reviewed, datetime.date(2026, 8, 17)) self.assertEqual([rel for rel, _ in stale], ["old"]) if __name__ == "__main__": unittest.main()