Allow passing security groups by ID when creating instances

This commit is contained in:
Chris St. Pierre 2014-05-07 09:16:28 -04:00
commit 745368242e
3 changed files with 24 additions and 5 deletions

View file

@ -173,11 +173,25 @@ def test_user_data_with_run_instance():
@mock_ec2
def test_run_instance_with_security_group():
def test_run_instance_with_security_group_name():
conn = boto.connect_ec2('the_key', 'the_secret')
group = conn.create_security_group('group1', "some description")
reservation = conn.run_instances('ami-1234abcd', security_groups=['group1'])
reservation = conn.run_instances('ami-1234abcd',
security_groups=['group1'])
instance = reservation.instances[0]
instance.groups[0].id.should.equal(group.id)
instance.groups[0].name.should.equal("group1")
@mock_ec2
def test_run_instance_with_security_group_id():
conn = boto.connect_ec2('the_key', 'the_secret')
group = conn.create_security_group('group1', "some description")
reservation = conn.run_instances('ami-1234abcd',
security_group_ids=[group.id])
instance = reservation.instances[0]
instance.groups[0].id.should.equal(group.id)