mirror of
https://github.com/simonw/datasette.git
synced 2025-12-10 16:51:24 +01:00
21 lines
923 B
Python
21 lines
923 B
Python
from .fixtures import app_client
|
|
|
|
|
|
def test_auth_token(app_client):
|
|
"The /-/auth-token endpoint sets the correct cookie"
|
|
assert app_client.ds._root_token is not None
|
|
path = "/-/auth-token?token={}".format(app_client.ds._root_token)
|
|
response = app_client.get(path, allow_redirects=False,)
|
|
assert 302 == response.status
|
|
assert "/" == response.headers["Location"]
|
|
assert {"id": "root"} == app_client.ds.unsign(response.cookies["ds_actor"], "actor")
|
|
# Check that a second with same token fails
|
|
assert app_client.ds._root_token is None
|
|
assert 403 == app_client.get(path, allow_redirects=False,).status
|
|
|
|
|
|
def test_actor_cookie(app_client):
|
|
"A valid actor cookie sets request.scope['actor']"
|
|
cookie = app_client.ds.sign({"id": "test"}, "actor")
|
|
response = app_client.get("/", cookies={"ds_actor": cookie})
|
|
assert {"id": "test"} == app_client.ds._last_request.scope["actor"]
|