wehub-resource-sync
67 lines
2.3 KiB
Python
67 lines
2.3 KiB
Python
"""Phase 2 gate — self-wake: timer + on-completion wake records and the tools."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from datetime import datetime, timedelta, timezone
|
|
|
|
from coworker.selfwake import WakeStore, selfwake_tools
|
|
|
|
|
|
def _now():
|
|
return datetime.now(timezone.utc)
|
|
|
|
|
|
def test_timer_due_only_after_fire_time(tmp_path):
|
|
store = WakeStore(tmp_path / "wakes.json")
|
|
soon = store.add_timer("s1", _now() + timedelta(seconds=60))
|
|
past = store.add_timer("s1", _now() - timedelta(seconds=1))
|
|
due_ids = {w.id for w in store.due()}
|
|
assert past.id in due_ids and soon.id not in due_ids
|
|
|
|
|
|
def test_completion_due_only_after_job_completes(tmp_path):
|
|
store = WakeStore(tmp_path / "wakes.json")
|
|
w = store.add_completion("s1", job_id="job-42")
|
|
assert w.id not in {x.id for x in store.due()} # not yet
|
|
marked = store.complete_job("job-42")
|
|
assert [x.id for x in marked] == [w.id]
|
|
assert w.id in {x.id for x in store.due()} # now due
|
|
|
|
|
|
def test_mark_fired_removes_from_due(tmp_path):
|
|
store = WakeStore(tmp_path / "wakes.json")
|
|
w = store.add_timer("s1", _now() - timedelta(seconds=1))
|
|
store.mark_fired(w.id)
|
|
assert w.id not in {x.id for x in store.due()}
|
|
assert w.id not in {x.id for x in store.pending("s1")}
|
|
|
|
|
|
def test_persistence(tmp_path):
|
|
store = WakeStore(tmp_path / "wakes.json")
|
|
w = store.add_completion("s1", "job-1")
|
|
reloaded = WakeStore(tmp_path / "wakes.json")
|
|
assert any(x.id == w.id for x in reloaded.pending("s1"))
|
|
|
|
|
|
def test_event_due_only_after_event_fires(tmp_path):
|
|
store = WakeStore(tmp_path / "wakes.json")
|
|
w = store.add_event("s1", event_key="pr-opened")
|
|
assert w.id not in {x.id for x in store.due()}
|
|
marked = store.fire_event("pr-opened")
|
|
assert [x.id for x in marked] == [w.id]
|
|
assert w.id in {x.id for x in store.due()}
|
|
|
|
|
|
def test_selfwake_tools(tmp_path):
|
|
store = WakeStore(tmp_path / "wakes.json")
|
|
sleep_for, sleep_until, wake_on, wake_on_event = selfwake_tools(store, "s1")
|
|
|
|
assert sleep_for(30)["ok"]
|
|
assert wake_on("job-9")["job_id"] == "job-9"
|
|
assert sleep_until((_now() + timedelta(minutes=5)).isoformat())["fire_at"]
|
|
assert wake_on_event("alert-fired")["event_key"] == "alert-fired"
|
|
|
|
pend = store.pending("s1")
|
|
assert len(pend) == 4
|
|
assert {w.kind for w in pend} == {"timer", "completion", "event"}
|