diff --git a/test/per_router/test_celery_worker.py b/test/per_router/test_celery_worker.py index 81ad1be8b87c8637d70694c4464239229e9d855d..fa8037043e5e8b3ba3e85ae2489c0f0f14bcb0d9 100644 --- a/test/per_router/test_celery_worker.py +++ b/test/per_router/test_celery_worker.py @@ -133,9 +133,11 @@ def test_reload_router_config(mocked_worker_module, router, mocker): 'inventory_provider.tasks.worker.snmp_refresh_interfaces.apply', _mocked_snmp_refresh_interfaces_apply) + def _mocked_update_status(self, **kwargs): + pass mocker.patch( - 'inventory_provider.tasks.worker.snmp_refresh_interfaces.apply', - _mocked_snmp_refresh_interfaces_apply) + 'inventory_provider.tasks.worker.InventoryTask.update_state', + _mocked_update_status) worker.reload_router_config(router) assert 'netconf:' + router in MockedRedis.db