def test_restrict_ami_to_stage(self):
self.assertEqual(
True, ec2.is_stage_ami(self._make_fake_ami(environment='stage')))
self.assertEqual(
False, ec2.is_stage_ami(self._make_fake_ami(environment='prod')))
self.assertEqual(
False,
ec2.is_stage_ami(
self._make_fake_ami(deployment='stage', play='stage')))
def restrict_ami_to_stage(ami_id):
"""
Method to allow only AMI deployments to stage - and no other environments.
"""
ami_id = ami_id.strip()
try:
is_stage = is_stage_ami(ami_id)
except Exception as err: # pylint: disable=broad-except
traceback.print_exc()
click.secho("Error restricting AMI to stage: {0}.\nMessage: {1}".format(ami_id, err.message), fg='red')
sys.exit(1)
sys.exit(0 if is_stage else 1)
def restrict_ami_to_stage(ami_id):
"""
Method to allow only AMI deployments to stage - and no other environments.
"""
ami_id = ami_id.strip()
try:
is_stage = is_stage_ami(ami_id)
except Exception as err: # pylint: disable=broad-except
traceback.print_exc()
click.secho(
"Error restricting AMI to stage: {0}.\nMessage: {1}".format(
ami_id, err),
fg='red')
sys.exit(1)
sys.exit(0 if is_stage else 1)
def test_restrict_ami_to_stage(self):
self.assertEqual(True, ec2.is_stage_ami(self._make_fake_ami(environment='stage')))
self.assertEqual(False, ec2.is_stage_ami(self._make_fake_ami(environment='prod')))
self.assertEqual(False, ec2.is_stage_ami(self._make_fake_ami(deployment='stage', play='stage')))
