claude-code/plugins/hookify/tests/test_config_loader.py at 7af972ea789b1ed31cedb3642266352d7237dbc3 · anthropics/claude-code · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
import os
import sys
import tempfile
import unittest
from pathlib import Path


sys.path.insert(0, str(Path(__file__).resolve().parents[2]))

from hookify.core.config_loader import load_rules, resolve_project_claude_dir


RULE_MARKDOWN = """---
name: project-rule
enabled: true
event: bash
conditions:
  - field: command
    operator: regex_match
    pattern: "rm"
action: warn
---
Warn about rm usage.
"""


class ConfigLoaderTests(unittest.TestCase):
    def test_load_rules_uses_claude_project_dir_when_available(self):
        with tempfile.TemporaryDirectory() as td:
            project_root = Path(td)
            rules_dir = project_root / ".claude"
            rules_dir.mkdir()
            (rules_dir / "hookify.project.local.md").write_text(RULE_MARKDOWN)
            subdir = project_root / "subdir"
            subdir.mkdir()

            original_cwd = Path.cwd()
            original_project_dir = os.environ.get("CLAUDE_PROJECT_DIR")
            try:
                os.environ["CLAUDE_PROJECT_DIR"] = str(project_root)
                os.chdir(subdir)
                self.assertEqual(
                    Path(resolve_project_claude_dir()).resolve(),
                    rules_dir.resolve(),
                )
                self.assertEqual(len(load_rules("bash")), 1)
            finally:
                os.chdir(original_cwd)
                if original_project_dir is None:
                    os.environ.pop("CLAUDE_PROJECT_DIR", None)
                else:
                    os.environ["CLAUDE_PROJECT_DIR"] = original_project_dir

    def test_load_rules_finds_project_root_from_nested_directory(self):
        with tempfile.TemporaryDirectory() as td:
            project_root = Path(td)
            rules_dir = project_root / ".claude"
            rules_dir.mkdir()
            (rules_dir / "hookify.project.local.md").write_text(RULE_MARKDOWN)

            nested_dir = project_root / "packages" / "app"
            nested_dir.mkdir(parents=True)

            original_cwd = Path.cwd()
            original_project_dir = os.environ.get("CLAUDE_PROJECT_DIR")
            try:
                os.environ.pop("CLAUDE_PROJECT_DIR", None)
                os.chdir(nested_dir)
                self.assertEqual(
                    Path(resolve_project_claude_dir()).resolve(),
                    rules_dir.resolve(),
                )
                self.assertEqual(len(load_rules("bash")), 1)
            finally:
                os.chdir(original_cwd)
                if original_project_dir is None:
                    os.environ.pop("CLAUDE_PROJECT_DIR", None)
                else:
                    os.environ["CLAUDE_PROJECT_DIR"] = original_project_dir


if __name__ == "__main__":
    unittest.main()