Modularise test_spec.py

Author: RamParameswaran

tests/web_api/test_spec.py

@@ -1,27 +1,25 @@
-# -*- coding: utf-8 -*-
-
-import json
-from http.client import OK
-
 import dpath
-from . import subject
+import json
+import pytest
+from http import client
 
 
 def assert_items_equal(x, y):
     assert sorted(x) == sorted(y)
 
 
-openAPI_response = subject.get('/spec')
-
-
-def test_return_code():
-    assert openAPI_response.status_code == OK
+def test_return_code(test_client):
+    openAPI_response = test_client.get('/spec')
+    assert openAPI_response.status_code == client.OK
 
 
-body = json.loads(openAPI_response.data.decode('utf-8'))
+@pytest.fixture(scope="module")
+def body(test_client):
+    openAPI_response = test_client.get('/spec')
+    return json.loads(openAPI_response.data.decode('utf-8'))
 
 
-def test_paths():
+def test_paths(body):
     assert_items_equal(
         body['paths'],
         ["/parameter/{parameterID}",
@@ -35,15 +33,15 @@ def test_paths():
         )
 
 
-def test_entity_definition():
+def test_entity_definition(body):
     assert 'parents' in dpath.get(body, 'definitions/Household/properties')
     assert 'children' in dpath.get(body, 'definitions/Household/properties')
     assert 'salary' in dpath.get(body, 'definitions/Person/properties')
     assert 'rent' in dpath.get(body, 'definitions/Household/properties')
     assert 'number' == dpath.get(body, 'definitions/Person/properties/salary/additionalProperties/type')
 
 
-def test_situation_definition():
+def test_situation_definition(body):
     situation_input = body['definitions']['SituationInput']
     situation_output = body['definitions']['SituationOutput']
     for situation in situation_input, situation_output:
@@ -53,5 +51,5 @@ def test_situation_definition():
         assert "#/definitions/Person" == dpath.get(situation, '/properties/persons/additionalProperties/$ref')
 
 
-def test_host():
+def test_host(body):
     assert 'http' not in body['host']