From a2dca62360ad4a961d4c46f68eae41b7d5c7b2c9 Mon Sep 17 00:00:00 2001 From: Simon Willison Date: Sat, 31 Dec 2022 11:21:15 -0800 Subject: [PATCH] Fix for extension tests I broke, refs #782 --- tests/test_load_extensions.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/tests/test_load_extensions.py b/tests/test_load_extensions.py index 360bc8f3..0e39f566 100644 --- a/tests/test_load_extensions.py +++ b/tests/test_load_extensions.py @@ -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"