NPA-5357: Update formatting and API sandbox test

Author: lelliott-hippo

sandbox/api/post_questionnaire_response.py

@@ -10,7 +10,9 @@
 )
 from .utils import generate_response_from_example
 
-QUESTIONNAIRE_RESPONSE_APP_BASE_PATH = "https://sandbox.api.service.nhs.uk/validated-relationships/FHIR/R4/QuestionnaireResponse"
+QUESTIONNAIRE_RESPONSE_APP_BASE_PATH = (
+    "https://sandbox.api.service.nhs.uk/validated-relationships/FHIR/R4/QuestionnaireResponse"
+)
 basicConfig(level=INFO, format="%(asctime)s - %(message)s")
 logger = getLogger(__name__)
 

sandbox/api/tests/test_post_questionnaire_response.py

@@ -10,21 +10,24 @@
     INTERNAL_SERVER_ERROR_EXAMPLE,
 )
 
+SANDBOX_API_URL = "https://sandbox.api.service.nhs.uk/validated-relationships"
 QUESTIONNAIRE_RESPONSE_API_ENDPOINT = "/FHIR/R4/QuestionnaireResponse"
 
 
 @pytest.mark.parametrize(
-    ("nhs_num", "response_file_name", "status_code"),
+    ("nhs_num", "response_file_name", "status_code", "id"),
     [
         (
             "9000000009",
             POST_QUESTIONNAIRE_RESPONSE__SUCCESS,
             200,
+            "156e1560-e532-4e2a-85ad-5aeff03dc43e",
         ),
         (
             "9000000017",
             POST_QUESTIONNAIRE_RESPONSE__SUCCESS,
             200,
+            "156e1560-e532-4e2a-85ad-5aeff03dc43e",
         ),
         (
             "9000000049",
@@ -42,6 +45,7 @@
 def test_post_questionnaire_response(
     mock_generate_response_from_example: MagicMock,
     nhs_num: str,
+    id: str,
     response_file_name: str,
     status_code: int,
     client: object,
@@ -57,6 +61,15 @@ def test_post_questionnaire_response(
     # Act
     response = client.post(QUESTIONNAIRE_RESPONSE_API_ENDPOINT, json=json)
     # Assert
-    mock_generate_response_from_example.assert_called_once_with(response_file_name, status_code)
+    if id is not None:
+        mock_generate_response_from_example.assert_called_once_with(
+            response_file_name,
+            status_code,
+            headers={
+                "location": f"{SANDBOX_API_URL}{QUESTIONNAIRE_RESPONSE_API_ENDPOINT}?ID={id}"
+            },
+        )
+    else:
+        mock_generate_response_from_example.assert_called_once_with(response_file_name, status_code)
     assert response.status_code == status_code
     assert response.json == loads(mocked_response.get_data(as_text=True))