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

Check SQS message size

Browse Source
This commit is contained in:
Ian Dees 2015-11-09 19:29:08 -05:00
parent ab3682a55c
commit 5b2a724219
2 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
moto/sqs/exceptions.py
View File

@ -11,6 +11,11 @@ class ReceiptHandleIsInvalid(Exception):
status_code = 400 status_code = 400
class InvalidParameterValue(Exception):
description = "One or more parameters are invalid. Reason: Message must be shorter than 262144 bytes."
status_code = 400
class MessageAttributesInvalid(Exception): class MessageAttributesInvalid(Exception):
status_code = 400 status_code = 400

6
moto/sqs/models.py
View File

@ -12,7 +12,8 @@ from moto.core.utils import camelcase_to_underscores, get_random_message_id
from .utils import generate_receipt_handle, unix_time_millis from .utils import generate_receipt_handle, unix_time_millis
from .exceptions import ( from .exceptions import (
ReceiptHandleIsInvalid, ReceiptHandleIsInvalid,
MessageNotInflight MessageNotInflight,
InvalidParameterValue,
) )
DEFAULT_ACCOUNT_ID = 123456789012 DEFAULT_ACCOUNT_ID = 123456789012
@ -251,6 +252,9 @@ 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)