increase code coverage

This commit is contained in:
Kenneth Kehl
2025-06-27 10:37:23 -07:00
parent fe81e6dfce
commit b2d98f7bd3

View File

@@ -607,32 +607,55 @@ def test_get_s3_client_default_credentials():
with patch.dict(os.environ, {}, clear=True): with patch.dict(os.environ, {}, clear=True):
client = get_s3_client() client = get_s3_client()
assert client is not None, "Client should be created with default credentials" assert client is not None, "Client should be created with default credentials"
assert client.client_config.region_name, "Client should have a region" assert client._client_config.region_name, "Client should have a region"
def test_get_s3_client_invalid_credentials(): # def test_get_s3_client_invalid_credentials():
with patch.dict( # with patch.dict(
# os.environ,
# {
# "AWS_ACCESS_KEY_ID": "invalid-key",
# "AWS_SECRET_ACCESS_KEY": "invalid-secret", # pragma: allowlist secret
# "AWS_DEFAULT_REGION": "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"
def test_get_s3_client_no_region():
with patch.ditc(
os.environ, os.environ,
{ {
"AWS_ACCESS_KEY_ID": "invalid-key", "AWS_ACCESS_KEY_ID": "test-key",
"AWS_SECRET_ACCESS_KEY": "invalid-secret", # pragma: allowlist secret "AWS_SECRET_ACCESS_KEY": "test-secret", # pragma: allowlist secret
"AWS_DEFAULT_REGION": "us-north-1",
}, },
clear=True,
): ):
with patch( with patch("boto3.client") as mock_client:
"boto3.client", mock_client.return_value = MagicMock(
side_effect=botocore.exceptions.ClientError( _client_config=MagicMock(region_name="us-north-1")
{ )
"Error": { client = get_s3_client()
"Code": "InvalidClientTokenId", assert client is not None
"Message": "Invalid credentials", mock_client.assert_called_with(
} "s3",
}, aws_access_key_id="test-key",
"HeadBucket", aws_secret_access_key="test-secret", # pragma: allowlist secret
), region_name="us-north-1",
): )
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"