| import pytest |
|
|
| from aegis.core.tool_registry import ( |
| DuplicateToolError, |
| ToolNotFoundError, |
| ToolRegistry, |
| ) |
|
|
|
|
| class StubTool: |
| def __init__(self, name: str) -> None: |
| self.name = name |
|
|
|
|
| def test_registry_registers_and_resolves_tool() -> None: |
| tool = StubTool("calculator") |
| registry = ToolRegistry() |
|
|
| registry.register(tool) |
|
|
| assert registry.get("calculator") is tool |
| assert registry.contains("calculator") is True |
| assert len(registry) == 1 |
|
|
|
|
| def test_registry_accepts_initial_tools() -> None: |
| calculator = StubTool("calculator") |
| search = StubTool("search") |
|
|
| registry = ToolRegistry([calculator, search]) |
|
|
| assert registry.get("calculator") is calculator |
| assert registry.get("search") is search |
| assert len(registry) == 2 |
|
|
|
|
| def test_registry_rejects_duplicate_tool_names() -> None: |
| registry = ToolRegistry([StubTool("calculator")]) |
|
|
| with pytest.raises(DuplicateToolError): |
| registry.register(StubTool("calculator")) |
|
|
|
|
| def test_registry_raises_when_tool_is_missing() -> None: |
| registry = ToolRegistry() |
|
|
| with pytest.raises(ToolNotFoundError): |
| registry.get("missing") |
|
|
|
|
| def test_registry_returns_sorted_names() -> None: |
| registry = ToolRegistry( |
| [ |
| StubTool("search"), |
| StubTool("calculator"), |
| ] |
| ) |
|
|
| assert registry.names() == ("calculator", "search") |
|
|
|
|
| def test_registry_normalizes_tool_names() -> None: |
| tool = StubTool(" calculator ") |
| registry = ToolRegistry([tool]) |
|
|
| assert registry.contains("calculator") is True |
| assert registry.get(" calculator ") is tool |
|
|
|
|
| def test_registry_rejects_empty_tool_name() -> None: |
| registry = ToolRegistry() |
|
|
| with pytest.raises(ValueError, match="Tool name cannot be empty"): |
| registry.register(StubTool(" ")) |
|
|
