def test_edp_for_untagged_ami(self):
ec2_connection = boto.connect_ec2()
reservation = ec2_connection.run_instances(random_ami_id())
instance_id = reservation.instances[0].id
ami_id = ec2_connection.create_image(instance_id, "Existing AMI")
# AMI Exists but isn't tagged.
self.assertRaises(MissingTagException, ec2.edp_for_ami, ami_id)
def test_edp_for_untagged_ami(self):
ec2_connection = boto.connect_ec2()
reservation = ec2_connection.run_instances(random_ami_id())
instance_id = reservation.instances[0].id
ami_id = ec2_connection.create_image(instance_id, "Existing AMI")
# AMI Exists but isn't tagged.
self.assertRaises(MissingTagException, ec2.edp_for_ami, ami_id)
def _make_fake_ami(self, environment='foo', deployment='bar', play='baz'):
"""
Method to make a fake AMI.
"""
ec2_connection = boto.connect_ec2()
reservation = ec2_connection.run_instances(random_ami_id())
instance_id = reservation.instances[0].id
ami_id = ec2_connection.create_image(instance_id, "Existing AMI")
ami = ec2_connection.get_all_images(ami_id)[0]
ami.add_tag("environment", environment)
ami.add_tag("deployment", deployment)
ami.add_tag("play", play)
return ami_id
def _make_fake_ami(self, environment='foo', deployment='bar', play='baz'):
"""
Method to make a fake AMI.
"""
ec2_connection = boto.connect_ec2()
reservation = ec2_connection.run_instances(random_ami_id())
instance_id = reservation.instances[0].id
ami_id = ec2_connection.create_image(instance_id, "Existing AMI")
ami = ec2_connection.get_all_images(ami_id)[0]
ami.add_tag("environment", environment)
ami.add_tag("deployment", deployment)
ami.add_tag("play", play)
return ami_id
