From 4d5719ea787883e3348d5accf0108c6e6fd778cb Mon Sep 17 00:00:00 2001
From: Robert Latta <robert.latta@geant.org>
Date: Wed, 2 Feb 2022 09:53:39 +0000
Subject: [PATCH] added SID retrieval to test
---
test/test_worker.py | 13 +++++++++++++
1 file changed, 13 insertions(+)
diff --git a/test/test_worker.py b/test/test_worker.py
index 825c9890..2dbce473 100644
--- a/test/test_worker.py
+++ b/test/test_worker.py
@@ -56,6 +56,14 @@ def test_extract_ims_data(mocker):
{'port_a_id': '2', 'value': '2A'}
]
)
+ mocker.patch(
+ 'inventory_provider.tasks.worker.ims_data.get_ids_and_sids',
+ return_value=(x for x in [
+ (111111, 'SID-01'),
+ (111112, 'SID-02'),
+ (111113, 'SID-03')
+ ])
+ )
res = extract_ims_data()
assert res['locations'] == {'loc_a': 'LOC A', 'loc_b': 'LOC B'}
assert res['lg_routers'] == ['lg router 1', 'lg router 2']
@@ -81,6 +89,11 @@ def test_extract_ims_data(mocker):
],
'2': [{'port_a_id': '2', 'value': '2A'}]
}
+ assert res['circuit_ids_sids'] == {
+ 111111: 'SID-01',
+ 111112: 'SID-02',
+ 111113: 'SID-03'
+ }
def test_transform_ims_data():
--
GitLab