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

Fix for extension tests I broke, refs #782

Browse source
This commit is contained in:
Simon Willison 2022-12-31 11:21:15 -08:00
commit a2dca62360
1 changed files with 6 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

12
tests/test_load_extensions.py
View file

@ -25,15 +25,15 @@ async def test_load_extension_default_entrypoint():
# should fail. # should fail.
ds = Datasette(sqlite_extensions=[COMPILED_EXTENSION_PATH]) 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.status_code == 200
assert response.json()["rows"][0][0] == "a" 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.status_code == 400
assert response.json()["error"] == "no such function: b" 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.status_code == 400
assert response.json()["error"] == "no such function: c" 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.status_code == 200
assert response.json()["rows"][0][0] == "a" 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.status_code == 200
assert response.json()["rows"][0][0] == "b" 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.status_code == 200
assert response.json()["rows"][0][0] == "c" assert response.json()["rows"][0][0] == "c"