mirror of
https://github.com/GSA/notifications-api.git
synced 2026-08-23 07:46:06 -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(
|
with patch.dict(
|
||||||
os.environ,
|
os.environ,
|
||||||
{
|
{
|
||||||
"AWS_ACCESS_KEY_ID": "test-key",
|
"AWS_ACCESS_KEY_ID": "invalid-key",
|
||||||
"AWS_SECRET_ACCESS_KEY": "test-sekret", # pragma: allowlist secret
|
"AWS_SECRET_ACCESS_KEY": "invalid-secret", # pragma: allowlist secret
|
||||||
"AWS_DEFAULT_REGION": "us-north-1",
|
"AWS_DEFAULT_REGION": "us-north-1",
|
||||||
},
|
},
|
||||||
):
|
):
|
||||||
client = get_s3_client()
|
with patch(
|
||||||
assert client is not None, "Client should be created with env credentials"
|
"boto3.client",
|
||||||
assert client._client_config.region_name == "us-north-1"
|
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