From 3d69d5e56f2f02df02ee0f394f2d22b164b524b6 Mon Sep 17 00:00:00 2001
From: Erik Reid <erik.reid@geant.org>
Date: Fri, 30 Nov 2018 14:04:03 +0100
Subject: [PATCH] put mocking in a fixture

---
 test/test_data_routes.py | 20 ++++++++++++--------
 1 file changed, 12 insertions(+), 8 deletions(-)

diff --git a/test/test_data_routes.py b/test/test_data_routes.py
index 03642b09..e7ffe89f 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
-- 
GitLab