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

add instance attribute description and modification

Browse source
This commit is contained in:
Steve Pulec 2013-02-19 23:01:13 -05:00
commit 1af038290d
3 changed files with 67 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

33
tests/test_ec2/test_ec2.py
View file

@ -1,5 +1,5 @@
import boto
from boto.ec2.instance import Reservation
from boto.ec2.instance import Reservation, InstanceAttribute
from sure import expect
from moto import mock_ec2
@ -44,9 +44,28 @@ def test_instance_start_and_stop():
started_instances = conn.start_instances(instances[0].id)
started_instances[0].state.should.equal('pending')
# @mock_ec2
# def test_instance_attributes():
# conn = boto.connect_ec2('the_key', 'the_secret')
# reservation = conn.run_instances('<ami-image-id>')
# instance = reservation.instances[0]
# instance_type_value = instance.get_attribute("instanceType")
@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")