ariadne/tests/test_metrics.py

18 lines
508 B
Python

from __future__ import annotations
from prometheus_client import generate_latest
from ariadne.metrics.metrics import record_task_run, record_schedule_state
def test_metrics_include_task_run() -> None:
record_task_run("unit", "ok", 0.2)
payload = generate_latest()
assert b"ariadne_task_runs_total" in payload
def test_metrics_include_schedule() -> None:
record_schedule_state("sched", 1, 1, 2, True)
payload = generate_latest()
assert b"ariadne_schedule_last_status" in payload