jpmschweitzerandClaude
78066fab1b
Mechanical only, and separated from the judgment calls that follow so the reviewable changes are not buried in a 98-file whitespace diff. 227 automatic fixes: 60 blank lines carrying whitespace, 60 unsorted import blocks, 34 Optional[X] to X | None, 28 unused imports, 16 deprecated typing imports, 12 datetime.timezone.utc to datetime.UTC, and assorted smaller modernisations. Then `ruff format` over src and tests: 98 files reformatted, 35 already conforming. No file among the unused-import findings defines __all__ or is an __init__.py, so nothing here removes a re-export. `make test`: 658 passed, unchanged from HEAD. Two things observed while verifying, neither addressed here: `pytest tests/` cannot collect — tests/e2e/test_orchestration_e2e.py uses an `e2e` marker that is not registered, and the config is strict about markers. This fails identically at HEAD, so it predates this change; `make test` passes because it ignores tests/e2e, tests/integration and tests/contracts. test_tatlock_tool_call_logging_calculator is flaky. It failed once in a full run with these changes and passed on the next, passes in isolation with them, and fails in isolation at HEAD. It is order- or timing-dependent, not a regression from this commit — established by running the full suite both ways rather than by reasoning about which change could have caused it. Co-Authored-By: Claude <noreply@anthropic.com>
140 lines
4.8 KiB
Python
140 lines
4.8 KiB
Python
"""
|
|
Tests for Housekeeper capability registration.
|
|
"""
|
|
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
import pytest
|
|
|
|
from src.agents.housekeeper.capability import (
|
|
HOUSEKEEPER_CAPABILITY,
|
|
get_housekeeper_capability,
|
|
register_housekeeper,
|
|
unregister_housekeeper,
|
|
)
|
|
from src.core.household_registry import HouseholdCapability
|
|
|
|
|
|
@pytest.mark.unit
|
|
class TestHousekeeperCapability:
|
|
"""Tests for the Housekeeper capability definition."""
|
|
|
|
def test_capability_is_household_capability(self):
|
|
"""Test capability is correct type."""
|
|
assert isinstance(HOUSEKEEPER_CAPABILITY, HouseholdCapability)
|
|
|
|
def test_capability_name(self):
|
|
"""Test capability has correct name."""
|
|
assert HOUSEKEEPER_CAPABILITY.name == "housekeeper"
|
|
|
|
def test_capability_role(self):
|
|
"""Test capability has correct role."""
|
|
assert HOUSEKEEPER_CAPABILITY.role == "The Housekeeper"
|
|
|
|
def test_capability_category(self):
|
|
"""Test capability is in automation category."""
|
|
assert HOUSEKEEPER_CAPABILITY.category == "automation"
|
|
|
|
def test_capability_domains(self):
|
|
"""Test capability covers expected domains."""
|
|
domains = HOUSEKEEPER_CAPABILITY.domains
|
|
|
|
assert "lights" in domains
|
|
assert "switches" in domains
|
|
assert "automation" in domains
|
|
assert "home" in domains
|
|
assert "scene" in domains
|
|
assert "turn on" in domains
|
|
assert "turn off" in domains
|
|
|
|
def test_capability_requires_network(self):
|
|
"""Test capability requires network access."""
|
|
assert HOUSEKEEPER_CAPABILITY.requires_network is True
|
|
|
|
def test_capability_cost_is_low(self):
|
|
"""Test capability is low cost (local API calls)."""
|
|
assert HOUSEKEEPER_CAPABILITY.cost == "low"
|
|
|
|
def test_get_housekeeper_capability(self):
|
|
"""Test getter returns same capability."""
|
|
cap = get_housekeeper_capability()
|
|
|
|
assert cap is HOUSEKEEPER_CAPABILITY
|
|
|
|
|
|
@pytest.mark.unit
|
|
class TestHousekeeperRegistration:
|
|
"""Tests for Housekeeper registration functions."""
|
|
|
|
def test_register_housekeeper(self):
|
|
"""Test registering housekeeper with registry."""
|
|
mock_registry = MagicMock()
|
|
mock_registry.__contains__ = MagicMock(return_value=False)
|
|
|
|
with patch(
|
|
"src.agents.housekeeper.capability.get_household_registry",
|
|
return_value=mock_registry,
|
|
):
|
|
with patch("src.agents.housekeeper.capability.get_housekeeper_agent") as mock_get_agent:
|
|
mock_agent = MagicMock()
|
|
mock_get_agent.return_value = mock_agent
|
|
|
|
register_housekeeper()
|
|
|
|
mock_registry.register.assert_called_once()
|
|
call_kwargs = mock_registry.register.call_args[1]
|
|
|
|
assert call_kwargs["name"] == "housekeeper"
|
|
assert call_kwargs["capability"] is HOUSEKEEPER_CAPABILITY
|
|
assert call_kwargs["agent"] is mock_agent
|
|
|
|
def test_register_housekeeper_already_registered(self):
|
|
"""Test registering when already registered does nothing."""
|
|
mock_registry = MagicMock()
|
|
mock_registry.__contains__ = MagicMock(return_value=True)
|
|
|
|
with patch(
|
|
"src.agents.housekeeper.capability.get_household_registry",
|
|
return_value=mock_registry,
|
|
):
|
|
register_housekeeper()
|
|
|
|
# Should not call register since already registered
|
|
mock_registry.register.assert_not_called()
|
|
|
|
def test_unregister_housekeeper(self):
|
|
"""Test unregistering housekeeper from registry."""
|
|
mock_registry = MagicMock()
|
|
|
|
with patch(
|
|
"src.agents.housekeeper.capability.get_household_registry",
|
|
return_value=mock_registry,
|
|
):
|
|
unregister_housekeeper()
|
|
|
|
mock_registry.unregister.assert_called_once_with("housekeeper")
|
|
|
|
|
|
@pytest.mark.unit
|
|
class TestCapabilityDescription:
|
|
"""Tests for capability description."""
|
|
|
|
def test_description_mentions_device_control(self):
|
|
"""Test description mentions device control capabilities."""
|
|
desc = HOUSEKEEPER_CAPABILITY.description.lower()
|
|
assert "turn on" in desc
|
|
# Description uses "ON/OFF" format
|
|
assert "off" in desc
|
|
|
|
def test_description_mentions_scenes(self):
|
|
"""Test description mentions scene capability."""
|
|
assert "scene" in HOUSEKEEPER_CAPABILITY.description.lower()
|
|
|
|
def test_description_mentions_scripts(self):
|
|
"""Test description mentions script capability."""
|
|
assert "script" in HOUSEKEEPER_CAPABILITY.description.lower()
|
|
|
|
def test_description_mentions_automations(self):
|
|
"""Test description mentions automation management."""
|
|
assert "automation" in HOUSEKEEPER_CAPABILITY.description.lower()
|