Fix for extension tests I broke, refs #782

pull/1974/head
Simon Willison 2022-12-31 11:21:15 -08:00
rodzic ca07fff3e2
commit a2dca62360
1 zmienionych plików z 6 dodań i 6 usunięć

Wyświetl plik

@ -25,15 +25,15 @@ async def test_load_extension_default_entrypoint():
# should fail.
ds = Datasette(sqlite_extensions=[COMPILED_EXTENSION_PATH])
response = await ds.client.get("/_memory.json?sql=select+a()")
response = await ds.client.get("/_memory.json?_shape=arrays&sql=select+a()")
assert response.status_code == 200
assert response.json()["rows"][0][0] == "a"
response = await ds.client.get("/_memory.json?sql=select+b()")
response = await ds.client.get("/_memory.json?_shape=arrays&sql=select+b()")
assert response.status_code == 400
assert response.json()["error"] == "no such function: b"
response = await ds.client.get("/_memory.json?sql=select+c()")
response = await ds.client.get("/_memory.json?_shape=arrays&sql=select+c()")
assert response.status_code == 400
assert response.json()["error"] == "no such function: c"
@ -52,14 +52,14 @@ async def test_load_extension_multiple_entrypoints():
]
)
response = await ds.client.get("/_memory.json?sql=select+a()")
response = await ds.client.get("/_memory.json?_shape=arrays&sql=select+a()")
assert response.status_code == 200
assert response.json()["rows"][0][0] == "a"
response = await ds.client.get("/_memory.json?sql=select+b()")
response = await ds.client.get("/_memory.json?_shape=arrays&sql=select+b()")
assert response.status_code == 200
assert response.json()["rows"][0][0] == "b"
response = await ds.client.get("/_memory.json?sql=select+c()")
response = await ds.client.get("/_memory.json?_shape=arrays&sql=select+c()")
assert response.status_code == 200
assert response.json()["rows"][0][0] == "c"