NRL-1050 Fix integration tests

Author: axelkrastek1-nhs

tests/features/consumer/readDocumentReference-success.feature

@@ -28,7 +28,8 @@ Feature: Consumer - readDocumentReference - Success Scenarios
           "coding": [
             {
               "system": "http://snomed.info/sct",
-              "code": "736253002"
+              "code": "736253002",
+              "display": "Mental health crisis plan"
             }
           ]
         },
@@ -102,7 +103,8 @@ Feature: Consumer - readDocumentReference - Success Scenarios
           "coding": [
             {
               "system": "http://snomed.info/sct",
-              "code": "736253002"
+              "code": "736253002",
+              "display": "Mental health crisis plan"
             }
           ]
         },

tests/features/producer/readDocumentReference-success.feature

@@ -30,7 +30,8 @@ Feature: Producer - readDocumentReference - Success Scenarios
           "coding": [
             {
               "system": "http://snomed.info/sct",
-              "code": "736253002"
+              "code": "736253002",
+              "display": "Mental health crisis plan"
             }
           ]
         },
@@ -104,7 +105,8 @@ Feature: Producer - readDocumentReference - Success Scenarios
           "coding": [
             {
               "system": "http://snomed.info/sct",
-              "code": "736253002"
+              "code": "736253002",
+              "display": "Mental health crisis plan"
             }
           ]
         },

tests/features/utils/data.py

@@ -1,4 +1,4 @@
-from layer.nrlf.core.constants import CATEGORY_ATTRIBUTES
+from layer.nrlf.core.constants import CATEGORY_ATTRIBUTES, TYPE_ATTRIBUTES
 from nrlf.producer.fhir.r4.model import (
     Attachment,
     CodeableConcept,
@@ -40,9 +40,13 @@ def create_test_document_reference(items: dict) -> DocumentReference:
     if items.get("id"):
         base_doc_ref.id = items["id"]
 
-    if items.get("type"):
+    if type_code := items.get("type"):
+        type_system = items.get("type_system", "http://snomed.info/sct")
+        type_str = f"{type_system}|{type_code}"
+        type_display = TYPE_ATTRIBUTES.get(type_str, {}).get("display")
+
         base_doc_ref.type = CodeableConcept(
-            coding=[Coding(system="http://snomed.info/sct", code=items["type"])]
+            coding=[Coding(system=type_system, code=type_code, display=type_display)]
         )
 
     if items.get("subject"):