Implement internet gateway mocking

tests/test_ec2/test_internet_gateways.py

@@ -1,9 +1,115 @@
+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_internet_gateways():
-    pass
+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)