didericis-claude
57290da1e8
Three new unit test modules: - tests/unit/test_contrib_gitea_client.py — GiteaClient (urllib mocked) and GiteaForge delegation - tests/unit/orchestrator/test_main.py — __main__ run/status commands - tests/unit/orchestrator/test_bootstrap.py — _token, BotBottleStateStore, _to_forge_state/_to_record, make_forge, make_sidecar, build Augments to existing suites: - test_events: non-"created" comment action ignored - test_lifecycle: _iso_now callable, untracked-issue comment ignored, untracked-PR closed ignored (covers _find_by_pr return-None path) - test_runner: destroy command, _default_run via subprocess mock - test_sidecar: _jsonable dataclass/list branches, OpLog.read on missing file, drain_done_events on corrupted file, socket _Handler invalid-JSON and empty-line paths, serve() with pre-existing socket path - test_watchdog: _loop body covered by patching _TICK_SECS to 0.01s - test_webhook: unknown GET path returns 404 Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
79 lines
2.8 KiB
Python
79 lines
2.8 KiB
Python
"""Unit: watchdog sweep."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import time
|
|
import unittest
|
|
import unittest.mock
|
|
from datetime import datetime, timedelta
|
|
|
|
from bot_bottle.orchestrator.model import STATUS_FROZEN, STATUS_RUNNING, RunRecord
|
|
from bot_bottle.orchestrator.store import InMemoryStateStore
|
|
from bot_bottle.orchestrator.watchdog import Watchdog
|
|
|
|
from ._fakes import FakeRunner
|
|
|
|
_NOW = datetime(2026, 7, 1, 12, 0, 0).astimezone()
|
|
|
|
|
|
def _record(issue: int, status: str, checkin: str) -> RunRecord:
|
|
return RunRecord(
|
|
owner="o", repo="r", issue_number=issue, slug=f"s{issue}",
|
|
agent_name="a", status=status, last_checkin_at=checkin,
|
|
)
|
|
|
|
|
|
class WatchdogSweepTest(unittest.TestCase):
|
|
def setUp(self):
|
|
self.store = InMemoryStateStore()
|
|
self.runner = FakeRunner()
|
|
self.wd = Watchdog(store=self.store, runner=self.runner, timeout_secs=1800)
|
|
|
|
def _status(self, issue: int) -> str:
|
|
rec = self.store.get("o", "r", issue)
|
|
assert rec is not None
|
|
return rec.status
|
|
|
|
def test_stale_running_is_frozen(self):
|
|
stale = (_NOW - timedelta(minutes=31)).isoformat()
|
|
self.store.upsert(_record(1, STATUS_RUNNING, stale))
|
|
fired = self.wd.sweep(_NOW)
|
|
self.assertEqual([1], [r.issue_number for r in fired])
|
|
self.assertEqual(STATUS_FROZEN, self._status(1))
|
|
self.assertIn(("freeze", "s1"), self.runner.calls)
|
|
|
|
def test_fresh_running_untouched(self):
|
|
fresh = (_NOW - timedelta(minutes=5)).isoformat()
|
|
self.store.upsert(_record(2, STATUS_RUNNING, fresh))
|
|
self.assertEqual([], self.wd.sweep(_NOW))
|
|
self.assertEqual(STATUS_RUNNING, self._status(2))
|
|
|
|
def test_non_running_ignored(self):
|
|
stale = (_NOW - timedelta(hours=2)).isoformat()
|
|
self.store.upsert(_record(3, STATUS_FROZEN, stale))
|
|
self.assertEqual([], self.wd.sweep(_NOW))
|
|
|
|
def test_unparseable_checkin_skipped(self):
|
|
self.store.upsert(_record(4, STATUS_RUNNING, "not-a-time"))
|
|
self.assertEqual([], self.wd.sweep(_NOW))
|
|
|
|
def test_start_and_stop(self):
|
|
# Exercises the daemon-thread start/stop path; stop sets the event
|
|
# so the loop's wait returns immediately.
|
|
self.wd.start()
|
|
self.wd.stop()
|
|
|
|
def test_loop_sweeps_stale_record(self):
|
|
# Patch tick to near-zero so the loop iterates quickly.
|
|
stale = (_NOW - timedelta(hours=1)).isoformat()
|
|
self.store.upsert(_record(5, STATUS_RUNNING, stale))
|
|
with unittest.mock.patch("bot_bottle.orchestrator.watchdog._TICK_SECS", 0.01):
|
|
self.wd.start()
|
|
time.sleep(0.05) # enough for several iterations at 0.01s tick
|
|
self.wd.stop()
|
|
self.assertEqual(STATUS_FROZEN, self.store.get("o", "r", 5).status)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|