moto/tests/test_ec2/test_amis.py

import boto
from boto.exception import EC2ResponseError

from sure import expect

from moto import mock_ec2


@mock_ec2
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")