From 33077e8d57d12f3732c3a7873de4e0f098a797ff Mon Sep 17 00:00:00 2001 From: Kim Gustyr Date: Thu, 27 Mar 2025 14:36:21 +0000 Subject: [PATCH] reset registry state before asserting --- src/common/test_tools/plugin.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/src/common/test_tools/plugin.py b/src/common/test_tools/plugin.py index b0914b5..8597e7b 100644 --- a/src/common/test_tools/plugin.py +++ b/src/common/test_tools/plugin.py @@ -11,6 +11,11 @@ def assert_metric_impl() -> Generator[AssertMetricFixture, None, None]: registry = prometheus_client.REGISTRY collectors = [*registry._collector_to_names] + # Reset registry state + for collector in collectors: + if isinstance(collector, MetricWrapperBase): + collector.clear() + def _assert_metric( *, name: str, @@ -25,10 +30,5 @@ def _assert_metric( yield _assert_metric - # Reset registry state - for collector in collectors: - if isinstance(collector, MetricWrapperBase): - collector.clear() - assert_metric = pytest.fixture(assert_metric_impl)