catalin/moto
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

clean up instance attribute modification and add base AMI stuff

Browse source
This commit is contained in:
Steve Pulec 2013-02-23 14:22:09 -05:00
commit 351aca3c68
6 changed files with 210 additions and 46 deletions
Download patch file Download diff file Expand all files Collapse all files

44
tests/test_ec2/test_amis.py
View file

@ -1,9 +1,49 @@
import boto
from boto.exception import EC2ResponseError
from sure import expect
from moto import mock_ec2
@mock_ec2
def test_amis():
pass
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')
# @mock_ec2
# def test_ami_attributes():
# 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")
# launch_permission = conn.get_image_attribute(image, 'description')
# expect(launch_permission.description).should.equal("this is a test ami")