NRL-518 Add step for upserts, refactor create and upsert into one method

Author: axelkrastek1-nhs

tests/features/steps/2_request.py

@@ -94,17 +94,16 @@ def create_post_document_reference_step(context: Context, ods_code: str):
         context.add_cleanup(lambda: context.repository.delete_by_id(doc_ref_id))
 
 
-@when(
-    "producer 'TSTCUS' requests creation of a DocumentReference with default test values except '{section}' is"
-)
-def create_post_body_step(context: Context, section: str):
+def _create_or_upsert_body_step(
+    context: Context, method: str, pointer_id: str = "TSTCUS-sample-id-00000"
+):
     client = producer_client_from_context(context, "TSTCUS")
 
     if not context.text:
         raise ValueError("No document reference text snippet provided")
 
-    doc_ref = create_test_document_reference_with_defaults(section, context.text)
-    context.response = client.create_text(doc_ref)
+    doc_ref = create_test_document_reference_with_defaults("content", context.text)
+    context.response = getattr(client, method)(doc_ref)
 
     if context.response.status_code == 201:
         doc_ref_id = context.response.headers["Location"].split("/")[-1]
@@ -114,6 +113,20 @@ def create_post_body_step(context: Context, section: str):
         context.add_cleanup(lambda: context.repository.delete_by_id(doc_ref_id))
 
 
+@when(
+    "producer 'TSTCUS' requests creation of a DocumentReference with default test values except '{section}' is"
+)
+def create_post_body_step(context: Context, section: str):
+    _create_or_upsert_body_step(context, "create_text")
+
+
+@when(
+    "producer 'TSTCUS' requests upsert of a DocumentReference with pointerId '{pointer_id}' and default test values except '{section}' is"
+)
+def upsert_post_body_step(context: Context, section: str, pointer_id: str):
+    _create_or_upsert_body_step(context, "upsert_text", pointer_id)
+
+
 @when("producer '{ods_code}' upserts a DocumentReference with values")
 def create_put_document_reference_step(context: Context, ods_code: str):
     client = producer_client_from_context(context, ods_code)

tests/utilities/api_clients.py

@@ -213,6 +213,14 @@ def upsert(self, doc_ref):
             cert=self.config.client_cert,
         )
 
+    def upsert_text(self, doc_ref):
+        return requests.put(
+            f"{self.api_url}/DocumentReference",
+            data=doc_ref,
+            headers=self.request_headers,
+            cert=self.config.client_cert,
+        )
+
     def update(self, doc_ref, doc_ref_id: str):
         return requests.put(
             f"{self.api_url}/DocumentReference/{doc_ref_id}",