moto/tests/test_sqs/test_server.py

from __future__ import unicode_literals

import re
import sure  # noqa
import threading
import time

import moto.server as server

"""
Test the different server responses
"""


def test_sqs_list_identities():
    backend = server.create_backend_app("sqs")
    test_client = backend.test_client()

    res = test_client.get("/?Action=ListQueues")
    res.data.should.contain(b"ListQueuesResponse")

    # Make sure that we can receive messages from queues whose name contains dots (".")
    # The AWS API mandates that the names of FIFO queues use the suffix ".fifo"
    # See: https://github.com/spulec/moto/issues/866

    for queue_name in ("testqueue", "otherqueue.fifo"):

        res = test_client.put("/?Action=CreateQueue&QueueName=%s" % queue_name)

        res = test_client.put(
            "/123/%s?MessageBody=test-message&Action=SendMessage" % queue_name
        )

        res = test_client.get(
            "/123/%s?Action=ReceiveMessage&MaxNumberOfMessages=1" % queue_name
        )

        message = re.search("<Body>(.*?)</Body>", res.data.decode("utf-8")).groups()[0]
        message.should.equal("test-message")

    res = test_client.get("/?Action=ListQueues&QueueNamePrefix=other")
    res.data.should.contain(b"otherqueue.fifo")
    res.data.should_not.contain(b"testqueue")


def test_messages_polling():
    backend = server.create_backend_app("sqs")
    test_client = backend.test_client()
    messages = []

    test_client.put("/?Action=CreateQueue&QueueName=testqueue")

    def insert_messages():
        messages_count = 5
        while messages_count > 0:
            test_client.put(
                "/123/testqueue?MessageBody=test-message&Action=SendMessage"
                "&Attribute.1.Name=WaitTimeSeconds&Attribute.1.Value=10"
            )
            messages_count -= 1
            time.sleep(0.5)

    def get_messages():
        count = 0
        while count < 5:
            msg_res = test_client.get(
                "/123/testqueue?Action=ReceiveMessage&MaxNumberOfMessages=1&WaitTimeSeconds=5"
            )
            new_msgs = re.findall("<Body>(.*?)</Body>", msg_res.data.decode("utf-8"))
            count += len(new_msgs)
            messages.append(new_msgs)

    get_messages_thread = threading.Thread(target=get_messages)
    insert_messages_thread = threading.Thread(target=insert_messages)

    get_messages_thread.start()
    insert_messages_thread.start()

    get_messages_thread.join()
    insert_messages_thread.join()

    # got each message in a separate call to ReceiveMessage, despite the long
    # WaitTimeSeconds
    assert len(messages) == 5
from __future__ import unicode_literals 2014-08-27 11:17:06 -04:00			`from __future__ import unicode_literals`
Added support for WaitTimeSeconds in SQS #392 2015-08-10 22:23:17 +02:00
Adding server mode 2013-03-05 08:14:43 -05:00			`import re`
Cleanup flake8 2013-08-03 17:21:25 -04:00			`import sure # noqa`
Added support for WaitTimeSeconds in SQS #392 2015-08-10 22:23:17 +02:00			`import threading`
			`import time`
Adding server mode 2013-03-05 08:14:43 -05:00
			`import moto.server as server`

Run black on moto & test directories. 2019-10-31 08:44:26 -07:00			`"""`
Adding server mode 2013-03-05 08:14:43 -05:00			`Test the different server responses`
Run black on moto & test directories. 2019-10-31 08:44:26 -07:00			`"""`
Adding server mode 2013-03-05 08:14:43 -05:00

fixes to sqs encoding. Closes #4. 2013-03-23 09:44:53 -04:00			`def test_sqs_list_identities():`
Clean up querystring logic. 2013-12-28 20:15:37 -05:00			`backend = server.create_backend_app("sqs")`
			`test_client = backend.test_client()`

Run black on moto & test directories. 2019-10-31 08:44:26 -07:00			`res = test_client.get("/?Action=ListQueues")`
Support Python 3 using six 2014-08-26 13:25:50 -04:00			`res.data.should.contain(b"ListQueuesResponse")`
Adding server mode 2013-03-05 08:14:43 -05:00
fix receiving of messages from queues with a dot character in their name 2017-08-05 15:47:40 +10:00			`# Make sure that we can receive messages from queues whose name contains dots (".")`
			`# The AWS API mandates that the names of FIFO queues use the suffix ".fifo"`
			`# See: https://github.com/spulec/moto/issues/866`

Run black on moto & test directories. 2019-10-31 08:44:26 -07:00			`for queue_name in ("testqueue", "otherqueue.fifo"):`
support QueueNamePrefix in the SQS Backend 2014-02-20 11:00:08 -08:00
Run black on moto & test directories. 2019-10-31 08:44:26 -07:00			`res = test_client.put("/?Action=CreateQueue&QueueName=%s" % queue_name)`
Adding server mode 2013-03-05 08:14:43 -05:00
fix receiving of messages from queues with a dot character in their name 2017-08-05 15:47:40 +10:00			`res = test_client.put(`
Run black on moto & test directories. 2019-10-31 08:44:26 -07:00			`"/123/%s?MessageBody=test-message&Action=SendMessage" % queue_name`
			`)`
Adding server mode 2013-03-05 08:14:43 -05:00
fix receiving of messages from queues with a dot character in their name 2017-08-05 15:47:40 +10:00			`res = test_client.get(`
Run black on moto & test directories. 2019-10-31 08:44:26 -07:00			`"/123/%s?Action=ReceiveMessage&MaxNumberOfMessages=1" % queue_name`
			`)`
Enhanced SQS support - Support for delaying messages - Support for visibility timeouts - Support for actually deleting messages - Support for message bodies longer than 27 characters - Support for message attributes - Support for accurate queue attributes Caveats: - All message attributes are returned regardless of whether or not attributes were requested when reading messages - I’m not sure why messages longer than 27 characters were breaking in my tests. Boto seems to expect the body to be base64 encoded and bodies less than 27 characters would be fine, but if I attempted to use a larger body it would mangle the content. I now base64 encode the body if the raw string is longer than 27 characters and all is fine. 2014-06-20 15:00:36 -05:00
Run black on moto & test directories. 2019-10-31 08:44:26 -07:00			`message = re.search("<Body>(.*?)</Body>", res.data.decode("utf-8")).groups()[0]`
			`message.should.equal("test-message")`
fix receiving of messages from queues with a dot character in their name 2017-08-05 15:47:40 +10:00
Run black on moto & test directories. 2019-10-31 08:44:26 -07:00			`res = test_client.get("/?Action=ListQueues&QueueNamePrefix=other")`
			`res.data.should.contain(b"otherqueue.fifo")`
			`res.data.should_not.contain(b"testqueue")`
Added support for WaitTimeSeconds in SQS #392 2015-08-10 22:23:17 +02:00

			`def test_messages_polling():`
			`backend = server.create_backend_app("sqs")`
			`test_client = backend.test_client()`
			`messages = []`

Run black on moto & test directories. 2019-10-31 08:44:26 -07:00			`test_client.put("/?Action=CreateQueue&QueueName=testqueue")`
Added support for WaitTimeSeconds in SQS #392 2015-08-10 22:23:17 +02:00
			`def insert_messages():`
			`messages_count = 5`
			`while messages_count > 0:`
			`test_client.put(`
Run black on moto & test directories. 2019-10-31 08:44:26 -07:00			`"/123/testqueue?MessageBody=test-message&Action=SendMessage"`
			`"&Attribute.1.Name=WaitTimeSeconds&Attribute.1.Value=10"`
Added support for WaitTimeSeconds in SQS #392 2015-08-10 22:23:17 +02:00			`)`
			`messages_count -= 1`
Run black on moto & test directories. 2019-10-31 08:44:26 -07:00			`time.sleep(0.5)`
Added support for WaitTimeSeconds in SQS #392 2015-08-10 22:23:17 +02:00
			`def get_messages():`
update test_messages_polling to assume messages are returned immediately 2015-08-31 11:54:17 -04:00			`count = 0`
			`while count < 5:`
			`msg_res = test_client.get(`
Run black on moto & test directories. 2019-10-31 08:44:26 -07:00			`"/123/testqueue?Action=ReceiveMessage&MaxNumberOfMessages=1&WaitTimeSeconds=5"`
update test_messages_polling to assume messages are returned immediately 2015-08-31 11:54:17 -04:00			`)`
Run black on moto & test directories. 2019-10-31 08:44:26 -07:00			`new_msgs = re.findall("<Body>(.*?)</Body>", msg_res.data.decode("utf-8"))`
update test_messages_polling to assume messages are returned immediately 2015-08-31 11:54:17 -04:00			`count += len(new_msgs)`
			`messages.append(new_msgs)`
Added support for WaitTimeSeconds in SQS #392 2015-08-10 22:23:17 +02:00
			`get_messages_thread = threading.Thread(target=get_messages)`
			`insert_messages_thread = threading.Thread(target=insert_messages)`

			`get_messages_thread.start()`
			`insert_messages_thread.start()`

			`get_messages_thread.join()`
			`insert_messages_thread.join()`

Lints. 2017-02-23 21:37:43 -05:00			`# got each message in a separate call to ReceiveMessage, despite the long`
			`# WaitTimeSeconds`
Added support for WaitTimeSeconds in SQS #392 2015-08-10 22:23:17 +02:00			`assert len(messages) == 5`