422 lines
16 KiB
Python
422 lines
16 KiB
Python
from __future__ import unicode_literals
|
|
|
|
import json
|
|
import sure # noqa
|
|
|
|
import moto.server as server
|
|
from moto import mock_secretsmanager
|
|
|
|
'''
|
|
Test the different server responses for secretsmanager
|
|
'''
|
|
|
|
|
|
@mock_secretsmanager
|
|
def test_get_secret_value():
|
|
|
|
backend = server.create_backend_app("secretsmanager")
|
|
test_client = backend.test_client()
|
|
|
|
create_secret = test_client.post('/',
|
|
data={"Name": "test-secret",
|
|
"SecretString": "foo-secret"},
|
|
headers={
|
|
"X-Amz-Target": "secretsmanager.CreateSecret"},
|
|
)
|
|
get_secret = test_client.post('/',
|
|
data={"SecretId": "test-secret",
|
|
"VersionStage": "AWSCURRENT"},
|
|
headers={
|
|
"X-Amz-Target": "secretsmanager.GetSecretValue"},
|
|
)
|
|
|
|
json_data = json.loads(get_secret.data.decode("utf-8"))
|
|
assert json_data['SecretString'] == 'foo-secret'
|
|
|
|
@mock_secretsmanager
|
|
def test_get_secret_that_does_not_exist():
|
|
|
|
backend = server.create_backend_app("secretsmanager")
|
|
test_client = backend.test_client()
|
|
|
|
get_secret = test_client.post('/',
|
|
data={"SecretId": "i-dont-exist",
|
|
"VersionStage": "AWSCURRENT"},
|
|
headers={
|
|
"X-Amz-Target": "secretsmanager.GetSecretValue"},
|
|
)
|
|
json_data = json.loads(get_secret.data.decode("utf-8"))
|
|
assert json_data['message'] == "Secrets Manager can't find the specified secret"
|
|
assert json_data['__type'] == 'ResourceNotFoundException'
|
|
|
|
@mock_secretsmanager
|
|
def test_get_secret_that_does_not_match():
|
|
backend = server.create_backend_app("secretsmanager")
|
|
test_client = backend.test_client()
|
|
|
|
create_secret = test_client.post('/',
|
|
data={"Name": "test-secret",
|
|
"SecretString": "foo-secret"},
|
|
headers={
|
|
"X-Amz-Target": "secretsmanager.CreateSecret"},
|
|
)
|
|
get_secret = test_client.post('/',
|
|
data={"SecretId": "i-dont-match",
|
|
"VersionStage": "AWSCURRENT"},
|
|
headers={
|
|
"X-Amz-Target": "secretsmanager.GetSecretValue"},
|
|
)
|
|
json_data = json.loads(get_secret.data.decode("utf-8"))
|
|
assert json_data['message'] == "Secrets Manager can't find the specified secret"
|
|
assert json_data['__type'] == 'ResourceNotFoundException'
|
|
|
|
@mock_secretsmanager
|
|
def test_create_secret():
|
|
|
|
backend = server.create_backend_app("secretsmanager")
|
|
test_client = backend.test_client()
|
|
|
|
res = test_client.post('/',
|
|
data={"Name": "test-secret",
|
|
"SecretString": "foo-secret"},
|
|
headers={
|
|
"X-Amz-Target": "secretsmanager.CreateSecret"},
|
|
)
|
|
|
|
json_data = json.loads(res.data.decode("utf-8"))
|
|
assert json_data['ARN'] == (
|
|
'arn:aws:secretsmanager:us-east-1:1234567890:secret:test-secret-rIjad')
|
|
assert json_data['Name'] == 'test-secret'
|
|
|
|
@mock_secretsmanager
|
|
def test_describe_secret():
|
|
|
|
backend = server.create_backend_app('secretsmanager')
|
|
test_client = backend.test_client()
|
|
|
|
create_secret = test_client.post('/',
|
|
data={"Name": "test-secret",
|
|
"SecretString": "foosecret"},
|
|
headers={
|
|
"X-Amz-Target": "secretsmanager.CreateSecret"
|
|
},
|
|
)
|
|
describe_secret = test_client.post('/',
|
|
data={"SecretId": "test-secret"},
|
|
headers={
|
|
"X-Amz-Target": "secretsmanager.DescribeSecret"
|
|
},
|
|
)
|
|
|
|
json_data = json.loads(describe_secret.data.decode("utf-8"))
|
|
assert json_data # Returned dict is not empty
|
|
assert json_data['ARN'] == (
|
|
'arn:aws:secretsmanager:us-east-1:1234567890:secret:test-secret-rIjad'
|
|
)
|
|
|
|
@mock_secretsmanager
|
|
def test_describe_secret_that_does_not_exist():
|
|
|
|
backend = server.create_backend_app('secretsmanager')
|
|
test_client = backend.test_client()
|
|
|
|
describe_secret = test_client.post('/',
|
|
data={"SecretId": "i-dont-exist"},
|
|
headers={
|
|
"X-Amz-Target": "secretsmanager.DescribeSecret"
|
|
},
|
|
)
|
|
|
|
json_data = json.loads(describe_secret.data.decode("utf-8"))
|
|
assert json_data['message'] == "Secrets Manager can't find the specified secret"
|
|
assert json_data['__type'] == 'ResourceNotFoundException'
|
|
|
|
@mock_secretsmanager
|
|
def test_describe_secret_that_does_not_match():
|
|
|
|
backend = server.create_backend_app('secretsmanager')
|
|
test_client = backend.test_client()
|
|
|
|
create_secret = test_client.post('/',
|
|
data={"Name": "test-secret",
|
|
"SecretString": "foosecret"},
|
|
headers={
|
|
"X-Amz-Target": "secretsmanager.CreateSecret"
|
|
},
|
|
)
|
|
describe_secret = test_client.post('/',
|
|
data={"SecretId": "i-dont-match"},
|
|
headers={
|
|
"X-Amz-Target": "secretsmanager.DescribeSecret"
|
|
},
|
|
)
|
|
|
|
json_data = json.loads(describe_secret.data.decode("utf-8"))
|
|
assert json_data['message'] == "Secrets Manager can't find the specified secret"
|
|
assert json_data['__type'] == 'ResourceNotFoundException'
|
|
|
|
@mock_secretsmanager
|
|
def test_rotate_secret():
|
|
backend = server.create_backend_app('secretsmanager')
|
|
test_client = backend.test_client()
|
|
|
|
create_secret = test_client.post('/',
|
|
data={"Name": "test-secret",
|
|
"SecretString": "foosecret"},
|
|
headers={
|
|
"X-Amz-Target": "secretsmanager.CreateSecret"
|
|
},
|
|
)
|
|
|
|
client_request_token = "EXAMPLE2-90ab-cdef-fedc-ba987SECRET2"
|
|
rotate_secret = test_client.post('/',
|
|
data={"SecretId": "test-secret",
|
|
"ClientRequestToken": client_request_token},
|
|
headers={
|
|
"X-Amz-Target": "secretsmanager.RotateSecret"
|
|
},
|
|
)
|
|
|
|
json_data = json.loads(rotate_secret.data.decode("utf-8"))
|
|
assert json_data # Returned dict is not empty
|
|
assert json_data['ARN'] == (
|
|
'arn:aws:secretsmanager:us-east-1:1234567890:secret:test-secret-rIjad'
|
|
)
|
|
assert json_data['Name'] == 'test-secret'
|
|
assert json_data['VersionId'] == client_request_token
|
|
|
|
# @mock_secretsmanager
|
|
# def test_rotate_secret_enable_rotation():
|
|
# backend = server.create_backend_app('secretsmanager')
|
|
# test_client = backend.test_client()
|
|
|
|
# create_secret = test_client.post(
|
|
# '/',
|
|
# data={
|
|
# "Name": "test-secret",
|
|
# "SecretString": "foosecret"
|
|
# },
|
|
# headers={
|
|
# "X-Amz-Target": "secretsmanager.CreateSecret"
|
|
# },
|
|
# )
|
|
|
|
# initial_description = test_client.post(
|
|
# '/',
|
|
# data={
|
|
# "SecretId": "test-secret"
|
|
# },
|
|
# headers={
|
|
# "X-Amz-Target": "secretsmanager.DescribeSecret"
|
|
# },
|
|
# )
|
|
|
|
# json_data = json.loads(initial_description.data.decode("utf-8"))
|
|
# assert json_data # Returned dict is not empty
|
|
# assert json_data['RotationEnabled'] is False
|
|
# assert json_data['RotationRules']['AutomaticallyAfterDays'] == 0
|
|
|
|
# rotate_secret = test_client.post(
|
|
# '/',
|
|
# data={
|
|
# "SecretId": "test-secret",
|
|
# "RotationRules": {"AutomaticallyAfterDays": 42}
|
|
# },
|
|
# headers={
|
|
# "X-Amz-Target": "secretsmanager.RotateSecret"
|
|
# },
|
|
# )
|
|
|
|
# rotated_description = test_client.post(
|
|
# '/',
|
|
# data={
|
|
# "SecretId": "test-secret"
|
|
# },
|
|
# headers={
|
|
# "X-Amz-Target": "secretsmanager.DescribeSecret"
|
|
# },
|
|
# )
|
|
|
|
# json_data = json.loads(rotated_description.data.decode("utf-8"))
|
|
# assert json_data # Returned dict is not empty
|
|
# assert json_data['RotationEnabled'] is True
|
|
# assert json_data['RotationRules']['AutomaticallyAfterDays'] == 42
|
|
|
|
@mock_secretsmanager
|
|
def test_rotate_secret_that_does_not_exist():
|
|
backend = server.create_backend_app('secretsmanager')
|
|
test_client = backend.test_client()
|
|
|
|
rotate_secret = test_client.post('/',
|
|
data={"SecretId": "i-dont-exist"},
|
|
headers={
|
|
"X-Amz-Target": "secretsmanager.RotateSecret"
|
|
},
|
|
)
|
|
|
|
json_data = json.loads(rotate_secret.data.decode("utf-8"))
|
|
assert json_data['message'] == "Secrets Manager can't find the specified secret"
|
|
assert json_data['__type'] == 'ResourceNotFoundException'
|
|
|
|
@mock_secretsmanager
|
|
def test_rotate_secret_that_does_not_match():
|
|
backend = server.create_backend_app('secretsmanager')
|
|
test_client = backend.test_client()
|
|
|
|
create_secret = test_client.post('/',
|
|
data={"Name": "test-secret",
|
|
"SecretString": "foosecret"},
|
|
headers={
|
|
"X-Amz-Target": "secretsmanager.CreateSecret"
|
|
},
|
|
)
|
|
|
|
rotate_secret = test_client.post('/',
|
|
data={"SecretId": "i-dont-match"},
|
|
headers={
|
|
"X-Amz-Target": "secretsmanager.RotateSecret"
|
|
},
|
|
)
|
|
|
|
json_data = json.loads(rotate_secret.data.decode("utf-8"))
|
|
assert json_data['message'] == "Secrets Manager can't find the specified secret"
|
|
assert json_data['__type'] == 'ResourceNotFoundException'
|
|
|
|
@mock_secretsmanager
|
|
def test_rotate_secret_client_request_token_too_short():
|
|
backend = server.create_backend_app('secretsmanager')
|
|
test_client = backend.test_client()
|
|
|
|
create_secret = test_client.post('/',
|
|
data={"Name": "test-secret",
|
|
"SecretString": "foosecret"},
|
|
headers={
|
|
"X-Amz-Target": "secretsmanager.CreateSecret"
|
|
},
|
|
)
|
|
|
|
client_request_token = "ED9F8B6C-85B7-B7E4-38F2A3BEB13C"
|
|
rotate_secret = test_client.post('/',
|
|
data={"SecretId": "test-secret",
|
|
"ClientRequestToken": client_request_token},
|
|
headers={
|
|
"X-Amz-Target": "secretsmanager.RotateSecret"
|
|
},
|
|
)
|
|
|
|
json_data = json.loads(rotate_secret.data.decode("utf-8"))
|
|
assert json_data['message'] == "ClientRequestToken must be 32-64 characters long."
|
|
assert json_data['__type'] == 'InvalidParameterException'
|
|
|
|
@mock_secretsmanager
|
|
def test_rotate_secret_client_request_token_too_long():
|
|
backend = server.create_backend_app('secretsmanager')
|
|
test_client = backend.test_client()
|
|
|
|
create_secret = test_client.post('/',
|
|
data={"Name": "test-secret",
|
|
"SecretString": "foosecret"},
|
|
headers={
|
|
"X-Amz-Target": "secretsmanager.CreateSecret"
|
|
},
|
|
)
|
|
|
|
client_request_token = (
|
|
'ED9F8B6C-85B7-446A-B7E4-38F2A3BEB13C-'
|
|
'ED9F8B6C-85B7-446A-B7E4-38F2A3BEB13C'
|
|
)
|
|
rotate_secret = test_client.post('/',
|
|
data={"SecretId": "test-secret",
|
|
"ClientRequestToken": client_request_token},
|
|
headers={
|
|
"X-Amz-Target": "secretsmanager.RotateSecret"
|
|
},
|
|
)
|
|
|
|
json_data = json.loads(rotate_secret.data.decode("utf-8"))
|
|
assert json_data['message'] == "ClientRequestToken must be 32-64 characters long."
|
|
assert json_data['__type'] == 'InvalidParameterException'
|
|
|
|
@mock_secretsmanager
|
|
def test_rotate_secret_rotation_lambda_arn_too_long():
|
|
backend = server.create_backend_app('secretsmanager')
|
|
test_client = backend.test_client()
|
|
|
|
create_secret = test_client.post('/',
|
|
data={"Name": "test-secret",
|
|
"SecretString": "foosecret"},
|
|
headers={
|
|
"X-Amz-Target": "secretsmanager.CreateSecret"
|
|
},
|
|
)
|
|
|
|
rotation_lambda_arn = '85B7-446A-B7E4' * 147 # == 2058 characters
|
|
rotate_secret = test_client.post('/',
|
|
data={"SecretId": "test-secret",
|
|
"RotationLambdaARN": rotation_lambda_arn},
|
|
headers={
|
|
"X-Amz-Target": "secretsmanager.RotateSecret"
|
|
},
|
|
)
|
|
|
|
json_data = json.loads(rotate_secret.data.decode("utf-8"))
|
|
assert json_data['message'] == "RotationLambdaARN must <= 2048 characters long."
|
|
assert json_data['__type'] == 'InvalidParameterException'
|
|
|
|
|
|
#
|
|
# The following tests should work, but fail on the embedded dict in
|
|
# RotationRules. The error message suggests a problem deeper in the code, which
|
|
# needs further investigation.
|
|
#
|
|
|
|
# @mock_secretsmanager
|
|
# def test_rotate_secret_rotation_period_zero():
|
|
# backend = server.create_backend_app('secretsmanager')
|
|
# test_client = backend.test_client()
|
|
|
|
# create_secret = test_client.post('/',
|
|
# data={"Name": "test-secret",
|
|
# "SecretString": "foosecret"},
|
|
# headers={
|
|
# "X-Amz-Target": "secretsmanager.CreateSecret"
|
|
# },
|
|
# )
|
|
|
|
# rotate_secret = test_client.post('/',
|
|
# data={"SecretId": "test-secret",
|
|
# "RotationRules": {"AutomaticallyAfterDays": 0}},
|
|
# headers={
|
|
# "X-Amz-Target": "secretsmanager.RotateSecret"
|
|
# },
|
|
# )
|
|
|
|
# json_data = json.loads(rotate_secret.data.decode("utf-8"))
|
|
# assert json_data['message'] == "RotationRules.AutomaticallyAfterDays must be within 1-1000."
|
|
# assert json_data['__type'] == 'InvalidParameterException'
|
|
|
|
# @mock_secretsmanager
|
|
# def test_rotate_secret_rotation_period_too_long():
|
|
# backend = server.create_backend_app('secretsmanager')
|
|
# test_client = backend.test_client()
|
|
|
|
# create_secret = test_client.post('/',
|
|
# data={"Name": "test-secret",
|
|
# "SecretString": "foosecret"},
|
|
# headers={
|
|
# "X-Amz-Target": "secretsmanager.CreateSecret"
|
|
# },
|
|
# )
|
|
|
|
# rotate_secret = test_client.post('/',
|
|
# data={"SecretId": "test-secret",
|
|
# "RotationRules": {"AutomaticallyAfterDays": 1001}},
|
|
# headers={
|
|
# "X-Amz-Target": "secretsmanager.RotateSecret"
|
|
# },
|
|
# )
|
|
|
|
# json_data = json.loads(rotate_secret.data.decode("utf-8"))
|
|
# assert json_data['message'] == "RotationRules.AutomaticallyAfterDays must be within 1-1000."
|
|
# assert json_data['__type'] == 'InvalidParameterException'
|