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

move some db mocking to conftest

parent c736e9a2
No related branches found
No related tags found
No related merge requests found
import ast
import contextlib
import json
import netifaces
import os
......@@ -184,6 +185,11 @@ def mocked_netifaces(mocker):
mocker.patch('netifaces.ifaddresses', lambda n: data[n])
@contextlib.contextmanager
def _mocked_db_connection(ignored):
yield None
@pytest.fixture
def mocked_worker_module(
mocker, mocked_redis, data_config_filename,
......@@ -205,3 +211,13 @@ def mocked_worker_module(
mocker.patch(
'inventory_provider.juniper.load_config',
_mocked_load_juniper_netconf_config)
def _mocked_get_service_users(cx, service_ids):
for id in service_ids:
yield {'service_id': id, 'user': 'AAAAA'}
yield {'service_id': id, 'user': 'BBBB'}
mocker.patch('inventory_provider.db.db.connection', _mocked_db_connection)
mocker.patch(
'inventory_provider.db.opsdb.get_service_users',
_mocked_get_service_users)
......@@ -21,28 +21,12 @@ def backend_db():
}).db
@contextlib.contextmanager
def mocked_connection(ignored):
yield None
def test_build_interface_services(mocker, mocked_worker_module):
def test_build_interface_services(mocked_worker_module):
"""
checks that valid interface service objects are created
:param mocked_worker_module: fixture
:return:
"""
def _mocked_get_service_users(cx, service_ids):
for id in service_ids:
yield {'service_id': id, 'user': 'AAAAA'}
yield {'service_id': id, 'user': 'BBBB'}
mocker.patch('inventory_provider.db.db.connection', mocked_connection)
mocker.patch(
'inventory_provider.db.opsdb.get_service_users',
_mocked_get_service_users)
ifc_schema = {
'$schema': 'http://json-schema.org/draft-07/schema#',
'type': 'object',
......
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