Have context manager return mock.
This commit is contained in:
parent
12aa5dddf7
commit
308712841c
@ -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()
|
||||||
|
10
tests/test_core/test_context_manager.py
Normal file
10
tests/test_core/test_context_manager.py
Normal 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'])
|
Loading…
Reference in New Issue
Block a user