2013-02-18 21:09:40 +00:00
|
|
|
import boto
|
2013-02-20 04:01:13 +00:00
|
|
|
from boto.ec2.instance import Reservation, InstanceAttribute
|
2013-02-18 21:09:40 +00:00
|
|
|
from sure import expect
|
|
|
|
|
|
|
|
|
|
from moto import mock_ec2
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@mock_ec2
|
|
|
|
|
def test_instance_launch_and_terminate():
|
|
|
|
|
conn = boto.connect_ec2('the_key', 'the_secret')
|
|
|
|
|
reservation = conn.run_instances('<ami-image-id>')
|
|
|
|
|
reservation.should.be.a(Reservation)
|
|
|
|
|
reservation.instances.should.have.length_of(1)
|
|
|
|
|
instance = reservation.instances[0]
|
|
|
|
|
|
|
|
|
|
reservations = conn.get_all_instances()
|
|
|
|
|
reservations.should.have.length_of(1)
|
|
|
|
|
reservations[0].id.should.equal(reservation.id)
|
|
|
|
|
instances = reservations[0].instances
|
|
|
|
|
instances.should.have.length_of(1)
|
|
|
|
|
instances[0].id.should.equal(instance.id)
|
|
|
|
|
instances[0].state.should.equal('pending')
|
|
|
|
|
|
|
|
|
|
conn.terminate_instances(instances[0].id)
|
|
|
|
|
|
|
|
|
|
reservations = conn.get_all_instances()
|
|
|
|
|
instance = reservations[0].instances[0]
|
|
|
|
|
instance.state.should.equal('shutting-down')
|
2013-02-19 02:56:22 +00:00
|
|
|
|
|
|
|
|
|
|
|
|
|
@mock_ec2
|
|
|
|
|
def test_instance_start_and_stop():
|
|
|
|
|
conn = boto.connect_ec2('the_key', 'the_secret')
|
2013-02-19 04:06:23 +00:00
|
|
|
reservation = conn.run_instances('<ami-image-id>', min_count=2)
|
2013-02-19 02:56:22 +00:00
|
|
|
instances = reservation.instances
|
2013-02-19 04:06:23 +00:00
|
|
|
instances.should.have.length_of(2)
|
2013-02-19 02:56:22 +00:00
|
|
|
|
|
|
|
|
stopped_instances = conn.stop_instances([instance.id for instance in instances])
|
|
|
|
|
|
|
|
|
|
for instance in stopped_instances:
|
|
|
|
|
instance.state.should.equal('stopping')
|
|
|
|
|
|
|
|
|
|
started_instances = conn.start_instances(instances[0].id)
|
|
|
|
|
started_instances[0].state.should.equal('pending')
|
2013-02-19 04:06:23 +00:00
|
|
|
|
2013-02-20 04:01:13 +00:00
|
|
|
|
2013-02-20 04:55:01 +00:00
|
|
|
@mock_ec2
|
|
|
|
|
def test_instance_reboot():
|
|
|
|
|
conn = boto.connect_ec2('the_key', 'the_secret')
|
|
|
|
|
reservation = conn.run_instances('<ami-image-id>')
|
|
|
|
|
instance = reservation.instances[0]
|
|
|
|
|
instance.reboot()
|
|
|
|
|
instance.state.should.equal('pending')
|
|
|
|
|
|
|
|
|
|
|
2013-02-20 04:01:13 +00:00
|
|
|
@mock_ec2
|
|
|
|
|
def test_instance_attribute_instance_type():
|
|
|
|
|
conn = boto.connect_ec2('the_key', 'the_secret')
|
|
|
|
|
reservation = conn.run_instances('<ami-image-id>')
|
|
|
|
|
instance = reservation.instances[0]
|
|
|
|
|
|
|
|
|
|
instance.modify_attribute("instanceType", "m1.small")
|
|
|
|
|
|
|
|
|
|
instance_attribute = instance.get_attribute("instanceType")
|
|
|
|
|
instance_attribute.should.be.a(InstanceAttribute)
|
|
|
|
|
instance_attribute.get('instanceType').should.equal("m1.small")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@mock_ec2
|
|
|
|
|
def test_instance_attribute_user_data():
|
|
|
|
|
conn = boto.connect_ec2('the_key', 'the_secret')
|
|
|
|
|
reservation = conn.run_instances('<ami-image-id>')
|
|
|
|
|
instance = reservation.instances[0]
|
|
|
|
|
|
|
|
|
|
instance.modify_attribute("userData", "this is my user data")
|
|
|
|
|
|
|
|
|
|
instance_attribute = instance.get_attribute("userData")
|
|
|
|
|
instance_attribute.should.be.a(InstanceAttribute)
|
|
|
|
|
instance_attribute.get("userData").should.equal("this is my user data")
|
