116 lines
3.9 KiB
Python
116 lines
3.9 KiB
Python
import re
|
|
|
|
import boto
|
|
from boto.exception import EC2ResponseError
|
|
|
|
import sure # noqa
|
|
|
|
from moto import mock_ec2
|
|
|
|
|
|
VPC_CIDR="10.0.0.0/16"
|
|
BAD_VPC="vpc-deadbeef"
|
|
BAD_IGW="igw-deadbeef"
|
|
|
|
@mock_ec2
|
|
def test_igw_create():
|
|
""" internet gateway create """
|
|
conn = boto.connect_vpc('the_key', 'the_secret')
|
|
|
|
conn.get_all_internet_gateways().should.have.length_of(0)
|
|
igw = conn.create_internet_gateway()
|
|
conn.get_all_internet_gateways().should.have.length_of(1)
|
|
igw.id.should.match(r'igw-[0-9a-f]+')
|
|
|
|
igw = conn.get_all_internet_gateways()[0]
|
|
igw.attachments.should.have.length_of(0)
|
|
|
|
@mock_ec2
|
|
def test_igw_attach():
|
|
""" internet gateway attach """
|
|
conn = boto.connect_vpc('the_key', 'the_secret')
|
|
igw = conn.create_internet_gateway()
|
|
vpc = conn.create_vpc(VPC_CIDR)
|
|
conn.attach_internet_gateway(igw.id, vpc.id)
|
|
|
|
igw = conn.get_all_internet_gateways()[0]
|
|
igw.attachments[0].vpc_id.should.be.equal(vpc.id)
|
|
|
|
@mock_ec2
|
|
def test_igw_attach_bad_vpc():
|
|
""" internet gateway fail to attach w/ bad vpc """
|
|
conn = boto.connect_vpc('the_key', 'the_secret')
|
|
igw = conn.create_internet_gateway()
|
|
conn.attach_internet_gateway.when.called_with(igw.id, BAD_VPC).should.throw(EC2ResponseError)
|
|
|
|
@mock_ec2
|
|
def test_igw_attach_twice():
|
|
""" internet gateway fail to attach twice """
|
|
conn = boto.connect_vpc('the_key', 'the_secret')
|
|
igw = conn.create_internet_gateway()
|
|
vpc1 = conn.create_vpc(VPC_CIDR)
|
|
vpc2 = conn.create_vpc(VPC_CIDR)
|
|
conn.attach_internet_gateway(igw.id, vpc1.id)
|
|
conn.attach_internet_gateway.when.called_with(igw.id, vpc2.id).should.throw(EC2ResponseError)
|
|
|
|
@mock_ec2
|
|
def test_igw_detach():
|
|
""" internet gateway detach"""
|
|
conn = boto.connect_vpc('the_key', 'the_secret')
|
|
igw = conn.create_internet_gateway()
|
|
vpc = conn.create_vpc(VPC_CIDR)
|
|
conn.attach_internet_gateway(igw.id, vpc.id)
|
|
conn.detach_internet_gateway(igw.id, vpc.id)
|
|
igw = conn.get_all_internet_gateways()[0]
|
|
igw.attachments.should.have.length_of(0)
|
|
|
|
@mock_ec2
|
|
def test_igw_detach_bad_vpc():
|
|
""" internet gateway fail to detach w/ bad vpc """
|
|
conn = boto.connect_vpc('the_key', 'the_secret')
|
|
igw = conn.create_internet_gateway()
|
|
vpc = conn.create_vpc(VPC_CIDR)
|
|
conn.attach_internet_gateway(igw.id, vpc.id)
|
|
conn.detach_internet_gateway.when.called_with(igw.id, BAD_VPC).should.throw(EC2ResponseError)
|
|
|
|
@mock_ec2
|
|
def test_igw_detach_unattached():
|
|
""" internet gateway fail to detach unattached """
|
|
conn = boto.connect_vpc('the_key', 'the_secret')
|
|
igw = conn.create_internet_gateway()
|
|
conn.detach_internet_gateway.when.called_with(igw.id, BAD_VPC).should.throw(EC2ResponseError)
|
|
|
|
@mock_ec2
|
|
def test_igw_delete():
|
|
""" internet gateway delete"""
|
|
conn = boto.connect_vpc('the_key', 'the_secret')
|
|
vpc = conn.create_vpc(VPC_CIDR)
|
|
conn.get_all_internet_gateways().should.have.length_of(0)
|
|
igw = conn.create_internet_gateway()
|
|
conn.get_all_internet_gateways().should.have.length_of(1)
|
|
conn.delete_internet_gateway(igw.id)
|
|
conn.get_all_internet_gateways().should.have.length_of(0)
|
|
|
|
@mock_ec2
|
|
def test_igw_delete_attached():
|
|
""" internet gateway fail to delete attached """
|
|
conn = boto.connect_vpc('the_key', 'the_secret')
|
|
igw = conn.create_internet_gateway()
|
|
vpc = conn.create_vpc(VPC_CIDR)
|
|
conn.attach_internet_gateway(igw.id, vpc.id)
|
|
conn.delete_internet_gateway.when.called_with(igw.id).should.throw(EC2ResponseError)
|
|
|
|
@mock_ec2
|
|
def test_igw_desribe():
|
|
""" internet gateway fetch by id """
|
|
conn = boto.connect_vpc('the_key', 'the_secret')
|
|
igw = conn.create_internet_gateway()
|
|
igw_by_search = conn.get_all_internet_gateways([igw.id])[0]
|
|
igw.id.should.equal(igw_by_search.id)
|
|
|
|
@mock_ec2
|
|
def test_igw_desribe_bad_id():
|
|
""" internet gateway fail to fetch by bad id """
|
|
conn = boto.connect_vpc('the_key', 'the_secret')
|
|
conn.get_all_internet_gateways.when.called_with([BAD_IGW]).should.throw(EC2ResponseError)
|