From a658900d69ca4ae36a4b265161809a529aabb211 Mon Sep 17 00:00:00 2001
From: JohnWC <johnwc@users.noreply.github.com>
Date: Sat, 25 Apr 2020 03:13:36 -0500
Subject: [PATCH 1/3] Add policy to apigateway

---
 moto/apigateway/models.py    | 4 ++++
 moto/apigateway/responses.py | 2 ++
 2 files changed, 6 insertions(+)

diff --git a/moto/apigateway/models.py b/moto/apigateway/models.py
index e5e5e3bfd..e011af601 100644
--- a/moto/apigateway/models.py
+++ b/moto/apigateway/models.py
@@ -461,6 +461,7 @@ class RestAPI(BaseModel):
         self.description = description
         self.create_date = int(time.time())
         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 {
             "types": ["EDGE"]
         }
@@ -485,6 +486,7 @@ class RestAPI(BaseModel):
             "apiKeySource": self.api_key_source,
             "endpointConfiguration": self.endpoint_configuration,
             "tags": self.tags,
+            "policy": self.policy,
         }
 
     def add_child(self, path, parent_id=None):
@@ -713,6 +715,7 @@ class APIGatewayBackend(BaseBackend):
         api_key_source=None,
         endpoint_configuration=None,
         tags=None,
+        policy=None,
     ):
         api_id = create_id()
         rest_api = RestAPI(
@@ -723,6 +726,7 @@ class APIGatewayBackend(BaseBackend):
             api_key_source=api_key_source,
             endpoint_configuration=endpoint_configuration,
             tags=tags,
+            policy=policy,
         )
         self.apis[api_id] = rest_api
         return rest_api
diff --git a/moto/apigateway/responses.py b/moto/apigateway/responses.py
index 822d4c0ce..a3c41a6d4 100644
--- a/moto/apigateway/responses.py
+++ b/moto/apigateway/responses.py
@@ -59,6 +59,7 @@ class APIGatewayResponse(BaseResponse):
             api_key_source = self._get_param("apiKeySource")
             endpoint_configuration = self._get_param("endpointConfiguration")
             tags = self._get_param("tags")
+            policy = self._get_param("policy")
 
             # Param validation
             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,
                 endpoint_configuration=endpoint_configuration,
                 tags=tags,
+                policy=policy,
             )
             return 200, {}, json.dumps(rest_api.to_dict())
 

From 0828c5af9dfff7430537cfb26cc62a8523d9cef3 Mon Sep 17 00:00:00 2001
From: JohnWC <johnwc@users.noreply.github.com>
Date: Sat, 25 Apr 2020 03:27:59 -0500
Subject: [PATCH 2/3] Add unit test for add apigateway with policy

---
 tests/test_apigateway/test_apigateway.py | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/tests/test_apigateway/test_apigateway.py b/tests/test_apigateway/test_apigateway.py
index 596ed2dd4..107dc5d05 100644
--- a/tests/test_apigateway/test_apigateway.py
+++ b/tests/test_apigateway/test_apigateway.py
@@ -69,6 +69,24 @@ def test_create_rest_api_with_tags():
     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
 def test_create_rest_api_invalid_apikeysource():
     client = boto3.client("apigateway", region_name="us-west-2")

From 4a800d8f2c8677b098ea0a2c41deface8236c267 Mon Sep 17 00:00:00 2001
From: JohnWC <johnwc@users.noreply.github.com>
Date: Sat, 25 Apr 2020 11:24:54 -0500
Subject: [PATCH 3/3] Updated for black

---
 tests/test_apigateway/test_apigateway.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/tests/test_apigateway/test_apigateway.py b/tests/test_apigateway/test_apigateway.py
index 107dc5d05..b04328a03 100644
--- a/tests/test_apigateway/test_apigateway.py
+++ b/tests/test_apigateway/test_apigateway.py
@@ -73,11 +73,9 @@ def test_create_rest_api_with_tags():
 def test_create_rest_api_with_policy():
     client = boto3.client("apigateway", region_name="us-west-2")
 
-    policy = "{\"Version\": \"2012-10-17\",\"Statement\": []}"
+    policy = '{"Version": "2012-10-17","Statement": []}'
     response = client.create_rest_api(
-        name="my_api",
-        description="this is my api",
-        policy=policy
+        name="my_api", description="this is my api", policy=policy
     )
     api_id = response["id"]