From 74790e57cfc6a6f1733eb971a0fe48791d9d1a61 Mon Sep 17 00:00:00 2001
From: Neda Moeini <neda.moeini@geant.org>
Date: Tue, 2 Jul 2024 14:25:22 +0200
Subject: [PATCH] Fixed the tests.

---
 inventory_provider/tasks/worker.py | 2 +-
 test/test_gap.py                   | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/inventory_provider/tasks/worker.py b/inventory_provider/tasks/worker.py
index 864dd803..fd0d4f5d 100644
--- a/inventory_provider/tasks/worker.py
+++ b/inventory_provider/tasks/worker.py
@@ -2068,4 +2068,4 @@ def collate_netconf_interfaces_all_cache(warning_callback=lambda s: None):
             "Failed to collate netconf-interfaces, logging exception")
 
     r.set(netconf_all_key, json.dumps(netconf_interface_docs))
-    r.set(lab_netconf_all_key, json.dumps(lab_netconf_interface_docs))
\ No newline at end of file
+    r.set(lab_netconf_all_key, json.dumps(lab_netconf_interface_docs))
diff --git a/test/test_gap.py b/test/test_gap.py
index 6801fa07..3bb88330 100644
--- a/test/test_gap.py
+++ b/test/test_gap.py
@@ -65,7 +65,7 @@ def test_extract_router_info(mocker):
 
 
 def test_load_routers_from_orchestrator(mocker):
-    mocker.patch('inventory_provider.gap.make_request', {
+    mocker.patch('inventory_provider.gap.make_request', return_value={
         'data': {
             'subscriptions': {
                 'page': [
@@ -75,7 +75,7 @@ def test_load_routers_from_orchestrator(mocker):
             }
         }
     })
-    mocker.patch('inventory_provider.gap.extract_router_info', return_value=[
+    mocker.patch('inventory_provider.gap.extract_router_info', side_effect=[
         {'fqdn': 'fqdn1', 'vendor': 'vendor1'},
         {'fqdn': 'fqdn2', 'vendor': 'vendor2'}
     ])
-- 
GitLab