Yasuke/moto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Raise exception on invalid event source type and use full spec

Browse Source
This commit is contained in:
Randy Westergren 2019-08-22 19:28:11 -04:00
parent 819dcfee24
commit 1efd9ee58d
No known key found for this signature in database
GPG Key ID: 9169FEB434BFAB92
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
moto/awslambda/models.py
View File

@ -435,7 +435,6 @@ class EventSourceMapping(BaseModel):
self.event_source_arn = spec['EventSourceArn']
self.uuid = str(uuid.uuid4())
self.last_modified = time.mktime(datetime.datetime.utcnow().timetuple())
self.batch_size = '' # Default to blank
# BatchSize service default/max mapping
batch_size_map = {
@ -453,6 +452,9 @@ class EventSourceMapping(BaseModel):
"BatchSize {} exceeds the max of {}".format(batch_size, batch_size_entry[1]))
else:
self.batch_size = batch_size
else:
raise ValueError("InvalidParameterValueException",
"Unsupported event source type")
# optional
self.starting_position = spec.get('StartingPosition', 'TRIM_HORIZON')
@ -668,7 +670,7 @@ class LambdaBackend(BaseBackend):
raise RESTError('InvalidParameterValueException', 'Missing {}'.format(param))
# Validate function name
func = self._lambdas.get_function_by_name_or_arn(spec.get('FunctionName', ''))
func = self._lambdas.get_function_by_name_or_arn(spec.pop('FunctionName', ''))
if not func:
raise RESTError('ResourceNotFoundException', 'Invalid FunctionName')
@ -682,11 +684,8 @@ class LambdaBackend(BaseBackend):
raise RESTError('InvalidParameterValueException',
'{} is FIFO'.format(queue.queue_arn))
else:
esm_spec = {
'EventSourceArn': spec['EventSourceArn'],
'FunctionArn': func.function_arn,
}
esm = EventSourceMapping(esm_spec)
spec.update({'FunctionArn': func.function_arn})
esm = EventSourceMapping(spec)
self._event_source_mappings[esm.uuid] = esm
# Set backend function on queue