mirror of
https://github.com/GSA/notifications-api.git
synced 2026-04-12 05:09:35 -04:00
increase code coverage
This commit is contained in:
@@ -603,15 +603,36 @@ def test_get_s3_files_handles_exception(mocker):
|
||||
)
|
||||
|
||||
|
||||
def test_get_s3_client_with_env_credentials():
|
||||
def test_get_s3_client_default_credentials():
|
||||
with patch.dict(os.environ, {}, clear=True):
|
||||
client = get_s3_client()
|
||||
assert client is not None, "Client should be created with default credentials"
|
||||
assert client.client_config.region_name, "Client should have a region"
|
||||
|
||||
|
||||
def test_get_s3_client_invalid_credentials():
|
||||
with patch.dict(
|
||||
os.environ,
|
||||
{
|
||||
"AWS_ACCESS_KEY_ID": "test-key",
|
||||
"AWS_SECRET_ACCESS_KEY": "test-sekret", # pragma: allowlist secret
|
||||
"AWS_ACCESS_KEY_ID": "invalid-key",
|
||||
"AWS_SECRET_ACCESS_KEY": "invalid-secret", # pragma: allowlist secret
|
||||
"AWS_DEFAULT_REGION": "us-north-1",
|
||||
},
|
||||
):
|
||||
client = get_s3_client()
|
||||
assert client is not None, "Client should be created with env credentials"
|
||||
assert client._client_config.region_name == "us-north-1"
|
||||
with patch(
|
||||
"boto3.client",
|
||||
side_effect=botocore.exceptions.ClientError(
|
||||
{
|
||||
"Error": {
|
||||
"Code": "InvalidClientTokenId",
|
||||
"Message": "Invalid credentials",
|
||||
}
|
||||
},
|
||||
"HeadBucket",
|
||||
),
|
||||
):
|
||||
try:
|
||||
get_s3_client()
|
||||
assert 1 == 0, "Should raise ClientError for invalid credentials"
|
||||
except botocore.exceptions.ClientError as e:
|
||||
assert e.response["Error"]["Code"] == "InvalidClientTokenId"
|
||||
|
||||
Reference in New Issue
Block a user