[NDR-293] Use EICAR file in virus test

Author: megan-bower4

lambdas/tests/e2e/api/fhir/test_upload_document_fhir_api.py

@@ -91,36 +91,45 @@ def test_create_document_presign_fails():
 
 
 def test_create_document_virus(test_data):
-    record = {
-        "ods": "H81109",
-        "nhs_number": "9730154260",
-    }
+    record = {"ods": "H81109", "nhs_number": "9730154260"}
     payload = pdm_data_helper.create_upload_payload(record)
 
     raw_upload_response = upload_document(payload)
     assert raw_upload_response.status_code == 200
     record["id"] = raw_upload_response.json()["id"].split("~")[1]
     test_data.append(record)
 
-    # Presigned upload
     upload_response = raw_upload_response.json()
     presign_uri = upload_response["content"][0]["attachment"]["url"]
     del upload_response["content"][0]["attachment"]["url"]
-    sample_pdf_path = os.path.join(os.path.dirname(__file__), "files", "dummy.pdf")
-    with open(sample_pdf_path, "rb") as f:
-        presign_response = requests.put(presign_uri, files={"file": f})
-        assert presign_response.status_code == 200
-
-    def condition(response_json):
-        logging.info(response_json)
-        return response_json.get("docStatus", False) == "cancelled"
-
-    raw_retrieve_response = retrieve_document_with_retry(
-        upload_response["id"], condition
-    )
-    retrieve_response = raw_retrieve_response.json()
 
-    assert retrieve_response["docStatus"] == "cancelled"
+    try:
+        # Create EICAR file
+        eicar_path = os.path.join(os.path.dirname(__file__), "files", "eicar.com")
+        eicar_base64 = "WDVPIVAlQEFQWzRcUFpYNTRQXlkyQ0MpN30kRUlDQVItU1RBTkRBUkQtQU5USVZJUlUtVEVTVC1GSUxFISRIK0gq"
+        with open(eicar_path, "wb") as f:
+            f.write(base64.b64decode(eicar_base64))
+
+        # Upload EICAR file to presigned URL
+        with open(eicar_path, "rb") as f:
+            presign_response = requests.put(presign_uri, files={"file": f})
+            assert presign_response.status_code == 200
+
+        def condition(response_json):
+            logging.info(response_json)
+            return response_json.get("docStatus", False) == "cancelled"
+
+        raw_retrieve_response = retrieve_document_with_retry(
+            upload_response["id"], condition
+        )
+        retrieve_response = raw_retrieve_response.json()
+
+        assert retrieve_response["docStatus"] == "cancelled"
+
+    finally:
+        # Delete the EICAR file
+        if os.path.exists(eicar_path):
+            os.remove(eicar_path)
 
 
 @pytest.mark.parametrize(