create methods

moto/apigateway/models.py

@@ -3,7 +3,67 @@ 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
+from .utils import create_id
+
+
+class MethodResponse(dict):
+    def __init__(self, status_code):
+        super(MethodResponse, self).__init__()
+        self['statusCode'] = status_code
+
+
+class Method(dict):
+    def __init__(self, method_type, authorization_type):
+        super(Method, self).__init__()
+        self.update(dict(
+            httpMethod=method_type,
+            authorizationType=authorization_type,
+            authorizerId=None,
+            apiKeyRequired=None,
+            requestParameters=None,
+            requestModels=None,
+            methodIntegration=None,
+        ))
+        self.method_responses = {}
+
+    def create_response(self, response_code):
+        method_response = MethodResponse(response_code)
+        self.method_responses[response_code] = method_response
+        return method_response
+
+    def get_response(self, response_code):
+        return self.method_responses[response_code]
+
+    def delete_response(self, response_code):
+        return self.method_responses.pop(response_code)
+
+
+class Resource(object):
+    def __init__(self, id, path_part, parent_id):
+        self.id = id
+        self.path_part = path_part
+        self.parent_id = parent_id
+        self.resource_methods = {
+            'GET': {}
+        }
+
+    def to_dict(self):
+        response = {
+            "path": self.path_part,
+            "id": self.id,
+            "resourceMethods": self.resource_methods,
+        }
+        if self.parent_id:
+            response['parent_id'] = self.parent_id
+        return response
+
+    def add_method(self, method_type, authorization_type):
+        method = Method(method_type=method_type, authorization_type=authorization_type)
+        self.resource_methods[method_type] = method
+        return method
+
+    def get_method(self, method_type):
+        return self.resource_methods[method_type]
 
 
 class RestAPI(object):
@@ -13,6 +73,9 @@ class RestAPI(object):
         self.description = description
         self.create_date = datetime.datetime.utcnow()
 
+        self.resources = {}
+        self.add_child('/')  # Add default child
+
     def to_dict(self):
         return {
             "id": self.id,
@@ -21,6 +84,12 @@ class RestAPI(object):
             "createdDate": iso_8601_datetime_with_milliseconds(self.create_date),
         }
 
+    def add_child(self, path, parent_id=None):
+        child_id = create_id()
+        child = Resource(id=child_id, path_part=path, parent_id=parent_id)
+        self.resources[child_id] = child
+        return child
+
 
 class APIGatewayBackend(BaseBackend):
     def __init__(self, region_name):
@@ -34,7 +103,7 @@ class APIGatewayBackend(BaseBackend):
         self.__init__(region_name)
 
     def create_rest_api(self, name, description):
-        api_id = create_rest_api_id()
+        api_id = create_id()
         rest_api = RestAPI(api_id, name, description)
         self.apis[api_id] = rest_api
         return rest_api
@@ -50,6 +119,52 @@ class APIGatewayBackend(BaseBackend):
         rest_api = self.apis.pop(function_id)
         return rest_api
 
+    def list_resources(self, function_id):
+        api = self.get_rest_api(function_id)
+        return api.resources.values()
+
+    def get_resource(self, function_id, resource_id):
+        api = self.get_rest_api(function_id)
+        resource = api.resources[resource_id]
+        return resource
+
+    def create_resource(self, function_id, parent_resource_id, path_part):
+        api = self.get_rest_api(function_id)
+        child = api.add_child(
+            path=path_part,
+            parent_id=parent_resource_id,
+        )
+        return child
+
+    def delete_resource(self, function_id, resource_id):
+        api = self.get_rest_api(function_id)
+        resource = api.resources.pop(resource_id)
+        return resource
+
+    def get_method(self, function_id, resource_id, method_type):
+        resource = self.get_resource(function_id, resource_id)
+        return resource.get_method(method_type)
+
+    def create_method(self, function_id, resource_id, method_type, authorization_type):
+        resource = self.get_resource(function_id, resource_id)
+        method = resource.add_method(method_type, authorization_type)
+        return method
+
+    def get_method_response(self, function_id, resource_id, method_type, response_code):
+        method = self.get_method(function_id, resource_id, method_type)
+        method_response = method.get_response(response_code)
+        return method_response
+
+    def create_method_response(self, function_id, resource_id, method_type, response_code):
+        method = self.get_method(function_id, resource_id, method_type)
+        method_response = method.create_response(response_code)
+        return method_response
+
+    def delete_method_response(self, function_id, resource_id, method_type, response_code):
+        method = self.get_method(function_id, resource_id, method_type)
+        method_response = method.delete_response(response_code)
+        return method_response
+
 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)