diff --git a/test/test_worker.py b/test/test_worker.py index 825c98909ffdd924a7100d95a245d023168e46ec..2dbce4732031de4decae67b95d0db64d0a524994 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():