Skip to content
Snippets Groups Projects
Commit 3add5eb1 authored by Erik Reid's avatar Erik Reid
Browse files

moved a fixture to conftest

parent b58af6fe
Branches
Tags
No related merge requests found
......@@ -6,9 +6,12 @@ import re
import shutil
import tempfile
from lxml import etree
import pytest
import inventory_provider
from inventory_provider import config
from inventory_provider.tasks import worker
TEST_DATA_DIRNAME = os.path.realpath(os.path.join(
inventory_provider.__path__[0],
......@@ -197,3 +200,22 @@ def mocked_netifaces(mocker):
data = ast.literal_eval(s)
mocker.patch('netifaces.interfaces', lambda: data.keys())
mocker.patch('netifaces.ifaddresses', lambda n: data[n])
@pytest.fixture
def mocked_worker_module(
mocker, mocked_redis, data_config,
cached_test_data, mocked_netifaces):
worker.InventoryTask.config = data_config
def _mocked_snmp_interfaces(hostname, community, _):
return json.loads(cached_test_data['snmp-interfaces:' + hostname])
mocker.patch(
'inventory_provider.snmp.get_router_interfaces',
_mocked_snmp_interfaces)
def _mocked_load_juniper_netconf_config(hostname, _):
return etree.XML(cached_test_data['netconf:' + hostname])
mocker.patch(
'inventory_provider.juniper.load_config',
_mocked_load_juniper_netconf_config)
......@@ -2,10 +2,6 @@
just checks that the worker methods call the right functions
and some data ends up in the right place ... otherwise not very detailed
"""
import json
from lxml import etree
import pytest
from inventory_provider.tasks import worker
from inventory_provider.tasks.common import get_redis
......@@ -16,26 +12,6 @@ def backend_db():
}).db
@pytest.fixture
def mocked_worker_module(
mocker, mocked_redis, data_config,
cached_test_data, mocked_netifaces):
worker.InventoryTask.config = data_config
def _mocked_snmp_interfaces(hostname, community, _):
return json.loads(cached_test_data['snmp-interfaces:' + hostname])
mocker.patch(
'inventory_provider.snmp.get_router_interfaces',
_mocked_snmp_interfaces)
def _mocked_load_juniper_netconf_config(hostname, _):
return etree.XML(cached_test_data['netconf:' + hostname])
mocker.patch(
'inventory_provider.juniper.load_config',
_mocked_load_juniper_netconf_config)
def test_netconf_refresh_config(mocked_worker_module, router):
del backend_db()['netconf:' + router]
worker.netconf_refresh_config(router)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment