Cleaning audit bucket after use per every test run

Author: shweta-nhs

tests/integration/conftest.py

@@ -208,6 +208,25 @@ def flask_function(lambda_client: BaseClient, iam_role: str, lambda_zip: Path) -
     lambda_client.delete_function(FunctionName=function_name)
 
 
+@pytest.fixture(autouse=True)
+def clean_audit_bucket(s3_client: BaseClient, audit_bucket: str):
+    objects_to_delete = []
+    paginator = s3_client.get_paginator('list_objects_v2')
+    pages = paginator.paginate(Bucket=audit_bucket)
+    for page in pages:
+        if "Contents" in page:
+            for obj in page["Contents"]:
+                objects_to_delete.append({"Key": obj["Key"]})
+
+    if objects_to_delete:
+        s3_client.delete_objects(
+            Bucket=audit_bucket,
+            Delete={"Objects": objects_to_delete, "Quiet": True},
+        )
+
+    yield
+
+
 @pytest.fixture(scope="session")
 def flask_function_url(lambda_client: BaseClient, flask_function: str) -> URL:
     response = lambda_client.create_function_url_config(FunctionName=flask_function, AuthType="NONE")