- metrics: update_ingest_queue_depth guarded on `not by_queue`, which conflated
None (memory backend no-op) with {} (postgres, ALL queues drained). When every
queue drains at once, get_ingest_job_counts_by_queue returns {} and the
pre-zero loop was skipped, leaving a stale ghost backlog in the gauge. Guard on
`by_queue is None` only; add an all-drained regression test.
- procrastinate: note that INGEST_TRANSIENT_MAX_ATTEMPTS is snapshotted at
blueprint-build time (restart to pick up changes).
Deck #323.
Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
56 lines
2.0 KiB
Python
56 lines
2.0 KiB
Python
"""Unit test for the per-tier ingest-queue-depth gauge (Deck #323).
|
|
|
|
Guards the round-2 fix: a queue that drains to empty (and so drops out of
|
|
procrastinate's ``list_queues_async``) must read 0, not its last non-zero value.
|
|
"""
|
|
|
|
import pytest
|
|
from pytest import approx
|
|
|
|
from nextcloud_mcp_server.observability.metrics import update_ingest_queue_depth
|
|
|
|
pytestmark = pytest.mark.unit
|
|
|
|
_METRIC = "astrolabe_ingest_queue_depth"
|
|
|
|
|
|
def test_drained_queue_zeroes_not_stale(metric_sample):
|
|
# ocr has a backlog this tick. (pytest.approx: the gauge sample is a float.)
|
|
update_ingest_queue_depth({"ingest-ocr": {"todo": 4}})
|
|
assert metric_sample(_METRIC, {"queue": "ingest-ocr", "status": "todo"}) == approx(
|
|
4
|
|
)
|
|
|
|
# Next tick ocr has drained → procrastinate omits it from by_queue entirely.
|
|
update_ingest_queue_depth({"ingest-fast": {"todo": 1}})
|
|
# The gauge must read 0 for the drained queue, not the stale 4.
|
|
assert metric_sample(_METRIC, {"queue": "ingest-ocr", "status": "todo"}) == approx(
|
|
0
|
|
)
|
|
assert metric_sample(_METRIC, {"queue": "ingest-fast", "status": "todo"}) == approx(
|
|
1
|
|
)
|
|
|
|
|
|
def test_none_is_noop(metric_sample):
|
|
update_ingest_queue_depth({"ingest-fast": {"doing": 2}})
|
|
# Memory backend passes None → must not wipe the last published values.
|
|
update_ingest_queue_depth(None)
|
|
assert metric_sample(
|
|
_METRIC, {"queue": "ingest-fast", "status": "doing"}
|
|
) == approx(2)
|
|
|
|
|
|
def test_all_queues_drained_empty_dict_zeroes(metric_sample):
|
|
# postgres backend with every queue drained → get_ingest_job_counts_by_queue
|
|
# returns {} (list_queues_async drops empty queues). An empty dict is NOT the
|
|
# memory-backend no-op: it must still zero every managed queue's gauge.
|
|
update_ingest_queue_depth({"ingest-fast": {"todo": 9}})
|
|
assert metric_sample(_METRIC, {"queue": "ingest-fast", "status": "todo"}) == approx(
|
|
9
|
|
)
|
|
update_ingest_queue_depth({})
|
|
assert metric_sample(_METRIC, {"queue": "ingest-fast", "status": "todo"}) == approx(
|
|
0
|
|
)
|