def test_generate_template_invalid_arguments(): cloudformation_simple_stack, _ = load_cloudformation_template( here( 'resources/simple_cloudformation_stack/cloudformation_invalid_arguments.py' )) context = {'foo': 'bar'} config_simple_stack = fix_deprecated_kumo_config( read_json_config( here('resources/simple_cloudformation_stack/gcdt_dev.json')) )['kumo'] with pytest.raises(Exception) as einfo: generate_template(context, config_simple_stack, cloudformation_simple_stack) assert einfo.match( r"Arguments of 'generate_template' not as expected: \['invalid_context', 'invalid_config'\]" )
def test_generate_template_no_arguments(): cloudformation_simple_stack, _ = load_cloudformation_template( here('resources/simple_cloudformation_stack/cloudformation.py')) context = {} config_simple_stack = fix_deprecated_kumo_config( read_json_config( here('resources/simple_cloudformation_stack/gcdt_dev.json')) )['kumo'] expected_template_body = read_json_config( here( 'resources/simple_cloudformation_stack/expected_template_body.json' )) template_body = json.loads( generate_template(context, config_simple_stack, cloudformation_simple_stack)) assert template_body == expected_template_body
def test_simple_cloudformation_stack(): # read the template template_path = here( 'resources/simple_cloudformation_stack/cloudformation.py') cloudformation, success = load_cloudformation_template(template_path) assert success config = { 'stack': { 'StackName': "infra-dev-kumo-sample-stack" }, 'parameters': { 'InstanceType': "t2.micro" } } expected_templ_file_name = '%s-generated-cf-template.json' % \ _get_stack_name(config) actual = write_template_to_file( config, generate_template({}, config, cloudformation)) assert actual == expected_templ_file_name
def test_s3_upload(cleanup_buckets, awsclient): #upload_conf = ConfigFactory.parse_file( # here('resources/simple_cloudformation_stack/settings_upload_dev.conf') #) upload_conf = { 'stack': { 'StackName': "infra-dev-kumo-sample-stack", 'artifactBucket': "unittest-kumo-artifact-bucket" }, 'parameters': { 'InstanceType': "t2.micro" } } region = awsclient.get_client('s3').meta.region_name account = os.getenv('ACCOUNT', None) # add account prefix to artifact bucket config if account: upload_conf['stack']['artifactBucket'] = \ '%s-unittest-kumo-artifact-bucket' % account artifact_bucket = _get_artifact_bucket(upload_conf) prepare_artifacts_bucket(awsclient, artifact_bucket) cleanup_buckets.append(artifact_bucket) dest_key = 'kumo/%s/%s-cloudformation.json' % (region, _get_stack_name(upload_conf)) expected_s3url = 'https://s3-%s.amazonaws.com/%s/%s' % (region, artifact_bucket, dest_key) cloudformation_simple_stack, _ = load_cloudformation_template( here('resources/simple_cloudformation_stack/cloudformation.py') ) actual_s3url = _s3_upload(awsclient, upload_conf, generate_template({}, upload_conf, cloudformation_simple_stack)) assert expected_s3url == actual_s3url
def test_create_update_stack_artifactbucket(awsclient, temp_cloudformation_policy, cleanup_roles, cleanup_buckets): # create a stack we use for the test lifecycle cloudformation_simple_stack, _ = load_cloudformation_template( here('resources/simple_cloudformation_stack/cloudformation.py') ) upload_conf = { 'stack': { 'StackName': "infra-dev-kumo-sample-stack", 'artifactBucket': "unittest-kumo-artifact-bucket" }, 'parameters': { 'InstanceType': "t2.micro", } } region = awsclient.get_client('s3').meta.region_name account = os.getenv('ACCOUNT', None) # add account prefix to artifact bucket config if account: upload_conf['stack']['artifactBucket'] = \ '%s-unittest-kumo-artifact-bucket' % account artifact_bucket = _get_artifact_bucket(upload_conf) prepare_artifacts_bucket(awsclient, artifact_bucket) cleanup_buckets.append(artifact_bucket) dest_key = 'kumo/%s/%s-cloudformation.json' % (region, _get_stack_name(upload_conf)) expected_s3url = 'https://s3-%s.amazonaws.com/%s/%s' % (region, artifact_bucket, dest_key) actual_s3url = _s3_upload(awsclient, upload_conf, generate_template({}, upload_conf, cloudformation_simple_stack)) assert expected_s3url == actual_s3url # create role to use for cloudformation update role = create_role_helper( awsclient, 'unittest_%s_kumo' % utils.random_string(), policies=[ temp_cloudformation_policy, 'arn:aws:iam::aws:policy/AWSCodeDeployReadOnlyAccess', 'arn:aws:iam::aws:policy/AmazonS3FullAccess' ], principal_service=['cloudformation.amazonaws.com'] ) cleanup_roles.append(role['RoleName']) # create exit_code = deploy_stack(awsclient, {}, upload_conf, cloudformation_simple_stack, override_stack_policy=False) assert exit_code == 0 stack_id = get_stack_id(awsclient, upload_conf['stack']['StackName']) wait_for_stack_create_complete(awsclient, stack_id) # update (as a change we add the RoleARN) upload_conf['stack']['RoleARN'] = role['Arn'] # update the stack changed = get_parameter_diff(awsclient, upload_conf) assert not changed exit_code = deploy_stack(awsclient, {}, upload_conf, cloudformation_simple_stack, override_stack_policy=False) assert exit_code == 0 wait_for_stack_update_complete(awsclient, stack_id) # cleanup exit_code = delete_stack(awsclient, upload_conf) assert exit_code == 0 wait_for_stack_delete_complete(awsclient, stack_id)