catalin/moto
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Testing new version of decorator.

Browse source
This commit is contained in:
Steve Pulec 2017-02-15 22:35:45 -05:00
commit fde721bed7
123 changed files with 2740 additions and 1114 deletions
Download patch file Download diff file Expand all files Collapse all files

4
moto/awslambda/__init__.py
View file

@ -1,7 +1,7 @@
from __future__ import unicode_literals
from .models import lambda_backends
from ..core.models import MockAWS, base_decorator
from ..core.models import MockAWS, base_decorator, HttprettyMockAWS, deprecated_base_decorator
lambda_backend = lambda_backends['us-east-1']
mock_lambda = base_decorator(lambda_backends)
mock_lambda_deprecated = deprecated_base_decorator(lambda_backends)

14
moto/awslambda/responses.py
View file

@ -9,7 +9,7 @@ from .models import lambda_backends
class LambdaResponse(BaseResponse):
@classmethod
def root(cls, request, full_url, headers):
if request.method == 'GET':
@ -38,11 +38,13 @@ class LambdaResponse(BaseResponse):
def _invoke(self, request, full_url, headers):
lambda_backend = self.get_lambda_backend(full_url)
function_name = request.path.split('/')[-2]
path = request.path if hasattr(request, 'path') else request.path_url
function_name = path.split('/')[-2]
if lambda_backend.has_function(function_name):
fn = lambda_backend.get_function(function_name)
payload = fn.invoke(request, headers)
headers['Content-Length'] = str(len(payload))
return 202, headers, payload
else:
return 404, headers, "{}"
@ -68,7 +70,8 @@ class LambdaResponse(BaseResponse):
def _delete_function(self, request, full_url, headers):
lambda_backend = self.get_lambda_backend(full_url)
function_name = request.path.split('/')[-1]
path = request.path if hasattr(request, 'path') else request.path_url
function_name = path.split('/')[-1]
if lambda_backend.has_function(function_name):
lambda_backend.delete_function(function_name)
@ -79,7 +82,8 @@ class LambdaResponse(BaseResponse):
def _get_function(self, request, full_url, headers):
lambda_backend = self.get_lambda_backend(full_url)
function_name = request.path.split('/')[-1]
path = request.path if hasattr(request, 'path') else request.path_url
function_name = path.split('/')[-1]
if lambda_backend.has_function(function_name):
fn = lambda_backend.get_function(function_name)
@ -87,7 +91,7 @@ class LambdaResponse(BaseResponse):
return 200, headers, json.dumps(code)
else:
return 404, headers, "{}"
def get_lambda_backend(self, full_url):
from moto.awslambda.models import lambda_backends
region = self._get_aws_region(full_url)