Try to get tests running
This commit is contained in:
parent
fd4e524855
commit
b20e190995
@ -303,6 +303,7 @@ BUILTIN_HANDLERS.append(('before-send', botocore_stubber))
|
|||||||
class BotocoreEventMockAWS(BaseMockAWS):
|
class BotocoreEventMockAWS(BaseMockAWS):
|
||||||
def reset(self):
|
def reset(self):
|
||||||
botocore_stubber.reset()
|
botocore_stubber.reset()
|
||||||
|
responses_mock.reset()
|
||||||
|
|
||||||
def enable_patching(self):
|
def enable_patching(self):
|
||||||
botocore_stubber.enabled = True
|
botocore_stubber.enabled = True
|
||||||
@ -312,10 +313,32 @@ class BotocoreEventMockAWS(BaseMockAWS):
|
|||||||
pattern = re.compile(key)
|
pattern = re.compile(key)
|
||||||
botocore_stubber.register_response(method, pattern, value)
|
botocore_stubber.register_response(method, pattern, value)
|
||||||
|
|
||||||
|
if not hasattr(responses_mock, '_patcher') or not hasattr(responses_mock._patcher, 'target'):
|
||||||
|
responses_mock.start()
|
||||||
|
|
||||||
|
for method in RESPONSES_METHODS:
|
||||||
|
# for backend in default_backends.values():
|
||||||
|
for backend in self.backends_for_urls.values():
|
||||||
|
for key, value in backend.urls.items():
|
||||||
|
responses_mock.add(
|
||||||
|
CallbackResponse(
|
||||||
|
method=method,
|
||||||
|
url=re.compile(key),
|
||||||
|
callback=convert_flask_to_responses_response(value),
|
||||||
|
stream=True,
|
||||||
|
match_querystring=False,
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
def disable_patching(self):
|
def disable_patching(self):
|
||||||
botocore_stubber.enabled = False
|
botocore_stubber.enabled = False
|
||||||
self.reset()
|
self.reset()
|
||||||
|
|
||||||
|
try:
|
||||||
|
responses_mock.stop()
|
||||||
|
except RuntimeError:
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
MockAWS = BotocoreEventMockAWS
|
MockAWS = BotocoreEventMockAWS
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user