import os from unittest.mock import patch from src.functions.data_generator_vendor_a import app @patch.dict(os.environ, { "RAW_BUCKET_NAME": 'mock-raw-bucket' }) def test_data_generator_vendor_a(mock_s3_client, mock_raw_bucket): expected_s3_data = { "name": "ABC GOODS INC.", "status": "DELIVERED", "object_id": 12345 } app.lambda_handler({}, "") response = mock_s3_client.list_objects_v2(Bucket=mock_raw_bucket) contents = response["Contents"][0] s3_key = contents["Key"] result = mock_s3_client.get_object(Bucket=mock_raw_bucket, Key=s3_key) data = eval(result["Body"].read().decode()) assert data == expected_s3_data