Have context manager return mock.

This commit is contained in:
Steve Pulec 2019-07-09 20:31:43 -05:00
parent 12aa5dddf7
commit 308712841c
2 changed files with 11 additions and 0 deletions

View File

@ -52,6 +52,7 @@ class BaseMockAWS(object):
def __enter__(self): def __enter__(self):
self.start() self.start()
return self
def __exit__(self, *args): def __exit__(self, *args):
self.stop() self.stop()

View File

@ -0,0 +1,10 @@
import sure # noqa
import boto3
from moto import mock_sqs
def test_reset_api():
with mock_sqs() as sqs_mock:
conn = boto3.client("sqs", region_name='us-west-1')
conn.create_queue(QueueName="queue1")
list(sqs_mock.backends['us-west-1'].queues.keys()).should.equal(['queue1'])