matchers.pop(found_index) break if response_callback is not None: for header, value in request.headers.items(): if isinstance(value, six.binary_type): request.headers[header] = value.decode('utf-8') status, headers, body = response_callback(request, request.url, request.headers) body = MockRawResponse(body) response = AWSResponse(request.url, status, headers, body) return response botocore_stubber = BotocoreStubber() BUILTIN_HANDLERS.append(('before-send', botocore_stubber)) class BotocoreEventMockAWS(BaseMockAWS): def reset(self): botocore_stubber.reset() responses_mock.reset() def enable_patching(self): botocore_stubber.enabled = True for method in BOTOCORE_HTTP_METHODS: for backend in self.backends_for_urls.values(): for key, value in backend.urls.items(): pattern = re.compile(key) botocore_stubber.register_response(method, pattern, value)
break if response_callback is not None: for header, value in request.headers.items(): if isinstance(value, bytes): request.headers[header] = value.decode("utf-8") status, headers, body = response_callback(request, request.url, request.headers) body = MockRawResponse(body) response = AWSResponse(request.url, status, headers, body) return response botocore_stubber = BotocoreStubber() BUILTIN_HANDLERS.append(("before-send", botocore_stubber)) def not_implemented_callback(request): status = 400 headers = {} response = "The method is not implemented" return status, headers, response class BotocoreEventMockAWS(BaseMockAWS): def reset(self): botocore_stubber.reset() responses_mock.reset()
def install(display_fn=display, stdout=False): atexit.register(display_fn, stdout) BUILTIN_HANDLERS.append((BOTOCORE_EVENT_TO_CATCH, register_botocore_event))