def test_secrets_provider_get_sdk_options_overwrite(mock_name, mock_value, config):
"""
Test SecretsProvider.get() with custom SDK options overwritten
"""
# Create a new provider
provider = parameters.SecretsProvider(config=config)
# Stub the boto3 client
stubber = stub.Stubber(provider.client)
response = {
"ARN": f"arn:aws:secretsmanager:us-east-1:132456789012:secret/{mock_name}",
"Name": mock_name,
"VersionId": "7a9155b8-2dc9-466e-b4f6-5bc46516c84d",
"SecretString": mock_value,
"CreatedDate": datetime(2015, 1, 1),
}
expected_params = {"SecretId": mock_name}
stubber.add_response("get_secret_value", response, expected_params)
stubber.activate()
try:
value = provider.get(mock_name, SecretId="THIS_SHOULD_BE_OVERWRITTEN")
assert value == mock_value
stubber.assert_no_pending_responses()
finally:
stubber.deactivate()
def test_secrets_provider_get_expired(mock_name, mock_value, config):
"""
Test SecretsProvider.get() with a cached but expired value
"""
# Create a new provider
provider = parameters.SecretsProvider(config=config)
# Inject value in the internal store
provider.store[(mock_name, None)] = ExpirableValue(mock_value, datetime.now() - timedelta(seconds=60))
# Stub the boto3 client
stubber = stub.Stubber(provider.client)
response = {
"ARN": f"arn:aws:secretsmanager:us-east-1:132456789012:secret/{mock_name}",
"Name": mock_name,
"VersionId": "7a9155b8-2dc9-466e-b4f6-5bc46516c84d",
"SecretString": mock_value,
"CreatedDate": datetime(2015, 1, 1),
}
expected_params = {"SecretId": mock_name}
stubber.add_response("get_secret_value", response, expected_params)
stubber.activate()
try:
value = provider.get(mock_name)
assert value == mock_value
stubber.assert_no_pending_responses()
finally:
stubber.deactivate()
def test_secrets_provider_get_default_config(monkeypatch, mock_name, mock_value):
"""
Test SecretsProvider.get() without specifying a config
"""
monkeypatch.setenv("AWS_DEFAULT_REGION", "us-east-1")
# Create a new provider
provider = parameters.SecretsProvider()
# Stub the boto3 client
stubber = stub.Stubber(provider.client)
response = {
"ARN": f"arn:aws:secretsmanager:us-east-1:132456789012:secret/{mock_name}",
"Name": mock_name,
"VersionId": "7a9155b8-2dc9-466e-b4f6-5bc46516c84d",
"SecretString": mock_value,
"CreatedDate": datetime(2015, 1, 1),
}
expected_params = {"SecretId": mock_name}
stubber.add_response("get_secret_value", response, expected_params)
stubber.activate()
try:
value = provider.get(mock_name)
assert value == mock_value
stubber.assert_no_pending_responses()
finally:
stubber.deactivate()
def test_secrets_provider_get_cached(mock_name, mock_value, config):
"""
Test SecretsProvider.get() with a cached value
"""
# Create a new provider
provider = parameters.SecretsProvider(config=config)
# Inject value in the internal store
provider.store[(mock_name, None)] = ExpirableValue(mock_value, datetime.now() + timedelta(seconds=60))
# Stub the boto3 client
stubber = stub.Stubber(provider.client)
stubber.activate()
try:
value = provider.get(mock_name)
assert value == mock_value
stubber.assert_no_pending_responses()
finally:
stubber.deactivate()