added support for carrier gateway (#4280)

This commit is contained in:
Macwan Nevil 2021-09-12 02:00:15 +05:30 committed by GitHub
parent 99c661781e
commit 39314906bc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
6 changed files with 173 additions and 0 deletions

View File

@ -703,3 +703,11 @@ class InvalidSubnetCidrBlockAssociationID(EC2ClientError):
association_id
),
)
class InvalidCarrierGatewayID(EC2ClientError):
def __init__(self, carrier_gateway_id):
super(InvalidCarrierGatewayID, self).__init__(
"InvalidCarrierGatewayID.NotFound",
"The CarrierGateway ID '{0}' does not exist".format(carrier_gateway_id),
)

View File

@ -116,6 +116,7 @@ from .exceptions import (
InvalidVpcEndPointIdError,
InvalidTaggableResourceType,
InvalidGatewayIDError,
InvalidCarrierGatewayID,
)
from .utils import (
EC2_RESOURCE_TO_PREFIX,
@ -160,6 +161,7 @@ from .utils import (
random_vpc_cidr_association_id,
random_vpc_peering_connection_id,
random_iam_instance_profile_association_id,
random_carrier_gateway_id,
generic_filter,
is_valid_resource_id,
get_prefix,
@ -5344,6 +5346,66 @@ class InternetGatewayBackend(object):
return self.describe_internet_gateways(internet_gateway_ids=igw_ids)[0]
class CarrierGateway(TaggedEC2Resource):
def __init__(self, ec2_backend, vpc_id, tags=None):
self.id = random_carrier_gateway_id()
self.ec2_backend = ec2_backend
self.vpc_id = vpc_id
self.state = "available"
self.add_tags(tags or {})
@property
def physical_resource_id(self):
return self.id
@property
def owner_id(self):
return ACCOUNT_ID
class CarrierGatewayBackend(object):
def __init__(self):
self.carrier_gateways = {}
super(CarrierGatewayBackend, self).__init__()
def create_carrier_gateway(self, vpc_id, tags=None):
vpc = self.get_vpc(vpc_id)
if not vpc:
raise InvalidVPCIdError(vpc_id)
carrier_gateway = CarrierGateway(self, vpc_id, tags)
self.carrier_gateways[carrier_gateway.id] = carrier_gateway
return carrier_gateway
def delete_carrier_gateway(self, id):
if not self.carrier_gateways.get(id):
raise InvalidCarrierGatewayID(id)
carrier_gateway = self.carrier_gateways.pop(id)
carrier_gateway.state = "deleted"
return carrier_gateway
def describe_carrier_gateways(self, ids=None, filters=None):
carrier_gateways = list(self.carrier_gateways.values())
if ids:
carrier_gateways = [
carrier_gateway
for carrier_gateway in carrier_gateways
if carrier_gateway.id in ids
]
attr_pairs = (
("carrier-gateway-id", "id"),
("state", "state"),
("vpc-id", "vpc_id"),
("owner-id", "owner_id"),
)
result = carrier_gateways
if filters:
result = filter_resources(carrier_gateways, filters, attr_pairs)
return result
class EgressOnlyInternetGateway(TaggedEC2Resource):
def __init__(self, ec2_backend, vpc_id, tags=None):
self.id = random_egress_only_internet_gateway_id()
@ -7962,6 +8024,7 @@ class EC2Backend(
TransitGatewayRelationsBackend,
LaunchTemplateBackend,
IamInstanceProfileAssociationBackend,
CarrierGatewayBackend,
):
def __init__(self, region_name):
self.region_name = region_name

View File

@ -39,6 +39,7 @@ from .transit_gateways import TransitGateways
from .transit_gateway_route_tables import TransitGatewayRouteTable
from .transit_gateway_attachments import TransitGatewayAttachment
from .iam_instance_profiles import IamInstanceProfiles
from .carrier_gateways import CarrierGateway
class EC2Response(
@ -81,6 +82,7 @@ class EC2Response(
TransitGatewayRouteTable,
TransitGatewayAttachment,
IamInstanceProfiles,
CarrierGateway,
):
@property
def ec2_backend(self):

View File

@ -0,0 +1,94 @@
from moto.core.responses import BaseResponse
from moto.ec2.utils import filters_from_querystring, add_tag_specification
class CarrierGateway(BaseResponse):
def create_carrier_gateway(self):
vpc_id = self._get_param("VpcId")
tags = self._get_multi_param("TagSpecification")
tags = add_tag_specification(tags)
carrier_gateway = self.ec2_backend.create_carrier_gateway(
vpc_id=vpc_id, tags=tags
)
template = self.response_template(CREATE_CARRIER_GATEWAY_RESPONSE)
return template.render(carrier_gateway=carrier_gateway)
def delete_carrier_gateway(self):
carrier_gateway_id = self._get_param("CarrierGatewayId")
carrier_gateway = self.ec2_backend.delete_carrier_gateway(carrier_gateway_id)
template = self.response_template(DELETE_CARRIER_GATEWAY_RESPONSE)
return template.render(carrier_gateway=carrier_gateway)
def describe_carrier_gateways(self):
carrier_gateway_ids = self._get_multi_param("CarrierGatewayId")
filters = filters_from_querystring(self.querystring)
carrier_gateways = self.ec2_backend.describe_carrier_gateways(
carrier_gateway_ids, filters
)
template = self.response_template(DESCRIBE_CARRIER_GATEWAYS_RESPONSE)
return template.render(carrier_gateways=carrier_gateways)
CREATE_CARRIER_GATEWAY_RESPONSE = """<CreateCarrierGatewayResponse xmlns="http://ec2.amazonaws.com/doc/2016-11-15/">
<requestId>c617595f-6c29-4a00-a941-example</requestId>
<carrierGateway>
<state>{{ carrier_gateway.state }}</state>
<vpcId>{{ carrier_gateway.vpc_id }}</vpcId>
<carrierGatewayId>{{ carrier_gateway.id }}</carrierGatewayId>
<ownerId>{{ carrier_gateway.owner_id }}</ownerId>
<tagSet>
{% for tag in carrier_gateway.get_tags() %}
<item>
<key>{{ tag.key }}</key>
<value>{{ tag.value }}</value>
</item>
{% endfor %}
</tagSet>
</carrierGateway>
</CreateCarrierGatewayResponse>
"""
DELETE_CARRIER_GATEWAY_RESPONSE = """<DeleteCarrierGatewayResponse xmlns="http://ec2.amazonaws.com/doc/2016-11-15/">
<requestId>c617595f-6c29-4a00-a941-example</requestId>
<carrierGateway>
<state>{{ carrier_gateway.state }}</state>
<vpcId>{{ carrier_gateway.vpc_id }}</vpcId>
<carrierGatewayId>{{ carrier_gateway.id }}</carrierGatewayId>
<ownerId>{{ carrier_gateway.owner_id }}</ownerId>
<tagSet>
{% for tag in carrier_gateway.get_tags() %}
<item>
<key>{{ tag.key }}</key>
<value>{{ tag.value }}</value>
</item>
{% endfor %}
</tagSet>
</carrierGateway>
</DeleteCarrierGatewayResponse>
"""
DESCRIBE_CARRIER_GATEWAYS_RESPONSE = """<DescribeCarrierGatewaysResponse xmlns="http://ec2.amazonaws.com/doc/2016-11-15/">
<requestId>151283df-f7dc-4317-89b4-01c9888b1d45</requestId>
<carrierGatewaySet>
{% for carrier_gateway in carrier_gateways %}
<item>
<state>{{ carrier_gateway.state }}</state>
<vpcId>{{ carrier_gateway.vpc_id }}</vpcId>
<carrierGatewayId>{{ carrier_gateway.id }}</carrierGatewayId>
<ownerId>{{ carrier_gateway.owner_id }}</ownerId>
<tagSet>
{% for tag in carrier_gateway.get_tags() %}
<item>
<key>{{ tag.key }}</key>
<value>{{ tag.value }}</value>
</item>
{% endfor %}
</tagSet>
</item>
{% endfor %}
</carrierGatewaySet>
</DescribeCarrierGatewaysResponse>
"""

View File

@ -52,6 +52,7 @@ EC2_RESOURCE_TO_PREFIX = {
"vpn-connection": "vpn",
"vpn-gateway": "vgw",
"iam-instance-profile-association": "iip-assoc",
"carrier-gateway": "cagw",
}
@ -220,6 +221,10 @@ def random_iam_instance_profile_association_id():
return random_id(prefix=EC2_RESOURCE_TO_PREFIX["iam-instance-profile-association"])
def random_carrier_gateway_id():
return random_id(prefix=EC2_RESOURCE_TO_PREFIX["carrier-gateway"], size=17)
def random_public_ip():
return "54.214.{0}.{1}".format(random.choice(range(255)), random.choice(range(255)))

View File

@ -123,3 +123,4 @@ TestAccAWSInternetGateway
TestAccAWSSecurityGroupRule_
TestAccAWSVpnGateway
TestAccAWSVpnGatewayAttachment
TestAccAWSEc2CarrierGateway