NRL-1165 reflect model changes in feature test utils

Author: katebobyn-nhs

tests/features/utils/data.py

@@ -10,7 +10,6 @@
 from nrlf.producer.fhir.r4.model import (
     Attachment,
     CodeableConcept,
-    Coding,
     ContentStabilityExtension,
     ContentStabilityExtensionCoding,
     ContentStabilityExtensionValueCodeableConcept,
@@ -19,6 +18,8 @@
     DocumentReferenceContext,
     DocumentReferenceRelatesTo,
     Identifier,
+    NRLCodeableConcept,
+    NRLCoding,
     NRLFormatCode,
     Reference,
 )
@@ -80,9 +81,9 @@ def create_test_document_reference(items: dict) -> DocumentReference:
             ],
         ),
         context=DocumentReferenceContext(
-            practiceSetting=CodeableConcept(
+            practiceSetting=NRLCodeableConcept(
                 coding=[
-                    Coding(
+                    NRLCoding(
                         system=SNOMED_SYSTEM_URL,
                         code=str(practice_setting_code),
                         display=practice_setting_display,
@@ -103,7 +104,7 @@ def create_test_document_reference(items: dict) -> DocumentReference:
         )
 
         base_doc_ref.type = CodeableConcept(
-            coding=[Coding(system=type_system, code=type_code, display=type_display)]
+            coding=[NRLCoding(system=type_system, code=type_code, display=type_display)]
         )
 
     if items.get("subject"):
@@ -136,9 +137,9 @@ def create_test_document_reference(items: dict) -> DocumentReference:
             f"{SNOMED_SYSTEM_URL}|{items['category']}", {}
         ).get("display")
         base_doc_ref.category = [
-            CodeableConcept(
+            NRLCodeableConcept(
                 coding=[
-                    Coding(
+                    NRLCoding(
                         system=SNOMED_SYSTEM_URL,
                         code=items["category"],
                         display=category_display,