Skip to content

Commit bb9c295

Browse files
author
Your Name
committed
Fix hooks tests
1 parent 312813a commit bb9c295

2 files changed

Lines changed: 8 additions & 2 deletions

File tree

cecli/hooks/registry.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -173,6 +173,12 @@ def _load_hooks_from_python_file(self, file_path: Path) -> List[str]:
173173
hook = obj()
174174
self.hook_manager.register_hook(hook)
175175
hooks.append(hook.name)
176+
except ValueError as e:
177+
# Hook might already be registered; still count as loaded
178+
if "already exists" in str(e):
179+
hooks.append(hook.name)
180+
else:
181+
print(f"Warning: Could not register hook '{name}': {e}")
176182
except Exception as e:
177183
print(f"Warning: Could not instantiate hook {name}: {e}")
178184

tests/subagents/test_service.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -42,8 +42,8 @@ def service(mock_coder):
4242
def registry():
4343
"""Pre-populated registry."""
4444
AgentService._global_registry = {
45-
"reviewer": MagicMock(name="reviewer", prompt="Review code.", model=None),
46-
"tester": MagicMock(name="tester", prompt="Write tests.", model="gpt-4"),
45+
"reviewer": MagicMock(name="reviewer", prompt="Review code.", model=None, hooks={}),
46+
"tester": MagicMock(name="tester", prompt="Write tests.", model="gpt-4", hooks={}),
4747
}
4848
yield
4949
AgentService._global_registry = {}

0 commit comments

Comments
 (0)