test_main.py 642 B

123456789101112131415161718192021222324
  1. import pytest
  2. import app.main as main_module
  3. from app.schemas import ResetRequest
  4. @pytest.mark.asyncio
  5. async def test_global_reset_calls_each_adapter(monkeypatch):
  6. class FakeAgent:
  7. def __init__(self) -> None:
  8. self.reset_count = 0
  9. async def reset(self) -> None:
  10. self.reset_count += 1
  11. first = FakeAgent()
  12. second = FakeAgent()
  13. monkeypatch.setattr(main_module, "agents", {"first": first, "second": second})
  14. result = await main_module.reset(ResetRequest())
  15. assert result == {"status": "ok", "system": None}
  16. assert first.reset_count == 1
  17. assert second.reset_count == 1