18 lines
508 B
Python
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
|