2013-02-22 04:13:01 +00:00
|
|
|
import boto
|
2013-02-23 19:22:09 +00:00
|
|
|
from boto.exception import EC2ResponseError
|
|
|
|
|
|
2013-02-22 04:13:01 +00:00
|
|
|
from sure import expect
|
|
|
|
|
|
|
|
|
|
from moto import mock_ec2
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@mock_ec2
|
2013-02-23 19:22:09 +00:00
|
|
|
def test_ami_create_and_delete():
|
|
|
|
|
conn = boto.connect_ec2('the_key', 'the_secret')
|
|
|
|
|
reservation = conn.run_instances('<ami-image-id>')
|
|
|
|
|
instance = reservation.instances[0]
|
|
|
|
|
image = instance.create_image("test-ami", "this is a test ami")
|
|
|
|
|
|
|
|
|
|
all_images = conn.get_all_images()
|
|
|
|
|
all_images[0].id.should.equal(image)
|
|
|
|
|
|
|
|
|
|
success = conn.deregister_image(image)
|
|
|
|
|
success.should.be.true
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@mock_ec2
|
|
|
|
|
def test_ami_create_from_missing_instance():
|
|
|
|
|
conn = boto.connect_ec2('the_key', 'the_secret')
|
|
|
|
|
conn.create_image.when.called_with("i-abcdefg", "test-ami", "this is a test ami").should.throw(EC2ResponseError)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@mock_ec2
|
|
|
|
|
def test_ami_pulls_attributes_from_instance():
|
|
|
|
|
conn = boto.connect_ec2('the_key', 'the_secret')
|
|
|
|
|
reservation = conn.run_instances('<ami-image-id>')
|
|
|
|
|
instance = reservation.instances[0]
|
|
|
|
|
instance.modify_attribute("kernel", "test-kernel")
|
|
|
|
|
|
|
|
|
|
image_id = instance.create_image("test-ami", "this is a test ami")
|
|
|
|
|
image = conn.get_image(image_id)
|
|
|
|
|
image.kernel_id.should.equal('test-kernel')
|
