From 540ee79ad9c024015936efe9fae26663ba764c7f Mon Sep 17 00:00:00 2001 From: Ian Dees Date: Mon, 9 Nov 2015 19:51:17 -0500 Subject: [PATCH] Put the size check in the queue --- moto/sqs/models.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/moto/sqs/models.py b/moto/sqs/models.py index 725465498..17c33e103 100644 --- a/moto/sqs/models.py +++ b/moto/sqs/models.py @@ -190,6 +190,9 @@ class Queue(object): return [message for message in self._messages if message.visible and not message.delayed] def add_message(self, message): + if len(message) > self.maximum_message_size: + raise InvalidParameterValue() + self._messages.append(message) def get_cfn_attribute(self, attribute_name): @@ -252,9 +255,6 @@ class SQSBackend(BaseBackend): else: delay_seconds = queue.delay_seconds - if len(message_body) > self.maximum_message_size: - raise InvalidParameterValue - message_id = get_random_message_id() message = Message(message_id, message_body)