NRL-638 add cleanup to upsert helper

Author: katebobyn-nhs

tests/features/steps/2_request.py

@@ -121,8 +121,12 @@ def create_put_document_reference_step(context: Context, ods_code: str):
     items = {row["property"]: row["value"] for row in context.table}
 
     doc_ref = create_test_document_reference(items)
+    doc_ref_id = items.get(id)
     context.response = client.upsert(doc_ref.dict(exclude_none=True))
 
+    if context.response.status_code == 201:
+        context.add_cleanup(lambda: context.repository.delete_by_id(doc_ref_id))
+
 
 @when(
     "producer '{ods_code}' requests to delete DocumentReference with id '{doc_ref_id}'"