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