moto/tests/test_core/test_context_manager.py

import sure  # noqa
import boto3
from moto import mock_sqs, settings


def test_context_manager_returns_mock():
    with mock_sqs() as sqs_mock:
        conn = boto3.client("sqs", region_name="us-west-1")
        conn.create_queue(QueueName="queue1")

        if not settings.TEST_SERVER_MODE:
            list(sqs_mock.backends["us-west-1"].queues.keys()).should.equal(["queue1"])
Have context manager return mock. 2019-07-09 20:31:43 -05:00			`import sure # noqa`
			`import boto3`
Dont test context manager in server mode. 2019-07-10 21:40:49 -05:00			`from moto import mock_sqs, settings`

Have context manager return mock. 2019-07-09 20:31:43 -05:00
Cleanup model ref resetting. 2019-07-10 21:16:11 -05:00			`def test_context_manager_returns_mock():`
Have context manager return mock. 2019-07-09 20:31:43 -05:00			`with mock_sqs() as sqs_mock:`
Run black on moto & test directories. 2019-10-31 08:44:26 -07:00			`conn = boto3.client("sqs", region_name="us-west-1")`
Have context manager return mock. 2019-07-09 20:31:43 -05:00			`conn.create_queue(QueueName="queue1")`

Dont test context manager in server mode. 2019-07-10 21:40:49 -05:00			`if not settings.TEST_SERVER_MODE:`
Run black on moto & test directories. 2019-10-31 08:44:26 -07:00			`list(sqs_mock.backends["us-west-1"].queues.keys()).should.equal(["queue1"])`