Yasuke/moto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

EC2 - Add tests for egress_only_internet_gateway (#4206)

Browse Source
This commit is contained in:
Bert Blommers 2021-08-22 10:35:04 +01:00 committed by GitHub
parent 7acdfe4feb
commit fbd93efdb7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 101 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

101
tests/test_ec2/test_egress_only_igw.py Normal file
View File

@ -0,0 +1,101 @@
import boto3
import pytest
import sure # noqa
from botocore.exceptions import ClientError
from moto import mock_ec2
@mock_ec2
def test_create():
ec2 = boto3.resource("ec2", region_name="us-west-1")
client = boto3.client("ec2", region_name="us-west-1")
vpc = list(ec2.vpcs.all())[0]
res = client.create_egress_only_internet_gateway(VpcId=vpc.id)
gateway = res["EgressOnlyInternetGateway"]
gateway.should.have.key("EgressOnlyInternetGatewayId").match("eigw-[a-z0-9]+")
gateway.should.have.key("Tags").equal([])
gateway.should.have.key("Attachments")
gateway["Attachments"].should.have.length_of(1)
gateway["Attachments"][0].should.equal({"State": "attached", "VpcId": vpc.id})
@mock_ec2
def test_create_with_unknown_vpc():
client = boto3.client("ec2", region_name="us-west-1")
with pytest.raises(ClientError) as ex:
client.create_egress_only_internet_gateway(VpcId="vpc-says-what")
err = ex.value.response["Error"]
err["Code"].should.equal("InvalidVpcID.NotFound")
err["Message"].should.equal("VpcID vpc-says-what does not exist.")
@mock_ec2
def test_describe_all():
ec2 = boto3.resource("ec2", region_name="us-west-1")
client = boto3.client("ec2", region_name="us-west-1")
vpc = list(ec2.vpcs.all())[0]
gw1 = client.create_egress_only_internet_gateway(VpcId=vpc.id)[
"EgressOnlyInternetGateway"
]
gw2 = client.create_egress_only_internet_gateway(VpcId=vpc.id)[
"EgressOnlyInternetGateway"
]
gateways = client.describe_egress_only_internet_gateways()[
"EgressOnlyInternetGateways"
]
gateways.should.have.length_of(2)
gateways.should.contain(gw1)
gateways.should.contain(gw2)
@mock_ec2
def test_describe_one():
ec2 = boto3.resource("ec2", region_name="us-west-1")
client = boto3.client("ec2", region_name="us-west-1")
vpc = list(ec2.vpcs.all())[0]
gw1 = client.create_egress_only_internet_gateway(VpcId=vpc.id)[
"EgressOnlyInternetGateway"
]
gw1_id = gw1["EgressOnlyInternetGatewayId"]
client.create_egress_only_internet_gateway(VpcId=vpc.id)[
"EgressOnlyInternetGateway"
]
gw3 = client.create_egress_only_internet_gateway(VpcId=vpc.id)[
"EgressOnlyInternetGateway"
]
gw3_id = gw3["EgressOnlyInternetGatewayId"]
gateways = client.describe_egress_only_internet_gateways(
EgressOnlyInternetGatewayIds=[gw1_id, gw3_id]
)["EgressOnlyInternetGateways"]
gateways.should.have.length_of(2)
gateways.should.contain(gw1)
gateways.should.contain(gw3)
@mock_ec2
def test_create_and_delete():
ec2 = boto3.resource("ec2", region_name="us-west-1")
client = boto3.client("ec2", region_name="us-west-1")
vpc = list(ec2.vpcs.all())[0]
gw1 = client.create_egress_only_internet_gateway(VpcId=vpc.id)[
"EgressOnlyInternetGateway"
]
gw1_id = gw1["EgressOnlyInternetGatewayId"]
client.describe_egress_only_internet_gateways()[
"EgressOnlyInternetGateways"
].should.have.length_of(1)
client.delete_egress_only_internet_gateway(EgressOnlyInternetGatewayId=gw1_id)
client.describe_egress_only_internet_gateways()[
"EgressOnlyInternetGateways"
].should.have.length_of(0)