fix test

Flagsmith · Nov 7, 2023 · b460236 · b460236
1 parent 04b59b7
commit b460236
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/api/environments/dynamodb/tests/test_dynamodb_identity_wrapper.py b/api/environments/dynamodb/tests/test_dynamodb_identity_wrapper.py
@@ -249,14 +249,17 @@ def test_is_enabled_is_true_if_dynamo_table_name_is_set(settings, mocker):
         "environments.dynamodb.dynamodb_wrapper.DynamoIdentityWrapper.table_name",
         table_name,
     )
+    mocked_config = mocker.patch("environments.dynamodb.dynamodb_wrapper.Config")
     mocked_boto3 = mocker.patch("environments.dynamodb.dynamodb_wrapper.boto3")
 
     # When
     dynamo_identity_wrapper = DynamoIdentityWrapper()
     # Then
 
     assert dynamo_identity_wrapper.is_enabled is True
-    mocked_boto3.resource.assert_called_with("dynamodb")
+    mocked_boto3.resource.assert_called_with(
+        "dynamodb", config=mocked_config(tcp_keepalive=True)
+    )
     mocked_boto3.resource.return_value.Table.assert_called_with(table_name)
 
 

diff --git a/api/poetry.lock b/api/poetry.lock