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

Put the size check in the queue

Browse Source
This commit is contained in:
Ian Dees 2015-11-09 19:51:17 -05:00
parent 5b2a724219
commit 540ee79ad9
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
moto/sqs/models.py
View File

@ -190,6 +190,9 @@ class Queue(object):
return [message for message in self._messages if message.visible and not message.delayed] return [message for message in self._messages if message.visible and not message.delayed]
def add_message(self, message): def add_message(self, message):
if len(message) > self.maximum_message_size:
raise InvalidParameterValue()
self._messages.append(message) self._messages.append(message)
def get_cfn_attribute(self, attribute_name): def get_cfn_attribute(self, attribute_name):
@ -252,9 +255,6 @@ class SQSBackend(BaseBackend):
else: else:
delay_seconds = queue.delay_seconds delay_seconds = queue.delay_seconds
if len(message_body) > self.maximum_message_size:
raise InvalidParameterValue
message_id = get_random_message_id() message_id = get_random_message_id()
message = Message(message_id, message_body) message = Message(message_id, message_body)