def make_mocked_request(
method: str,
path: str,
headers: Dict = None,
query_string: bytes = b"",
payload: bytes = b"",
*,
app=None,
client_max_size=1024**2,
):
if headers is None:
headers = {}
if "Host" not in headers:
headers["Host"] = "localhost"
raw_hdrs = list(
(k.encode("utf-8"), v.encode("utf-8")) for k, v in headers.items())
q: asyncio.Queue[Dict] = asyncio.Queue()
chunks = [payload[i:i + 1024]
for i in range(0, len(payload), 1024)] or [b""]
for i, chunk in enumerate(chunks):
q.put_nowait({
"type": "http.request",
"body": chunk,
"more_body": i < len(chunks) - 1
})
return Request("http",
method,
path,
query_string,
raw_hdrs,
client_max_size=client_max_size,
receive=q.get)
async def __call__(self, scope, receive, send):
"""
This method always returns a response object or raises a Exception for
unhandled errors
"""
request_settings = {
k: v for k, v in self.asgi_app.server_settings.items() if k in ("client_max_size",)
}
request = Request.factory(scope, send, receive, **request_settings)
task_vars.request.set(request)
try:
return await self.request_handler(request)
except Response as exc:
return exc
except Exception as exc:
# Try to render exception using IErrorResponseException
eid = uuid.uuid4().hex
view_result = query_adapter(
exc, IErrorResponseException, kwargs={"error": "ServiceError", "eid": eid}
)
if view_result is not None:
return await apply_rendering(View(None, request), request, view_result)
# Raise unhandled exceptions to ErrorMiddleware
raise
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
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