catalin/moto
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Have context manager return mock.

Browse source
This commit is contained in:
Steve Pulec 2019-07-09 20:31:43 -05:00
commit 308712841c
2 changed files with 11 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
moto/core/models.py
View file

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