Finished feature NGM-6-availability-proc.

test/test_correlator.py

+import tempfile
+
+from mapping_provider.backends import cache, correlator
+
+
+def test_handle_correlator_state_broadcast():
+    """
+    tmp bogus test - just to have something
+    """
+    with tempfile.TemporaryDirectory() as tmp_dir:
+        cache.init(tmp_dir)
+        correlator.handle_correlator_state_broadcast(
+            message={'alarms': [], 'endpoints': []})
+        
+        cached_data = cache.get(correlator.CACHED_CORRELATOR_STATE_FILENAME)
+        assert cached_data == {'alarms': [], 'endpoints': []}

test/test_inventory.py

+import tempfile
+
+import responses
+
+from mapping_provider.backends import cache, inventory
+
+from .common import load_test_data
+
+
+@responses.activate
+def test_inventory_service_download():
+    """
+    tmp bogus test - just to have something
+    """
+
+    inventory_base_uri = 'https://dummy-hostname.dummy.domain'
+    reporting_base_uri = 'https://another-dummy-hostname.dummy.domain'
+
+    responses.add(
+        method=responses.GET,
+        url=f'{inventory_base_uri}/poller/interfaces',
+        json=load_test_data('poller-interfaces.json')
+    )
+    responses.add(
+        method=responses.GET,
+        url=f'{reporting_base_uri}/scid/current',
+        json=load_test_data('scid-current.json')
+    )
+    responses.add(
+        method=responses.GET,
+        url=f'{inventory_base_uri}/map/services',
+        json=load_test_data('inprov-services.json')
+    )
+    
+    with tempfile.TemporaryDirectory() as tmp_dir:
+
+        cache.init(tmp_dir)
+        inventory._load_all_inventory(
+            inventory_base_uri='https://dummy-hostname.dummy.domain',
+            reporting_base_uri='https://another-dummy-hostname.dummy.domain')
+
+        # assert os.path.exists(os.path.join(tmp_dir, services.POLLER_INTERFACES_CACHE_FILENAME))
+
+        cached_data = cache.get(inventory.INPROV_POLLER_INTERFACES_CACHE_FILENAME)
+        assert cached_data == load_test_data('poller-interfaces.json')
+
+        cached_data = cache.get(inventory.INPROV_MAP_SERVICES_CACHE_FILENAME)
+        assert cached_data == load_test_data('inprov-services.json')
+
+        cached_data = cache.get(inventory.REPORTING_SCID_CURRENT_CACHE_FILENAME)
+        assert cached_data == load_test_data('scid-current.json')

test/test_map_endpoints.py

-import json
-import os
 import re
 
 import responses
 
-from mapping_provider.api.map import RouterList, ServiceList, SiteList
+from mapping_provider.api.map import RouterList, SiteList
+from mapping_provider.backends.services import ServiceList
 
-DATA_DIR = os.path.join(os.path.dirname(__file__), 'data')
+from .common import load_test_data
 
 
-def _load_test_data(filename: str) -> dict:
-    with open(os.path.join(DATA_DIR, filename)) as f:
-        return json.load(f)
-
-# @responses.activate
-# def test_inventory_uri_validation():
-#     responses.add(
-#         method=responses.GET, url=re.compile(r".*/version$"), json={"api": "0.9"}
-#     )
-#     assert (
-#         classifier.verify_inventory_provider_uri(None, None, "http://a.b.c:9999")
-#         == "http://a.b.c:9999/"
-#     )
-
 @responses.activate
 def test_get_sites(client):
 
     responses.add(
         method=responses.GET,
         url=re.compile(r'.*/map/sites$'),
-        json=_load_test_data('inprov-sites.json')
+        json=load_test_data('inprov-sites.json')
     )
 
     rv = client.get("/map/sites")
@@ -44,7 +29,7 @@ def test_get_routers(client):
     responses.add(
         method=responses.GET,
         url=re.compile(r'.*/map/routers$'),
-        json=_load_test_data('inprov-routers.json')
+        json=load_test_data('inprov-routers.json')
     )
 
     rv = client.get("/map/routers")
@@ -58,7 +43,7 @@ def test_get_trunks(client):
     responses.add(
         method=responses.GET,
         url=re.compile(r'.*/map/services.*'),
-        json=_load_test_data('inprov-services.json')
+        json=load_test_data('inprov-services.json')
     )
 
     rv = client.get("/map/trunks")

tox.ini

 [tox]
-envlist = lint, typecheck, docs
+envlist = coverage, lint, typecheck, docs
 
 [testenv:coverage]
 description = Run unit tests and save coverage
@@ -23,6 +23,7 @@ deps =
     mypy
     types-jsonschema
     types-requests
+    types-pika
 commands = mypy mapping_provider
 
 [testenv:sbom]