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

added UpdateFunctionCode and UpdateFunctionConfiguration and associated test cases

Browse source
This commit is contained in:
Seth Black 2019-10-08 15:59:03 -05:00
commit 41af98c98b
4 changed files with 210 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

36
moto/awslambda/responses.py
View file

@ -122,6 +122,18 @@ class LambdaResponse(BaseResponse):
if request.method == 'POST':
return self._add_policy(request, full_url, headers)
def configuration(self, request, full_url, headers):
if request.method == 'PUT':
return self._put_configuration(request, full_url)
else:
raise ValueError("Cannot handle request")
def code(self, request, full_url, headers):
if request.method == 'PUT':
return self._put_code(request, full_url, headers)
else:
raise ValueError("Cannot handle request")
def _add_policy(self, request, full_url, headers):
path = request.path if hasattr(request, 'path') else path_url(request.url)
function_name = path.split('/')[-2]
@ -308,3 +320,27 @@ class LambdaResponse(BaseResponse):
return 204, {}, "{}"
else:
return 404, {}, "{}"
def _put_configuration(self, request, full_url):
function_name = self._get_param('FunctionName', None)
qualifier = self._get_param('Qualifier', None)
fn = self.lambda_backend.get_function(function_name, qualifier)
if fn:
config = fn.update_configuration(json.loads(request.body))
return 200, {}, json.dumps(config)
else:
return 404, {}, "{}"
def _put_code(self, request, full_url, headers):
function_name = self._get_param('FunctionName', None)
qualifier = self._get_param('Qualifier', None)
fn = self.lambda_backend.get_function(function_name, qualifier)
if fn:
config = fn.update_function_code(json.loads(request.body))
return 200, {}, json.dumps(config)
else:
return 404, {}, "{}"