Reset prometheus client global registry in unit tests

tests/test_bundler.py

@@ -1,6 +1,18 @@
 # test_bundler.py
 """Unit tests for lta/bundler.py."""
 
+# -----------------------------------------------------------------------------
+# reset prometheus registry for unit tests
+from prometheus_client import REGISTRY
+collectors = list(REGISTRY._collector_to_names.keys())
+for collector in collectors:
+ REGISTRY.unregister(collector)
+from prometheus_client import gc_collector, platform_collector, process_collector
+process_collector.ProcessCollector()
+platform_collector.PlatformCollector()
+gc_collector.GCCollector()
+# -----------------------------------------------------------------------------
+
 import os
 from typing import Dict
 from unittest.mock import AsyncMock, call, mock_open, patch

tests/test_deleter.py

@@ -1,6 +1,18 @@
 # test_deleter.py
 """Unit tests for lta/deleter.py."""
 
+# -----------------------------------------------------------------------------
+# reset prometheus registry for unit tests
+from prometheus_client import REGISTRY
+collectors = list(REGISTRY._collector_to_names.keys())
+for collector in collectors:
+ REGISTRY.unregister(collector)
+from prometheus_client import gc_collector, platform_collector, process_collector
+process_collector.ProcessCollector()
+platform_collector.PlatformCollector()
+gc_collector.GCCollector()
+# -----------------------------------------------------------------------------
+
 from typing import Dict
 from unittest.mock import AsyncMock, call, MagicMock
 

tests/test_desy_move_verifier.py

@@ -1,6 +1,18 @@
 # test_desy_move_verifier.py
 """Unit tests for lta/desy_move_verifier.py."""
 
+# -----------------------------------------------------------------------------
+# reset prometheus registry for unit tests
+from prometheus_client import REGISTRY
+collectors = list(REGISTRY._collector_to_names.keys())
+for collector in collectors:
+ REGISTRY.unregister(collector)
+from prometheus_client import gc_collector, platform_collector, process_collector
+process_collector.ProcessCollector()
+platform_collector.PlatformCollector()
+gc_collector.GCCollector()
+# -----------------------------------------------------------------------------
+
 from typing import Dict
 from unittest.mock import AsyncMock, call
 

tests/test_desy_verifier.py

@@ -1,6 +1,18 @@
 # test_desy_verifier.py
 """Unit tests for lta/desy_verifier.py."""
 
+# -----------------------------------------------------------------------------
+# reset prometheus registry for unit tests
+from prometheus_client import REGISTRY
+collectors = list(REGISTRY._collector_to_names.keys())
+for collector in collectors:
+ REGISTRY.unregister(collector)
+from prometheus_client import gc_collector, platform_collector, process_collector
+process_collector.ProcessCollector()
+platform_collector.PlatformCollector()
+gc_collector.GCCollector()
+# -----------------------------------------------------------------------------
+
 from typing import Dict
 from unittest.mock import AsyncMock, call
 from uuid import uuid1

tests/test_locator.py

@@ -1,6 +1,18 @@
 # test_locator.py
 """Unit tests for lta/locator.py."""
 
+# -----------------------------------------------------------------------------
+# reset prometheus registry for unit tests
+from prometheus_client import REGISTRY
+collectors = list(REGISTRY._collector_to_names.keys())
+for collector in collectors:
+ REGISTRY.unregister(collector)
+from prometheus_client import gc_collector, platform_collector, process_collector
+process_collector.ProcessCollector()
+platform_collector.PlatformCollector()
+gc_collector.GCCollector()
+# -----------------------------------------------------------------------------
+
 from math import floor
 from secrets import token_hex
 from typing import Any, Dict, List, Union

tests/test_nersc_mover.py

@@ -1,6 +1,18 @@
 # test_nersc_mover.py
 """Unit tests for lta/nersc_mover.py."""
 
+# -----------------------------------------------------------------------------
+# reset prometheus registry for unit tests
+from prometheus_client import REGISTRY
+collectors = list(REGISTRY._collector_to_names.keys())
+for collector in collectors:
+ REGISTRY.unregister(collector)
+from prometheus_client import gc_collector, platform_collector, process_collector
+process_collector.ProcessCollector()
+platform_collector.PlatformCollector()
+gc_collector.GCCollector()
+# -----------------------------------------------------------------------------
+
 from typing import Dict
 from unittest.mock import AsyncMock, call, MagicMock
 

tests/test_nersc_retriever.py

@@ -1,6 +1,18 @@
 # test_nersc_retriever.py
 """Unit tests for lta/nersc_retriever.py."""
 
+# -----------------------------------------------------------------------------
+# reset prometheus registry for unit tests
+from prometheus_client import REGISTRY
+collectors = list(REGISTRY._collector_to_names.keys())
+for collector in collectors:
+ REGISTRY.unregister(collector)
+from prometheus_client import gc_collector, platform_collector, process_collector
+process_collector.ProcessCollector()
+platform_collector.PlatformCollector()
+gc_collector.GCCollector()
+# -----------------------------------------------------------------------------
+
 from typing import Dict
 from unittest.mock import AsyncMock, call, MagicMock
 

tests/test_nersc_verifier.py

@@ -1,6 +1,18 @@
 # test_nersc_verifier.py
 """Unit tests for lta/nersc_verifier.py."""
 
+# -----------------------------------------------------------------------------
+# reset prometheus registry for unit tests
+from prometheus_client import REGISTRY
+collectors = list(REGISTRY._collector_to_names.keys())
+for collector in collectors:
+ REGISTRY.unregister(collector)
+from prometheus_client import gc_collector, platform_collector, process_collector
+process_collector.ProcessCollector()
+platform_collector.PlatformCollector()
+gc_collector.GCCollector()
+# -----------------------------------------------------------------------------
+
 from typing import Dict
 from unittest.mock import AsyncMock, call, MagicMock
 from uuid import uuid1

tests/test_picker.py

@@ -1,6 +1,18 @@
 # test_picker.py
 """Unit tests for lta/picker.py."""
 
+# -----------------------------------------------------------------------------
+# reset prometheus registry for unit tests
+from prometheus_client import REGISTRY
+collectors = list(REGISTRY._collector_to_names.keys())
+for collector in collectors:
+ REGISTRY.unregister(collector)
+from prometheus_client import gc_collector, platform_collector, process_collector
+process_collector.ProcessCollector()
+platform_collector.PlatformCollector()
+gc_collector.GCCollector()
+# -----------------------------------------------------------------------------
+
 from secrets import token_hex
 from typing import Any, Dict, List, Union
 from unittest.mock import AsyncMock, call, MagicMock

tests/test_rate_limiter.py

@@ -1,6 +1,18 @@
 # test_rate_limiter.py
 """Unit tests for lta/rate_limiter.py."""
 
+# -----------------------------------------------------------------------------
+# reset prometheus registry for unit tests
+from prometheus_client import REGISTRY
+collectors = list(REGISTRY._collector_to_names.keys())
+for collector in collectors:
+ REGISTRY.unregister(collector)
+from prometheus_client import gc_collector, platform_collector, process_collector
+process_collector.ProcessCollector()
+platform_collector.PlatformCollector()
+gc_collector.GCCollector()
+# -----------------------------------------------------------------------------
+
 from typing import Dict
 from unittest.mock import AsyncMock, call, MagicMock
 

tests/test_site_move_verifier.py

@@ -1,6 +1,18 @@
 # test_site_move_verifier.py
 """Unit tests for lta/site_move_verifier.py."""
 
+# -----------------------------------------------------------------------------
+# reset prometheus registry for unit tests
+from prometheus_client import REGISTRY
+collectors = list(REGISTRY._collector_to_names.keys())
+for collector in collectors:
+ REGISTRY.unregister(collector)
+from prometheus_client import gc_collector, platform_collector, process_collector
+process_collector.ProcessCollector()
+platform_collector.PlatformCollector()
+gc_collector.GCCollector()
+# -----------------------------------------------------------------------------
+
 from typing import Dict
 from unittest.mock import AsyncMock, call, MagicMock
 

tests/test_transfer_request_finisher.py

@@ -1,6 +1,18 @@
 # test_transfer_request_finisher.py
 """Unit tests for lta/transfer_request_finisher.py."""
 
+# -----------------------------------------------------------------------------
+# reset prometheus registry for unit tests
+from prometheus_client import REGISTRY
+collectors = list(REGISTRY._collector_to_names.keys())
+for collector in collectors:
+ REGISTRY.unregister(collector)
+from prometheus_client import gc_collector, platform_collector, process_collector
+process_collector.ProcessCollector()
+platform_collector.PlatformCollector()
+gc_collector.GCCollector()
+# -----------------------------------------------------------------------------
+
 from typing import Dict
 from unittest.mock import AsyncMock, call
 

tests/test_unpacker.py

@@ -1,6 +1,18 @@
 # test_unpacker.py
 """Unit tests for lta/unpacker.py."""
 
+# -----------------------------------------------------------------------------
+# reset prometheus registry for unit tests
+from prometheus_client import REGISTRY
+collectors = list(REGISTRY._collector_to_names.keys())
+for collector in collectors:
+ REGISTRY.unregister(collector)
+from prometheus_client import gc_collector, platform_collector, process_collector
+process_collector.ProcessCollector()
+platform_collector.PlatformCollector()
+gc_collector.GCCollector()
+# -----------------------------------------------------------------------------
+
 from typing import Any, Dict
 from unittest.mock import AsyncMock, call, MagicMock, mock_open, patch