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
No related branches found
No related tags found
No related merge requests found
...@@ -6,9 +6,12 @@ import re ...@@ -6,9 +6,12 @@ import re
import shutil import shutil
import tempfile import tempfile
from lxml import etree
import pytest import pytest
import inventory_provider import inventory_provider
from inventory_provider import config from inventory_provider import config
from inventory_provider.tasks import worker
TEST_DATA_DIRNAME = os.path.realpath(os.path.join( TEST_DATA_DIRNAME = os.path.realpath(os.path.join(
inventory_provider.__path__[0], inventory_provider.__path__[0],
...@@ -197,3 +200,22 @@ def mocked_netifaces(mocker): ...@@ -197,3 +200,22 @@ def mocked_netifaces(mocker):
data = ast.literal_eval(s) data = ast.literal_eval(s)
mocker.patch('netifaces.interfaces', lambda: data.keys()) mocker.patch('netifaces.interfaces', lambda: data.keys())
mocker.patch('netifaces.ifaddresses', lambda n: data[n]) 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 @@ ...@@ -2,10 +2,6 @@
just checks that the worker methods call the right functions just checks that the worker methods call the right functions
and some data ends up in the right place ... otherwise not very detailed 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 import worker
from inventory_provider.tasks.common import get_redis from inventory_provider.tasks.common import get_redis
...@@ -16,26 +12,6 @@ def backend_db(): ...@@ -16,26 +12,6 @@ def backend_db():
}).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): def test_netconf_refresh_config(mocked_worker_module, router):
del backend_db()['netconf:' + router] del backend_db()['netconf:' + router]
worker.netconf_refresh_config(router) worker.netconf_refresh_config(router)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment