diff --git a/test/test_data_routes.py b/test/test_data_routes.py index 03642b09a4e6dbf1fb60cb43835e7e399191072e..e7ffe89fb3381e248cdfd1d0ca4be0de798a6c0a 100644 --- a/test/test_data_routes.py +++ b/test/test_data_routes.py @@ -192,8 +192,18 @@ class MockedRedis(object): def keys(self, *args, **kwargs): return list([k.encode("utf-8") for k in MockedRedis.db.keys()]) +@pytest.fixture +def client_with_mocked_data(mocker, client): + mocker.patch( + 'inventory_provider.router_details.redis.StrictRedis', + MockedRedis) + mocker.patch( + 'inventory_provider.routes.data.redis.StrictRedis', + MockedRedis) + return client + -def test_routers_list(mocker, client): +def test_routers_list(client_with_mocked_data): routers_list_schema = { "$schema": "http://json-schema.org/draft-07/schema#", @@ -201,13 +211,7 @@ def test_routers_list(mocker, client): "items": {"type": "string"} } - mocker.patch( - 'inventory_provider.router_details.redis.StrictRedis', - MockedRedis) - mocker.patch( - 'inventory_provider.routes.data.redis.StrictRedis', - MockedRedis) - rv = client.post( + rv = client_with_mocked_data.post( "data/routers", headers=DEFAULT_REQUEST_HEADERS) assert rv.status_code == 200