base rest api endpoints.

moto/apigateway/__init__.py

@@ -0,0 +1,12 @@
+from __future__ import unicode_literals
+from .models import apigateway_backends
+from ..core.models import MockAWS
+
+apigateway_backend = apigateway_backends['us-east-1']
+
+
+def mock_apigateway(func=None):
+    if func:
+        return MockAWS(apigateway_backends)(func)
+    else:
+        return MockAWS(apigateway_backends)

moto/apigateway/models.py

@@ -0,0 +1,55 @@
+from __future__ import unicode_literals
+
+import datetime
+from moto.core import BaseBackend
+from moto.core.utils import iso_8601_datetime_with_milliseconds
+from .utils import create_rest_api_id
+
+
+class RestAPI(object):
+    def __init__(self, id, name, description):
+        self.id = id
+        self.name = name
+        self.description = description
+        self.create_date = datetime.datetime.utcnow()
+
+    def to_dict(self):
+        return {
+            "id": self.id,
+            "name": self.name,
+            "description": self.description,
+            "createdDate": iso_8601_datetime_with_milliseconds(self.create_date),
+        }
+
+
+class APIGatewayBackend(BaseBackend):
+    def __init__(self, region_name):
+        super(APIGatewayBackend, self).__init__()
+        self.apis = {}
+        self.region_name = region_name
+
+    def reset(self):
+        region_name = self.region_name
+        self.__dict__ = {}
+        self.__init__(region_name)
+
+    def create_rest_api(self, name, description):
+        api_id = create_rest_api_id()
+        rest_api = RestAPI(api_id, name, description)
+        self.apis[api_id] = rest_api
+        return rest_api
+
+    def get_rest_api(self, function_id):
+        rest_api = self.apis[function_id]
+        return rest_api
+
+    def list_apis(self):
+        return self.apis.values()
+
+    def delete_rest_api(self, function_id):
+        rest_api = self.apis.pop(function_id)
+        return rest_api
+
+apigateway_backends = {}
+for region_name in ['us-east-1', 'us-west-2', 'eu-west-1', 'ap-northeast-1']:  # Not available in boto yet
+    apigateway_backends[region_name] = APIGatewayBackend(region_name)

moto/apigateway/responses.py

@@ -0,0 +1,40 @@
+from __future__ import unicode_literals
+
+import json
+
+from moto.core.responses import BaseResponse
+from .models import apigateway_backends
+
+
+class APIGatewayResponse(BaseResponse):
+
+    def _get_param(self, key):
+        return json.loads(self.body).get(key)
+
+    @property
+    def backend(self):
+        return apigateway_backends[self.region]
+
+    def restapis(self, request, full_url, headers):
+        self.setup_class(request, full_url, headers)
+
+        if self.method == 'GET':
+            apis = self.backend.list_apis()
+            return 200, headers, json.dumps({"item": [
+                api.to_dict() for api in apis
+            ]})
+        elif self.method == 'POST':
+            name = self._get_param('name')
+            description = self._get_param('description')
+            rest_api = self.backend.create_rest_api(name, description)
+            return 200, headers, json.dumps(rest_api.to_dict())
+
+    def restapis_individual(self, request, full_url, headers):
+        self.setup_class(request, full_url, headers)
+        function_id = self.path.split("/")[-1]
+        if self.method == 'GET':
+            rest_api = self.backend.get_rest_api(function_id)
+            return 200, headers, json.dumps(rest_api.to_dict())
+        elif self.method == 'DELETE':
+            rest_api = self.backend.delete_rest_api(function_id)
+            return 200, headers, json.dumps(rest_api.to_dict())

moto/apigateway/urls.py

@@ -0,0 +1,11 @@
+from __future__ import unicode_literals
+from .responses import APIGatewayResponse
+
+url_bases = [
+    "https?://apigateway.(.+).amazonaws.com"
+]
+
+url_paths = {
+    '{0}/restapis': APIGatewayResponse().restapis,
+    '{0}/restapis/(?P<function_id>[^/]+)/?$': APIGatewayResponse().restapis_individual,
+}

moto/apigateway/utils.py

@@ -0,0 +1,9 @@
+from __future__ import unicode_literals
+import six
+import random
+
+
+def create_rest_api_id():
+    size = 10
+    chars = list(range(10)) + ['A-Z']
+    return ''.join(six.text_type(random.choice(chars)) for x in range(size))