moto/tests/test_core/test_context_manager.py

14 lines
445 B
Python

import boto3
from moto import mock_sqs, settings
from tests import DEFAULT_ACCOUNT_ID
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:
backend = sqs_mock.backends[DEFAULT_ACCOUNT_ID]["us-west-1"]
assert list(backend.queues.keys()) == ["queue1"]