datasette/tests/test_query_form_action.py
2026-09-14 06:03:21 -05:00

53 lines
1.8 KiB
Python

import pytest
from bs4 import BeautifulSoup as Soup
@pytest.mark.asyncio
async def test_ad_hoc_query_form_targets_query_route(ds_client):
response = await ds_client.get("/fixtures/-/query?sql=select+1")
assert response.status_code == 200
form = Soup(response.text, "html.parser").select_one("form.sql.core")
assert form is not None
assert form["action"] == "/fixtures/-/query"
def test_ad_hoc_query_form_action_preserves_base_url(app_client_base_url_prefix):
response = app_client_base_url_prefix.get("/prefix/fixtures/-/query?sql=select+1")
assert response.status_code == 200
form = Soup(response.text, "html.parser").select_one("form.sql.core")
assert form is not None
assert form["action"] == "/prefix/fixtures/-/query"
@pytest.mark.asyncio
async def test_stored_query_form_action_encodes_query_name(ds_client):
query_name = "query with spaces"
await ds_client.ds.update_query("fixtures", query_name, sql="select 1")
query_url = ds_client.ds.urls.query("fixtures", query_name)
response = await ds_client.get(query_url)
assert response.status_code == 200
form = Soup(response.text, "html.parser").select_one("form.sql.core")
assert form is not None
assert form["action"] == query_url
@pytest.mark.asyncio
async def test_stored_query_form_action_preserves_base_url(
app_client_base_url_prefix,
):
query_name = "query with spaces"
await app_client_base_url_prefix.ds.update_query(
"fixtures", query_name, sql="select 1"
)
query_url = app_client_base_url_prefix.ds.urls.query("fixtures", query_name)
response = app_client_base_url_prefix.get(query_url)
assert response.status_code == 200
form = Soup(response.text, "html.parser").select_one("form.sql.core")
assert form is not None
assert form["action"] == query_url