fix: Mock STS calls for some JA upload tests (#229)

Signed-off-by: Caden Marofke <marofke@amazon.com>
aws-deadline · Mar 22, 2024 · 119aabd · 119aabd
1 parent 143cc6c
commit 119aabd
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/test/unit/deadline_job_attachments/test_upload.py b/test/unit/deadline_job_attachments/test_upload.py
@@ -2348,6 +2348,7 @@ def test_separate_files_by_size(
         )
         assert actual_queues == expected_queues
 
+    @mock_sts
     @pytest.mark.parametrize(
         "manifest_version",
         [
@@ -2395,6 +2396,7 @@ def test_upload_object_to_cas_skips_upload_with_cache(
             assert file_size == 5
             s3_cache.put_entry.assert_not_called()
 
+    @mock_sts
     @pytest.mark.parametrize(
         "manifest_version",
         [