Skip to content

Commit dc3306e

Browse files
committed
fix test
1 parent 021dccf commit dc3306e

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

tests/test_api.py

+5-3
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,7 @@ def test_secrets_manager_fips_endpoint(self, mock_boto3_client):
3838
mock_boto3_client.assert_called_with(
3939
"secretsmanager",
4040
endpoint_url="https://secretsmanager-fips.us-gov-east-1.amazonaws.com",
41+
region_name="us-gov-east-1",
4142
)
4243
self.assertEqual(api_key, "test-api-key")
4344

@@ -56,7 +57,8 @@ def test_secrets_manager_different_region(self, mock_boto3_client):
5657

5758
mock_boto3_client.assert_called_with(
5859
"secretsmanager",
59-
endpoint_url="https://secretsmanager.us-west-1.amazonaws.com",
60+
endpoint_url=None,
61+
region_name="us-west-1",
6062
)
6163
self.assertEqual(api_key, "test-api-key")
6264

@@ -103,8 +105,8 @@ def test_no_fips_for_standard_regions(self, mock_boto3_client):
103105

104106
os.environ.clear()
105107
os.environ["AWS_REGION"] = "us-west-2"
106-
os.environ["DD_API_KEY_SECRET_ARN"] = "test-arn"
108+
os.environ["DD_API_KEY_SECRET_ARN"] = "arn:aws:secretsmanager:us-west-2:1234567890:secret:key-name-123ABC"
107109

108110
api.get_api_key()
109111

110-
mock_boto3_client.assert_called_with("secretsmanager", endpoint_url=None)
112+
mock_boto3_client.assert_called_with("secretsmanager", endpoint_url=None, region_name="us-west-2")

0 commit comments

Comments
 (0)