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

Merge pull request #1763 from nate/fix/receive_message_wait_time_seconds

Browse Source 
Force receive_message_wait_time_seconds to be int
This commit is contained in:
Steve Pulec 2018-09-22 16:49:24 -04:00 committed by GitHub
commit 0ab07bc482
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 14 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
moto/sqs/responses.py
View File

@ -336,7 +336,7 @@ class SQSResponse(BaseResponse):
try: try:
wait_time = int(self.querystring.get("WaitTimeSeconds")[0]) wait_time = int(self.querystring.get("WaitTimeSeconds")[0])
except TypeError: except TypeError:
wait_time = queue.receive_message_wait_time_seconds wait_time = int(queue.receive_message_wait_time_seconds)
if wait_time < 0 or wait_time > 20: if wait_time < 0 or wait_time > 20:
return self._error( return self._error(

13
tests/test_sqs/test_sqs.py
View File

@ -1222,3 +1222,16 @@ def test_receive_messages_with_message_group_id_on_visibility_timeout():
messages = queue.receive_messages() messages = queue.receive_messages()
messages.should.have.length_of(1) messages.should.have.length_of(1)
messages[0].message_id.should.equal(message.message_id) messages[0].message_id.should.equal(message.message_id)
@mock_sqs
def test_receive_message_for_queue_with_receive_message_wait_time_seconds_set():
sqs = boto3.resource('sqs', region_name='us-east-1')
queue = sqs.create_queue(
QueueName='test-queue',
Attributes={
'ReceiveMessageWaitTimeSeconds': '2',
}
)
queue.receive_messages()