[NDR-293] Add Eicar to LG E2E test

Author: megan-bower4

lambdas/tests/e2e/api/test_upload_document_api.py

@@ -175,25 +175,36 @@ def test_create_document_virus(test_data, snapshot_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:
-        files = {"file": f}
-        presign_response = requests.put(presign_uri, files=files)
-        assert presign_response.status_code == 200
+    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
+
+        retrieve_url = (
+            f"https://{API_ENDPOINT}/FhirDocumentReference/{upload_response['id']}"
+        )
 
-    retrieve_url = (
-        f"https://{API_ENDPOINT}/FhirDocumentReference/{upload_response['id']}"
-    )
+        def condition(response_json):
+            logging.info(response_json)
+            return response_json.get("docStatus", False) == "cancelled"
 
-    def condition(response_json):
-        logging.info(response_json)
-        return response_json.get("docStatus", False) == "cancelled"
+        raw_retrieve_response = fetch_with_retry(retrieve_url, condition)
+        retrieve_response = raw_retrieve_response.json()
 
-    raw_retrieve_response = fetch_with_retry(retrieve_url, condition)
-    retrieve_response = raw_retrieve_response.json()
+        assert upload_response == snapshot_json(exclude=paths("id", "date"))
+        assert retrieve_response == snapshot_json(exclude=paths("id", "date"))
 
-    assert upload_response == snapshot_json(exclude=paths("id", "date"))
-    assert retrieve_response == snapshot_json(exclude=paths("id", "date"))
+    finally:
+        # Delete the EICAR file
+        if os.path.exists(eicar_path):
+            os.remove(eicar_path)
 
 
 def test_create_document_does_not_save_raw(test_data):