From a7d63cf111c9cc5e4da0802f8d985fe531f41948 Mon Sep 17 00:00:00 2001
From: Matteo Figus <matteofigus@gmail.com>
Date: Tue, 30 Jan 2024 18:48:13 +0000
Subject: [PATCH] Cleanup test

---
 tests/unit/ecs_tasks/test_main.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/tests/unit/ecs_tasks/test_main.py b/tests/unit/ecs_tasks/test_main.py
index d976ec10..919eb956 100644
--- 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}