def test_secure_ssm_parameter_create():
lambda_custom_resource.create(event, context)
response = ssm.get_parameter('/parameter/store/test/key')
assert (response['Parameter']) is not None
assert response['Parameter']['Name'] == '/parameter/store/test/key'
assert response['Parameter']['Value'] == 'test-value'
assert response['Parameter']['Type'] == 'SecureString'
def test_s3ConsoleDeploy_create():
s3_conn = boto3.resource('s3')
s3_conn.create_bucket(Bucket='ConsoleBucket')
s3_conn.create_bucket(Bucket='SrcBucket')
file_path = os.path.join(os.path.dirname(__file__), "console-manifest.json")
if not os.path.exists(file_path):
open(file_path, 'w')
lambda_custom_resource.create(event, context)
def test_secure_ssm_parameter_delete():
lambda_custom_resource.create(event, context)
response = ssm.get_parameter('/parameter/store/test/key')
assert (response['Parameter']) is not None
lambda_custom_resource.delete(event, context)
# testing the exception after the parameter is deleted and we try _get_ function
try:
ssm.get_parameter('/parameter/store/test/key')
raise RuntimeError(
'As this parameter has been deleted, this should fail')
except botocore.exceptions.ClientError as err:
assert err.operation_name == 'GetParameter'
def test_no_empty_string_in_prefix_list_arns():
arn_list = create(event_with_extra_comma, context)
logger.info(arn_list)
empty_string = ''
for arn in arn_list[1].get('PrefixListArns'):
logger.info(arn)
assert empty_string != arn.split('/')[1]
def test_create_empty_get_prefix_list_arns():
with pytest.raises(ValueError, match=r"STNO CFN Parameter Missing: You must"
r" provide at least one valid prefix "
r"list id."):
create(event_no_values, context)
def test_create_get_prefix_list_arns():
arn_list = create(event, context)
logger.info(arn_list)
for arn in arn_list[1].get('PrefixListArns'):
logger.info(arn)
assert arn.startswith('arn:aws:ec2')
def test_update_replace_accounts_with_org_id():
lambda_custom_resource.create(event_with_accounts, context)
lambda_custom_resource.update(event_replace_accounts_with_org_id, context)
def test_update_remove_account_as_principal_from_both():
lambda_custom_resource.create(event_with_org_arn_and_account, context)
lambda_custom_resource.update(event_remove_accounts_from_org_id, context)
def test_update_add_account_as_principal_to_org_id():
lambda_custom_resource.create(event_with_org_arn, context)
lambda_custom_resource.update(event_add_accounts_to_org_id, context)
def test_update_same_account_as_principal():
lambda_custom_resource.create(event_with_accounts, context)
lambda_custom_resource.update(event_same_accounts, context)
def test_create_delete_both_as_principal():
lambda_custom_resource.create(event_with_org_arn_and_account, context)
lambda_custom_resource.delete(event_with_org_arn_and_account, context)
def test_create_delete_org_id_as_principal():
lambda_custom_resource.create(event_with_org_arn, context)
lambda_custom_resource.delete(event_with_org_arn, context)
def test_create_delete_account_as_principal():
lambda_custom_resource.create(event_with_accounts, context)
lambda_custom_resource.delete(event_with_accounts, context)
