From 89882afed8d51ad335c7cf37b7434b6198eaa8ce Mon Sep 17 00:00:00 2001
From: Sam Roberts <sam.roberts@geant.org>
Date: Fri, 11 Oct 2024 16:12:23 +0100
Subject: [PATCH] fix tests

---
 inventory_provider/routes/poller.py | 1 +
 test/test_worker.py                 | 4 ++--
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/inventory_provider/routes/poller.py b/inventory_provider/routes/poller.py
index ce107c47..c5a32aa6 100644
--- a/inventory_provider/routes/poller.py
+++ b/inventory_provider/routes/poller.py
@@ -548,6 +548,7 @@ def _get_dashboards(interface, region=None):
     if region == 'EAP':
         yield BRIAN_DASHBOARDS.EAP
 
+
 def _get_dashboard_data(ifc, customers, regions=None):
     if regions is None:
         regions = []
diff --git a/test/test_worker.py b/test/test_worker.py
index 1d61282f..452550b3 100644
--- a/test/test_worker.py
+++ b/test/test_worker.py
@@ -788,7 +788,7 @@ def test_populate_poller_interfaces_cache(
                     "type": "SERVICE TYPE",
                     "status": "operational"
                 }],
-                'customers': {'geant'}
+                'customers': [{'name': 'GEANT', 'type': 'UNKNOWN'}]
             },
             "AE_A.456": {
                 'services': [{
@@ -797,7 +797,7 @@ def test_populate_poller_interfaces_cache(
                     "type": "SERVICE TYPE",
                     "status": "operational"
                 }],
-                'customers': {'geant'}
+                'customers': [{'name': 'GEANT', 'type': 'UNKNOWN'}]
             }
         }
     }
-- 
GitLab