From 14219b1fca86f2229d2b69455bdbb4eb952cf504 Mon Sep 17 00:00:00 2001 From: Laerte Pereira Date: Wed, 22 Jan 2025 07:16:22 -0300 Subject: [PATCH] fix: test_s3_export fails with boto3 >= 1.36.0 --- tests/test_feedexport.py | 16 +++++++++++----- 1 file changed, 11 insertions(+), 5 deletions(-) diff --git a/tests/test_feedexport.py b/tests/test_feedexport.py index 0f149f172..81d05e2a3 100644 --- a/tests/test_feedexport.py +++ b/tests/test_feedexport.py @@ -2622,18 +2622,24 @@ class BatchDeliveriesTest(FeedExportTestBase): stubs = [] def open(self, *args, **kwargs): + from botocore import __version__ as botocore_version from botocore.stub import ANY, Stubber + from packaging.version import Version + + expected_params = { + "Body": ANY, + "Bucket": bucket, + "Key": ANY, + } + if Version(botocore_version) >= Version("1.36.0"): + expected_params["ChecksumAlgorithm"] = ANY stub = Stubber(self.s3_client) stub.activate() CustomS3FeedStorage.stubs.append(stub) stub.add_response( "put_object", - expected_params={ - "Body": ANY, - "Bucket": bucket, - "Key": ANY, - }, + expected_params=expected_params, service_response={}, ) return super().open(*args, **kwargs)