"""Construction-time state.db maintenance must keep the startup watchdog's progress lease alive (#111092).

Archive, prune, orphan sweep and VACUUM are I/O-bound with near-zero CPU, so the watchdog's
CPU fallback cannot tell them from a parked deadlock; only a phase lease can. Leases are
clamped to ``_MAX_LEASE_S`` per call, so each long step renews rather than one lease at entry.
"""

from __future__ import annotations

import time

import hermes_startup_watchdog as sw
from hermes_state import SessionDB


def test_maintenance_steps_renew_the_armed_watchdog_lease(tmp_path, monkeypatch):
    monkeypatch.setenv("HERMES_HOME", str(tmp_path))
    sw._reset_for_tests()
    handle = sw.arm_startup_watchdog(timeout_s=300.0)
    db = SessionDB(db_path=tmp_path / "state.db")
    try:
        # Maintenance is throttled by state_meta; a fresh DB has never run it.
        monkeypatch.setattr(db, "prune_sessions", lambda **kw: 120)  # VACUUM only runs after a real prune
        monkeypatch.setattr(db, "sweep_orphaned_sessions", lambda **kw: [])
        monkeypatch.setattr(db, "_freelist_ratio", lambda: 1.0)
        monkeypatch.setattr(db, "vacuum", lambda: None)
        monkeypatch.setattr(db, "archive_stale_sessions", lambda *a, **kw: 0)
        before = handle._lease_count

        db.maybe_auto_archive(idle_days=3, min_interval_hours=0)
        db.maybe_auto_prune_and_vacuum(
            retention_days=90, min_interval_hours=0, vacuum=True, min_vacuum_interval_days=0,
        )

        # One renewal per long step (archive, prune, sweep, vacuum), the last one
        # taken right before VACUUM so a multi-minute rewrite never outlives the clamp.
        assert handle._lease_count - before == 4
        assert handle._lease_phase == "state_db_auto_vacuum"
        assert handle._lease_until > time.monotonic()
    finally:
        db.close()
        sw._reset_for_tests()