github/datasette
1
0
Fork 0
mirror of https://github.com/simonw/datasette.git synced 2025-12-10 16:51:24 +01:00
Code Issues Projects Releases Packages Wiki Activity

Request.fake(... url_vars), plus .fake() is now documented

Browse source 
Also made 'from datasette import Request' shortcut work.

Closes #1697
This commit is contained in:
Simon Willison 2022-03-31 19:01:58 -07:00
commit 5c5e9b3657
4 changed files with 38 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
datasette/__init__.py
View file

@ -1,5 +1,5 @@
from datasette.version import __version_info__, __version__ # noqa
from datasette.utils.asgi import Forbidden, NotFound, Response # noqa
from datasette.utils.asgi import Forbidden, NotFound, Request, Response # noqa
from datasette.utils import actor_matches_allow # noqa
from .hookspecs import hookimpl # noqa
from .hookspecs import hookspec # noqa

4
datasette/utils/asgi.py
View file

@ -118,7 +118,7 @@ class Request:
return dict(parse_qsl(body.decode("utf-8"), keep_blank_values=True))
@classmethod
def fake(cls, path_with_query_string, method="GET", scheme="http"):
def fake(cls, path_with_query_string, method="GET", scheme="http", url_vars=None):
"""Useful for constructing Request objects for tests"""
path, _, query_string = path_with_query_string.partition("?")
scope = {
@ -130,6 +130,8 @@ class Request:
"scheme": scheme,
"type": "http",
}
if url_vars:
scope["url_route"] = {"kwargs": url_vars}
return cls(scope, None)