Cleanup test

awslabs · Jan 30, 2024 · a7d63cf · a7d63cf
1 parent 2a4f0b7
commit a7d63cf
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/unit/ecs_tasks/test_main.py b/tests/unit/ecs_tasks/test_main.py
@@ -265,7 +265,10 @@ def test_it_assumes_role(
 ):
     mock_fs.S3FileSystem.return_value = mock_fs
     mock_file = MagicMock()
-    mock_file.metadata.return_value = {"VersionId": b"abc123"}
+    mock_file.metadata.return_value = {
+        "VersionId": b"abc123",
+        "Content-Length": b"5558",
+    }
     mock_fs.open_input_stream.return_value.__enter__.return_value = mock_file
     mock_get_object_info.return_value = {"Metadata": {}}, None
     mock_delete.return_value = pa.BufferOutputStream(), {"DeletedRows": 1}