diff --git a/test/per_router/test_celery_worker.py b/test/per_router/test_celery_worker.py index c943e40af1079e95d4b5d8be07c8be339dcff732..81ad1be8b87c8637d70694c4464239229e9d855d 100644 --- a/test/per_router/test_celery_worker.py +++ b/test/per_router/test_celery_worker.py @@ -118,8 +118,8 @@ def test_reload_router_config(mocked_worker_module, router, mocker): assert 'netconf:' + router not in MockedRedis.db assert 'snmp-interfaces:' + router not in MockedRedis.db - def _mocked_netconf_refresh_config_apply(hostname): - key = 'netconf:' + hostname + def _mocked_netconf_refresh_config_apply(args): + key = 'netconf:' + args[0] MockedRedis.db[key] = saved_data[key] mocker.patch( 'inventory_provider.tasks.worker.netconf_refresh_config.apply',