moto/tests/test_core/test_context_manager.py

13 lines
382 B
Python
Raw Normal View History

2019-07-09 20:31:43 -05:00
import sure # noqa
import boto3
from moto import mock_sqs, settings
2019-07-09 20:31:43 -05:00
2019-07-10 21:16:11 -05:00
def test_context_manager_returns_mock():
2019-07-09 20:31:43 -05:00
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'])