test_registry.py

"""Tests for the Prometheus metrics registry."""

from __future__ import annotations

from collections.abc import Iterator

import pytest
from prometheus_client import CollectorRegistry

from lean_spec.subspecs.metrics.registry import (
    ATTESTATION_AGGREGATE_COVERAGE_DIFF_DIRECTIONS,
    ATTESTATION_AGGREGATE_COVERAGE_SECTIONS,
    BLOCK_PROPOSAL_ATTESTATION_BUILD_PHASES,
    registry,
)


@pytest.fixture(autouse=True)
def _reset_registry() -> Iterator[None]:
    """Ensure metrics are uninitialized before and after each test."""
    registry.reset()
    registry._initialized = False
    yield
    registry.reset()
    registry._initialized = False


def test_attestation_aggregate_coverage_metrics_registered() -> None:
    """Coverage gauges are created with default combined-subnet series."""
    test_reg = CollectorRegistry()
    registry.init(registry=test_reg)

    for section in ATTESTATION_AGGREGATE_COVERAGE_SECTIONS:
        assert (
            test_reg.get_sample_value(
                "lean_attestation_aggregate_coverage_validators",
                {"section": section, "subnet": "combined"},
            )
            == 0.0
        )
        assert (
            test_reg.get_sample_value(
                "lean_attestation_aggregate_coverage_subnets",
                {"section": section},
            )
            == 0.0
        )

    for direction in ATTESTATION_AGGREGATE_COVERAGE_DIFF_DIRECTIONS:
        assert (
            test_reg.get_sample_value(
                "lean_attestation_aggregate_coverage_diff_validators",
                {"direction": direction},
            )
            == 0.0
        )


def test_block_proposal_attestation_build_metrics_registered() -> None:
    """Block-proposal attestation selection metrics are registered on init."""
    test_reg = CollectorRegistry()
    registry.init(registry=test_reg)

    for phase in BLOCK_PROPOSAL_ATTESTATION_BUILD_PHASES:
        assert (
            test_reg.get_sample_value(
                "lean_block_proposal_attestation_build_phase_seconds_count",
                {"phase": phase},
            )
            is None
        )

    assert test_reg.get_sample_value("lean_block_proposal_attestation_builds_total") == 0.0
    assert test_reg.get_sample_value("lean_block_proposal_child_payloads_consumed_total") == 0.0