diff --git a/scripts/mock_client.py b/scripts/mock_client.py index d4be7cd..53b4dde 100644 --- a/scripts/mock_client.py +++ b/scripts/mock_client.py @@ -6,8 +6,8 @@ from pyinfra.storage.storages import get_s3_storage def build_message_bodies(): - minio_client = get_s3_storage() - for dossier_id, pdf_name in minio_client.list_bucket_files(CONFIG.test.bucket): + storage = get_s3_storage() + for dossier_id, pdf_name in storage.list_bucket_files(CONFIG.test.bucket): file_id = pdf_name.split(".")[0] yield json.dumps({"dossierId": dossier_id, "fileId": file_id})