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

Merge pull request #2922 from johnwc/enhancement/apigateway-policy

Browse Source 
Add policy to API Gateway
This commit is contained in:
Jack Danger 2020-04-25 12:51:10 -07:00 committed by GitHub
commit bd650d9b48
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 22 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
moto/apigateway/models.py
View File

@ -461,6 +461,7 @@ class RestAPI(BaseModel):
self.description = description self.description = description
self.create_date = int(time.time()) self.create_date = int(time.time())
self.api_key_source = kwargs.get("api_key_source") or "HEADER" self.api_key_source = kwargs.get("api_key_source") or "HEADER"
self.policy = kwargs.get("policy") or None
self.endpoint_configuration = kwargs.get("endpoint_configuration") or { self.endpoint_configuration = kwargs.get("endpoint_configuration") or {
"types": ["EDGE"] "types": ["EDGE"]
} }
@ -485,6 +486,7 @@ class RestAPI(BaseModel):
"apiKeySource": self.api_key_source, "apiKeySource": self.api_key_source,
"endpointConfiguration": self.endpoint_configuration, "endpointConfiguration": self.endpoint_configuration,
"tags": self.tags, "tags": self.tags,
"policy": self.policy,
} }
def add_child(self, path, parent_id=None): def add_child(self, path, parent_id=None):
@ -713,6 +715,7 @@ class APIGatewayBackend(BaseBackend):
api_key_source=None, api_key_source=None,
endpoint_configuration=None, endpoint_configuration=None,
tags=None, tags=None,
policy=None,
): ):
api_id = create_id() api_id = create_id()
rest_api = RestAPI( rest_api = RestAPI(
@ -723,6 +726,7 @@ class APIGatewayBackend(BaseBackend):
api_key_source=api_key_source, api_key_source=api_key_source,
endpoint_configuration=endpoint_configuration, endpoint_configuration=endpoint_configuration,
tags=tags, tags=tags,
policy=policy,
) )
self.apis[api_id] = rest_api self.apis[api_id] = rest_api
return rest_api return rest_api

2
moto/apigateway/responses.py
View File

@ -59,6 +59,7 @@ class APIGatewayResponse(BaseResponse):
api_key_source = self._get_param("apiKeySource") api_key_source = self._get_param("apiKeySource")
endpoint_configuration = self._get_param("endpointConfiguration") endpoint_configuration = self._get_param("endpointConfiguration")
tags = self._get_param("tags") tags = self._get_param("tags")
policy = self._get_param("policy")
# Param validation # Param validation
if api_key_source and api_key_source not in API_KEY_SOURCES: if api_key_source and api_key_source not in API_KEY_SOURCES:
@ -94,6 +95,7 @@ class APIGatewayResponse(BaseResponse):
api_key_source=api_key_source, api_key_source=api_key_source,
endpoint_configuration=endpoint_configuration, endpoint_configuration=endpoint_configuration,
tags=tags, tags=tags,
policy=policy,
) )
return 200, {}, json.dumps(rest_api.to_dict()) return 200, {}, json.dumps(rest_api.to_dict())

16
tests/test_apigateway/test_apigateway.py
View File

@ -69,6 +69,22 @@ def test_create_rest_api_with_tags():
response["tags"].should.equal({"MY_TAG1": "MY_VALUE1"}) response["tags"].should.equal({"MY_TAG1": "MY_VALUE1"})
@mock_apigateway
def test_create_rest_api_with_policy():
client = boto3.client("apigateway", region_name="us-west-2")
policy = '{"Version": "2012-10-17","Statement": []}'
response = client.create_rest_api(
name="my_api", description="this is my api", policy=policy
)
api_id = response["id"]
response = client.get_rest_api(restApiId=api_id)
assert "policy" in response
response["policy"].should.equal(policy)
@mock_apigateway @mock_apigateway
def test_create_rest_api_invalid_apikeysource(): def test_create_rest_api_invalid_apikeysource():
client = boto3.client("apigateway", region_name="us-west-2") client = boto3.client("apigateway", region_name="us-west-2")