mock snmp_refresh_interface again, where needed

test/conftest.py

@@ -242,6 +242,9 @@ def mocked_worker_module(
         'inventory_provider.snmp.get_peer_state_info',
         _mocked_snmp_peerings)
 
+    _mocked_get_organizations = mocker.patch(
+        'brian_dashboard_manager.get_organizations')
+    _mocked_get_organizations.return_value = ['A', 'B', 'C']
     def _mocked_load_juniper_netconf_config(hostname, _):
         return etree.XML(cached_test_data['netconf:' + hostname])
     mocker.patch(

test/per_router/test_celery_worker.py

@@ -2,6 +2,7 @@
 just checks that the worker methods call the right functions
 and some data ends up in the right place ... otherwise not very detailed
 """
+import re
 from inventory_provider.tasks import worker
 from inventory_provider.tasks.common import _get_redis
 
@@ -58,6 +59,14 @@ def test_reload_router_config(mocked_worker_module, router, mocker):
     saved_data = {}
     for key in ('netconf:' + router, 'snmp-interfaces:' + router):
         saved_data[key] = backend_db().pop(key)
+
+    saved_peerings = {}
+    for key in backend_db().keys():
+        if re.match(r'^snmp-peerings:\d.*', key):
+            saved_peerings[key] = backend_db()[key]
+    for key in saved_peerings.keys():
+        del backend_db()[key]
+
     assert 'netconf:' + router not in backend_db()
     assert 'snmp-interfaces:' + router not in backend_db()
 
@@ -68,10 +77,18 @@ def test_reload_router_config(mocked_worker_module, router, mocker):
         'inventory_provider.tasks.worker.netconf_refresh_config.apply',
         _mocked_netconf_refresh_config_apply)
 
+    def _mocked_snmp_refresh_peerings_apply(args):
+        assert len(args) == 3
+        backend_db().update(saved_peerings)
+    mocker.patch(
+        'inventory_provider.tasks.worker.snmp_refresh_peerings.apply',
+        _mocked_snmp_refresh_peerings_apply)
+
     def _mocked_snmp_refresh_interfaces_apply(args):
-        assert len(args) == 2
+        assert len(args) == 3
         key = 'snmp-interfaces:' + args[0]
         backend_db()[key] = saved_data[key]
+
     mocker.patch(
         'inventory_provider.tasks.worker.snmp_refresh_interfaces.apply',
         _mocked_snmp_refresh_interfaces_apply)