updated celery api

test/test_celery_worker.py

@@ -3,6 +3,8 @@ just checks that the worker methods call the right functions
 and some data ends up in the right place ... otherwise not very detailed
 """
 import logging
+
+from lxml import etree
 import pytest
 from inventory_provider.tasks import worker
 
@@ -31,52 +33,20 @@ def mocked_worker_module(mocker, data_config):
         MockedRedis)
 
 
-def test_juniper_refresh_bgp(
-        mocked_worker_module, mocker, cached_test_data):
-
-    def _mocked_fetch_bpg_config(hostname, _):
-        return cached_test_data[hostname]["bgp"]
-
-    mocker.patch(
-        'inventory_provider.tasks.worker.juniper.fetch_bgp_config',
-        _mocked_fetch_bpg_config)
-
-    for hostname in cached_test_data.keys():
-        assert 'hostname' not in MockedRedis.db
-        worker.juniper_refresh_bgp(hostname)
-        assert MockedRedis.db[hostname]['bgp']
-
-
-def test_juniper_refresh_interfaces(
-        mocked_worker_module, mocker, cached_test_data):
-
-    def _mocked_fetch_interfaces(hostname, _):
-        return cached_test_data[hostname]["interfaces"]
-
-    mocker.patch(
-        'inventory_provider.tasks.worker.juniper.fetch_interfaces',
-        _mocked_fetch_interfaces)
-
-    for hostname in cached_test_data.keys():
-        assert 'hostname' not in MockedRedis.db
-        worker.juniper_refresh_interfaces(hostname)
-        assert MockedRedis.db[hostname]['interfaces']
-
-
-def test_juniper_refresh_vrr(
+def test_netconf_refresh_config(
         mocked_worker_module, mocker, cached_test_data):
 
-    def _mocked_fetch_vrr_config(hostname, _):
-        return cached_test_data[hostname]["vrr"]
+    def _mocked_load_juniper_netconf_config(hostname, _):
+        return etree.XML(cached_test_data[hostname]["netconf"])
 
     mocker.patch(
-        'inventory_provider.tasks.worker.juniper.fetch_vrr_config',
-        _mocked_fetch_vrr_config)
+        'inventory_provider.tasks.worker.juniper.load_config',
+        _mocked_load_juniper_netconf_config)
 
     for hostname in cached_test_data.keys():
         assert 'hostname' not in MockedRedis.db
-        worker.juniper_refresh_vrr(hostname)
-        assert MockedRedis.db[hostname]['vrr']
+        worker.netconf_refresh_config(hostname)
+        assert MockedRedis.db[hostname]['netconf']
 
 
 def test_snmp_refresh_interfaces(