def test_create_on_update(self):
stack_name = 'create-on-update-stack'
with open('tests/fixtures/create_stack_template.yaml') as tpl_file:
cf.create_stack(self.config['cf_conn'], stack_name, tpl_file,
self.config, update=True, create_on_update=True)
stack = self.config['cf_conn'].describe_stacks(stack_name)[0]
self.assertEqual('b08c2e9d7003f62ba8ffe5c985c50a63', stack.tags['MD5Sum'])
def test_create_stack_no_stack_name(self):
stack_name = None
with open('tests/fixtures/no_metadata_template.yaml') as tpl_file:
with self.assertRaises(SystemExit) as err:
cf.create_stack(self.config['cf_conn'], stack_name, tpl_file,
self.config)
self.assertEqual(err.exception.code, 1)
def test_update_stack(self):
stack_name = None
with open('tests/fixtures/create_stack_template.yaml') as tpl_file:
cf.create_stack(self.config['cf_conn'], stack_name, tpl_file,
self.config, update=True)
stack = self.config['cf_conn'].describe_stacks('unittest-infra')[0]
self.assertEqual('b08c2e9d7003f62ba8ffe5c985c50a63', stack.tags['MD5Sum'])
def test_create_stack_no_metadata(self):
stack_name = 'my-stack'
with open('tests/fixtures/no_metadata_template.yaml') as tpl_file:
cf.create_stack(self.config['cf_conn'], stack_name, tpl_file, self.config)
stack = self.config['cf_conn'].describe_stacks('my-stack')[0]
self.assertEqual('my-stack', stack.stack_name)
self.assertEqual(self.config['env'], stack.tags['Env'])
self.assertEqual('b08c2e9d7003f62ba8ffe5c985c50a63', stack.tags['MD5Sum'])
def test_create_stack_no_metadata(self):
stack_name = 'my-stack'
with open('tests/fixtures/no_metadata_template.yaml') as tpl_file:
cf.create_stack(self.config['cf_conn'], stack_name, tpl_file,
self.config)
stack = self.config['cf_conn'].describe_stacks('my-stack')[0]
self.assertEqual('my-stack', stack.stack_name)
self.assertEqual(self.config['env'], stack.tags['Env'])
self.assertEqual('b08c2e9d7003f62ba8ffe5c985c50a63',
stack.tags['MD5Sum'])
def test_update_stack(self):
stack_name = None
with open('tests/fixtures/create_stack_template.yaml') as tpl_file:
cf.create_stack(self.config['cf_conn'],
stack_name,
tpl_file,
self.config,
update=True)
stack = self.config['cf_conn'].describe_stacks('unittest-infra')[0]
self.assertEqual('b08c2e9d7003f62ba8ffe5c985c50a63',
stack.tags['MD5Sum'])
def test_create_on_update(self):
stack_name = 'create-on-update-stack'
with open('tests/fixtures/create_stack_template.yaml') as tpl_file:
cf.create_stack(self.config['cf_conn'],
stack_name,
tpl_file,
self.config,
update=True,
create_on_update=True)
stack = self.config['cf_conn'].describe_stacks(stack_name)[0]
self.assertEqual('b08c2e9d7003f62ba8ffe5c985c50a63',
stack.tags['MD5Sum'])
def test_create_stack_no_metadata(self):
stack_name = 'my-stack'
with open('tests/fixtures/no_metadata_template.yaml') as tpl_file:
cf.create_stack(self.config['cf_conn'], stack_name, tpl_file,
self.config)
stack = self.config['cf_conn'].describe_stacks(
StackName='my-stack')['Stacks'][0]
stack_tags = {tag['Key']: tag['Value'] for tag in stack['Tags']}
self.assertEqual('my-stack', stack['StackName'])
self.assertEqual(self.config['env'], stack_tags['Env'])
self.assertEqual('b08c2e9d7003f62ba8ffe5c985c50a63',
stack_tags['MD5Sum'])
def test_update_stack(self):
stack_name = None
with open('tests/fixtures/create_stack_template.yaml') as tpl_file:
cf.create_stack(self.config['cf_conn'],
stack_name,
tpl_file,
self.config,
update=True)
stack = self.config['cf_conn'].describe_stacks(
StackName='unittest-infra')['Stacks'][0]
stack_tags = {tag['Key']: tag['Value'] for tag in stack['Tags']}
self.assertEqual('b08c2e9d7003f62ba8ffe5c985c50a63',
stack_tags['MD5Sum'])
def test_list_stacks(self):
first_stack = 'stack-to-be-listed'
with open('tests/fixtures/create_stack_template.yaml') as tpl_file:
cf.create_stack(self.config['cf_conn'], first_stack, tpl_file,
self.config)
second_stack = 'another-stack-to-be-listed'
with open('tests/fixtures/create_stack_template.yaml') as tpl_file:
cf.create_stack(self.config['cf_conn'], second_stack, tpl_file,
self.config)
listed_stacks = cf.list_stacks(self.config['cf_conn'],
name_filter='*-to-be-listed')
print(listed_stacks)
self.assertEqual(len(listed_stacks.split('\n')), 2)
def test_create_stack_no_stack_name(self):
stack_name = None
with open('tests/fixtures/no_metadata_template.yaml') as tpl_file:
with self.assertRaises(SystemExit) as err:
cf.create_stack(self.config['cf_conn'], stack_name, tpl_file, self.config)
self.assertEqual(err.exception.code, 1)
def main():
for sig in [signal.SIGTERM, signal.SIGINT, signal.SIGHUP, signal.SIGQUIT]:
signal.signal(sig, handler)
parser, args = cli.parse_options()
if not args.subcommand:
parser.print_help()
sys.exit(0)
config_file = vars(args).get('config', None)
config_dir = vars(args).get('config_dir', None)
env = vars(args).get('env', None)
config = config_load(env, config_file, config_dir)
now = datetime.now(tz=pytz.UTC)
if args.subcommand == 'config':
print_config(config,
args.property_name,
output_format=args.output_format)
sys.exit(0)
config['get_ami_id'] = aws.get_ami_id
config['get_vpc_id'] = aws.get_vpc_id
config['get_zone_id'] = aws.get_zone_id
config['get_stack_output'] = aws.get_stack_output
config['get_stack_resource'] = aws.get_stack_resource
# Figure out profile value in the following order
# - cli arg
# - env variable
# - default profile if exists
if args.profile:
profile = args.profile
elif os.environ.get('AWS_DEFAULT_PROFILE'):
profile = os.environ.get('AWS_DEFAULT_PROFILE')
elif profile_exists('default'):
profile = 'default'
elif args.profile and not profile_exists(args.profile):
print('Profile "{}" does not exist.'.format(args.profile))
sys.exit(1)
else:
profile = None
# Figure out region value in the following order
# - cli arg
# - env variable
# - region from config
if args.region:
region = args.region
elif os.environ.get('AWS_DEFAULT_REGION'):
region = os.environ.get('AWS_DEFAULT_REGION')
elif get_region_name(profile):
region = get_region_name(profile)
else:
region = get_default_region_name()
if not region:
print('Region is not specified.')
sys.exit(1)
config['region'] = region
# Not great, but try to catch everything. Above should be refactored in a
# function which handles setting up connections to different aws services
try:
ec2_conn = boto.ec2.connect_to_region(region, profile_name=profile)
vpc_conn = boto.vpc.connect_to_region(region, profile_name=profile)
cf_conn = boto.cloudformation.connect_to_region(region,
profile_name=profile)
r53_conn = boto.route53.connect_to_region(region, profile_name=profile)
s3_conn = boto.s3.connect_to_region(region, profile_name=profile)
config['ec2_conn'] = ec2_conn
config['vpc_conn'] = vpc_conn
config['cf_conn'] = cf_conn
config['r53_conn'] = r53_conn
config['s3_conn'] = s3_conn
# TODO(alekna): Fix too broad exception
except:
print(sys.exc_info()[1])
sys.exit(1)
if args.subcommand == 'resources':
output = cf.stack_resources(cf_conn, args.name, args.logical_id)
if output:
print(output)
cf_conn.close()
if args.subcommand == 'outputs':
output = cf.stack_outputs(cf_conn, args.name, args.output_name)
if output:
print(output)
cf_conn.close()
if args.subcommand == 'list':
output = cf.list_stacks(cf_conn, args.name, args.verbose)
if output:
print(output)
cf_conn.close()
if args.subcommand == 'create' or args.subcommand == 'update':
if args.property:
properties = validate_properties(args.property)
config.update(properties)
if args.subcommand == 'create':
stack_name = cf.create_stack(cf_conn,
args.name,
args.template,
config,
dry=args.dry_run)
if args.events_follow and not args.dry_run:
stack_status = cf.print_events(cf_conn, stack_name,
args.events_follow)
if stack_status in FAILED_STACK_STATES + ROLLBACK_STACK_STATES:
sys.exit(1)
else:
stack_name = cf.create_stack(
cf_conn,
args.name,
args.template,
config,
update=True,
dry=args.dry_run,
create_on_update=args.create_on_update)
if args.events_follow and not args.dry_run:
stack_status = cf.print_events(cf_conn,
stack_name,
args.events_follow,
from_dt=now)
if stack_status in FAILED_STACK_STATES + ROLLBACK_STACK_STATES:
sys.exit(1)
if args.subcommand == 'delete':
cf.delete_stack(cf_conn, args.name, region, profile, args.yes)
if args.events_follow:
stack_status = cf.print_events(cf_conn,
args.name,
args.events_follow,
from_dt=now)
if stack_status in FAILED_STACK_STATES:
sys.exit(1)
if args.subcommand == 'events':
cf.print_events(cf_conn, args.name, args.events_follow, args.lines)
if args.subcommand == 'diff':
if args.property:
properties = validate_properties(args.property)
config.update(properties)
cf.print_stack_diff(cf_conn, args.name, args.template, config)
def main():
for sig in [signal.SIGTERM, signal.SIGINT, signal.SIGHUP, signal.SIGQUIT]:
signal.signal(sig, handler)
parser, args = cli.parse_options()
if not args.subcommand:
parser.print_help()
sys.exit(0)
config_file = vars(args).get('config', None)
config_dir = vars(args).get('config_dir', None)
env = vars(args).get('env', None)
config = config_load(env, config_file, config_dir)
now = datetime.now(tz=pytz.UTC)
if args.subcommand == 'config':
print_config(config, args.property_name, output_format=args.output_format)
sys.exit(0)
config['get_ami_id'] = aws.get_ami_id
config['get_vpc_id'] = aws.get_vpc_id
config['get_zone_id'] = aws.get_zone_id
config['get_stack_output'] = aws.get_stack_output
config['get_stack_resource'] = aws.get_stack_resource
# Figure out profile value in the following order
# - cli arg
# - env variable
# - default profile if exists
if args.profile:
profile = args.profile
elif os.environ.get('AWS_DEFAULT_PROFILE'):
profile = os.environ.get('AWS_DEFAULT_PROFILE')
elif profile_exists('default'):
profile = 'default'
elif args.profile and not profile_exists(args.profile):
print('Profile "{}" does not exist.'.format(args.profile))
sys.exit(1)
else:
profile = None
# Figure out region value in the following order
# - cli arg
# - env variable
# - region from config
if args.region:
region = args.region
elif os.environ.get('AWS_DEFAULT_REGION'):
region = os.environ.get('AWS_DEFAULT_REGION')
elif get_region_name(profile):
region = get_region_name(profile)
else:
region = get_default_region_name()
if not region:
print('Region is not specified.')
sys.exit(1)
config['region'] = region
# Not great, but try to catch everything. Above should be refactored in a
# function which handles setting up connections to different aws services
try:
ec2_conn = boto.ec2.connect_to_region(region, profile_name=profile)
vpc_conn = boto.vpc.connect_to_region(region, profile_name=profile)
cf_conn = boto.cloudformation.connect_to_region(region, profile_name=profile)
r53_conn = boto.route53.connect_to_region(region, profile_name=profile)
s3_conn = boto.s3.connect_to_region(region, profile_name=profile)
config['ec2_conn'] = ec2_conn
config['vpc_conn'] = vpc_conn
config['cf_conn'] = cf_conn
config['r53_conn'] = r53_conn
config['s3_conn'] = s3_conn
# TODO(alekna): Fix too broad exception
except:
print(sys.exc_info()[1])
sys.exit(1)
if args.subcommand == 'resources':
output = cf.stack_resources(cf_conn, args.name, args.logical_id)
if output:
print(output)
cf_conn.close()
if args.subcommand == 'outputs':
output = cf.stack_outputs(cf_conn, args.name, args.output_name)
if output:
print(output)
cf_conn.close()
if args.subcommand == 'list':
output = cf.list_stacks(cf_conn, args.name, args.verbose)
if output:
print(output)
cf_conn.close()
if args.subcommand == 'create' or args.subcommand == 'update':
if args.property:
properties = validate_properties(args.property)
config.update(properties)
if args.subcommand == 'create':
stack_name = cf.create_stack(cf_conn, args.name, args.template, config, dry=args.dry_run)
if args.events_follow and not args.dry_run:
stack_status = cf.print_events(cf_conn, stack_name, args.events_follow)
if stack_status in FAILED_STACK_STATES + ROLLBACK_STACK_STATES:
sys.exit(1)
else:
stack_name = cf.create_stack(cf_conn, args.name, args.template, config, update=True, dry=args.dry_run,
create_on_update=args.create_on_update)
if args.events_follow and not args.dry_run:
stack_status = cf.print_events(cf_conn, stack_name, args.events_follow, from_dt=now)
if stack_status in FAILED_STACK_STATES + ROLLBACK_STACK_STATES:
sys.exit(1)
if args.subcommand == 'delete':
cf.delete_stack(cf_conn, args.name, region, profile, args.yes)
if args.events_follow:
stack_status = cf.print_events(cf_conn, args.name, args.events_follow, from_dt=now)
if stack_status in FAILED_STACK_STATES:
sys.exit(1)
if args.subcommand == 'events':
cf.print_events(cf_conn, args.name, args.events_follow, args.lines)
if args.subcommand == 'diff':
if args.property:
properties = validate_properties(args.property)
config.update(properties)
cf.print_stack_diff(cf_conn, args.name, args.template, config)
def main():
for sig in [signal.SIGTERM, signal.SIGINT, signal.SIGHUP, signal.SIGQUIT]:
signal.signal(sig, handler)
parser, args = cli.parse_options()
if not args.subcommand:
parser.print_help()
sys.exit(0)
config_file = vars(args).get('config', None)
config_dir = vars(args).get('config_dir', None)
env = vars(args).get('env', None)
config = config_load(env, config_file, config_dir)
if args.subcommand == 'config':
print_config(config,
args.property_name,
output_format=args.output_format)
sys.exit(0)
config['get_ami_id'] = aws.get_ami_id
config['get_vpc_id'] = aws.get_vpc_id
config['get_zone_id'] = aws.get_zone_id
config['get_stack_output'] = aws.get_stack_output
config['get_stack_resource'] = aws.get_stack_resource
session_kwargs = {}
if args.profile:
session_kwargs['profile_name'] = args.profile
if args.region:
session_kwargs['region_name'] = args.region
try:
botosession = boto3.Session(**session_kwargs)
config['region'] = botosession.region_name
s3_conn = botosession.client('s3')
ec2_conn = botosession.resource('ec2')
vpc_conn = ec2_conn
r53_conn = botosession.client('route53')
cf_conn = botosession.client('cloudformation')
config['ec2_conn'] = ec2_conn
config['vpc_conn'] = vpc_conn
config['cf_conn'] = cf_conn
config['r53_conn'] = r53_conn
config['s3_conn'] = s3_conn
except botocore.exceptions.ClientError as e:
print(e)
sys.exit(1)
if args.subcommand == 'resources':
output = cf.stack_resources(cf_conn, args.name, args.logical_id)
if output:
print(output)
if args.subcommand == 'outputs':
output = cf.stack_outputs(cf_conn, args.name, args.output_name)
if output:
print(output)
if args.subcommand == 'list':
output = cf.list_stacks(cf_conn, args.name, args.verbose)
if output:
print(output)
if args.subcommand == 'create' or args.subcommand == 'update':
if args.property:
properties = validate_properties(args.property)
config.update(properties)
if args.subcommand == 'create':
cf.create_stack(cf_conn,
args.name,
args.template,
config,
dry=args.dry_run,
follow=args.events_follow)
else:
cf.create_stack(cf_conn,
args.name,
args.template,
config,
update=True,
dry=args.dry_run,
follow=args.events_follow,
create_on_update=args.create_on_update)
if args.subcommand == 'delete':
cf.delete_stack(cf_conn, args.name, botosession.region_name,
botosession.profile_name, args.yes)
if args.events_follow:
cf.get_events(cf_conn, args.name, args.events_follow, 10)
if args.subcommand == 'events':
cf.get_events(cf_conn, args.name, args.events_follow, args.lines)
