Just write blob for storage upload (#583)

Modify write to storage, to only write the blob message
CDCgov · May 17, 2023 · 81fcbbb · 81fcbbb
1 parent 0ec1b2b
commit 81fcbbb
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/containers/ingestion/app/routers/cloud_storage.py b/containers/ingestion/app/routers/cloud_storage.py
@@ -71,7 +71,7 @@ def write_blob_to_cloud_storage_endpoint(
 
     full_file_name = input["file_name"]
     cloud_provider_connection.upload_object(
-        message=input,
+        message=input["blob"],
         container_name=input["bucket_name"],
         filename=full_file_name,
     )

diff --git a/containers/ingestion/tests/test_cloud_storage.py b/containers/ingestion/tests/test_cloud_storage.py
@@ -34,7 +34,9 @@ def test_cloud_storage_params_success(patched_blob_write, patched_get_provider):
     actual_response = client.post(client_url, json=test_request)
 
     patched_get_provider.return_value.upload_object.assert_called_with(
-        message=test_request, container_name="test_bucket", filename="test_file_name"
+        message=test_request["blob"],
+        container_name="test_bucket",
+        filename="test_file_name",
     )
 
     expected_message = (