async def clone_request(self, method, endpoint, payload, headers): container_url = IAbsoluteURL(self.request.container, self.request)() url = posixpath.join(container_url, endpoint) parsed = urlparse(url) dct = {'method': method, 'url': URL(url), 'path': parsed.path} dct['headers'] = CIMultiDict(headers) dct['raw_headers'] = tuple( (k.encode('utf-8'), v.encode('utf-8')) for k, v in headers.items()) message = self.request._message._replace(**dct) payload_writer = mock.Mock() payload_writer.write_eof.side_effect = noop payload_writer.drain.side_effect = noop protocol = mock.Mock() protocol.transport = test_utils._create_transport(None) protocol.writer = payload_writer request = self.request.__class__( message, SimplePayload(payload), protocol, payload_writer, self.request._task, self.request._loop, client_max_size=self.request._client_max_size, state=self.request._state.copy(), scheme=self.request.scheme, host=self.request.host, remote=self.request.remote) request._db_write_enabled = True request._db_id = self.request._db_id request._tm = self.request._tm request._txn = self.request._txn request._container_id = self.context.id request.container = self.context annotations_container = IAnnotations(self.context) request.container_settings = await annotations_container.async_get( REGISTRY_DATA_KEY) layers = request.container_settings.get(ACTIVE_LAYERS_KEY, []) for layer in layers: try: alsoProvides(request, import_class(layer)) except ModuleNotFoundError: pass request._futures = self.request._futures return request
def make_mocked_request(method, path, headers=None, *, version=HttpVersion(1, 1), closing=False, app=None, writer=sentinel, payload_writer=sentinel, protocol=sentinel, transport=sentinel, payload=sentinel, sslcontext=None, client_max_size=1024**2): """ XXX copied from aiohttp but using guillotina request object Creates mocked web.Request testing purposes. Useful in unit tests, when spinning full web server is overkill or specific conditions and errors are hard to trigger. """ task = mock.Mock() loop = mock.Mock() loop.create_future.return_value = () if version < HttpVersion(1, 1): closing = True if headers: headers = CIMultiDict(headers) raw_hdrs = tuple( (k.encode('utf-8'), v.encode('utf-8')) for k, v in headers.items()) else: headers = CIMultiDict() raw_hdrs = () chunked = 'chunked' in headers.get(hdrs.TRANSFER_ENCODING, '').lower() message = RawRequestMessage( method, path, version, headers, raw_hdrs, closing, False, False, chunked, URL(path)) if app is None: app = test_utils._create_app_mock() if protocol is sentinel: protocol = mock.Mock() if transport is sentinel: transport = test_utils._create_transport(sslcontext) if writer is sentinel: writer = mock.Mock() writer.transport = transport if payload_writer is sentinel: payload_writer = mock.Mock() payload_writer.write_eof.side_effect = noop payload_writer.drain.side_effect = noop protocol.transport = transport protocol.writer = writer if payload is sentinel: payload = mock.Mock() time_service = mock.Mock() time_service.time.return_value = 12345 time_service.strtime.return_value = "Tue, 15 Nov 1994 08:12:31 GMT" @contextmanager def timeout(*args, **kw): yield time_service.timeout = mock.Mock() time_service.timeout.side_effect = timeout req = Request(message, payload, protocol, payload_writer, time_service, task, client_max_size=client_max_size) match_info = UrlMappingMatchInfo({}, mock.Mock()) match_info.add_app(app) req._match_info = match_info return req
async def create_request(self): req_data = self.data['req_data'] url = req_data['url'] parsed = urlparse(url) dct = { 'method': req_data['method'], 'url': yarl.URL(url), 'path': parsed.path, 'headers': CIMultiDict(req_data['headers']), 'raw_headers': tuple((k.encode('utf-8'), v.encode('utf-8')) for k, v in req_data['headers'].items()) } message = self.base_request._message._replace(**dct) payload_writer = mock.Mock() payload_writer.write_eof.side_effect = noop payload_writer.drain.side_effect = noop protocol = mock.Mock() protocol.transport = test_utils._create_transport(None) protocol.writer = payload_writer request = self.base_request.__class__( message, EmptyPayload(), protocol, payload_writer, self.task, self.task._loop, client_max_size=self.base_request._client_max_size, state=self.base_request._state.copy()) g_task_vars.request.set(request) request.annotations = req_data.get('annotations', {}) if self.data.get('db_id'): root = get_utility(IApplication, name='root') db = await root.async_get(self.data['db_id']) g_task_vars.db.set(db) # Add a transaction Manager to request tm = db.get_transaction_manager() g_task_vars.tm.set(tm) # Start a transaction txn = await tm.begin() # Get the root of the tree context = await tm.get_root(txn=txn) if self.data.get('container_id'): container = await context.async_get(self.data['container_id']) if container is None: raise Exception( f'Could not find container: {self.data["container_id"]}' ) g_task_vars.container.set(container) annotations_container = IAnnotations(container) container_settings = await annotations_container.async_get( REGISTRY_DATA_KEY) layers = container_settings.get(ACTIVE_LAYERS_KEY, []) for layer in layers: try: alsoProvides(request, import_class(layer)) except ModuleNotFoundError: pass g_task_vars.registry.set(container_settings) return request
def make_mocked_request(method, path, headers=None, *, version=HttpVersion(1, 1), closing=False, app=None, writer=sentinel, payload_writer=sentinel, protocol=sentinel, transport=sentinel, payload=sentinel, sslcontext=None, client_max_size=1024**2): """ XXX copied from aiohttp but using guillotina request object Creates mocked web.Request testing purposes. Useful in unit tests, when spinning full web server is overkill or specific conditions and errors are hard to trigger. """ task = mock.Mock() loop = mock.Mock() loop.create_future.return_value = () if version < HttpVersion(1, 1): closing = True if headers: headers = CIMultiDict(headers) raw_hdrs = tuple( (k.encode('utf-8'), v.encode('utf-8')) for k, v in headers.items()) else: headers = CIMultiDict() raw_hdrs = () chunked = 'chunked' in headers.get(hdrs.TRANSFER_ENCODING, '').lower() message = RawRequestMessage( method, path, version, headers, raw_hdrs, closing, False, False, chunked, URL(path)) if app is None: app = test_utils._create_app_mock() if protocol is sentinel: protocol = mock.Mock() if transport is sentinel: transport = test_utils._create_transport(sslcontext) if writer is sentinel: writer = mock.Mock() writer.transport = transport if payload_writer is sentinel: payload_writer = mock.Mock() payload_writer.write_eof.side_effect = noop payload_writer.drain.side_effect = noop protocol.transport = transport protocol.writer = writer if payload is sentinel: payload = mock.Mock() time_service = mock.Mock() time_service.time.return_value = 12345 time_service.strtime.return_value = "Tue, 15 Nov 1994 08:12:31 GMT" @contextmanager def timeout(*args, **kw): yield time_service.timeout = mock.Mock() time_service.timeout.side_effect = timeout req = Request(message, payload, protocol, payload_writer, time_service, task, client_max_size=client_max_size) match_info = UrlMappingMatchInfo({}, mock.Mock()) match_info.add_app(app) req._match_info = match_info return req