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

Compound primary key support for /db/-/create - closes #1911

Browse source 
Needed for tests in #1864
This commit is contained in:
Simon Willison 2022-11-29 10:47:46 -08:00
commit 1154048f79
3 changed files with 69 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

54
tests/test_api_write.py
View file

@ -643,6 +643,27 @@ async def test_drop_table(ds_write, scenario):
"row_count": 1,
},
),
# Create table with compound primary key
(
{
"table": "five",
"row": {"type": "article", "key": 123, "title": "Article 1"},
"pks": ["type", "key"],
},
201,
{
"ok": True,
"database": "data",
"table": "five",
"table_url": "http://localhost/data/five",
"table_api_url": "http://localhost/data/five.json",
"schema": (
"CREATE TABLE [five] (\n [type] TEXT,\n [key] INTEGER,\n"
" [title] TEXT,\n PRIMARY KEY ([type], [key])\n)"
),
"row_count": 1,
},
),
# Error: Table is required
(
{
@ -799,6 +820,39 @@ async def test_drop_table(ds_write, scenario):
"errors": ["pk must be a string"],
},
),
# Error: Cannot specify both pk and pks
(
{
"table": "bad",
"row": {"id": 1, "name": "Row 1"},
"pk": "id",
"pks": ["id", "name"],
},
400,
{
"ok": False,
"errors": ["Cannot specify both pk and pks"],
},
),
# Error: pks must be a list
(
{
"table": "bad",
"row": {"id": 1, "name": "Row 1"},
"pks": "id",
},
400,
{
"ok": False,
"errors": ["pks must be a list"],
},
),
# Error: pks must be a list of strings
(
{"table": "bad", "row": {"id": 1, "name": "Row 1"}, "pks": [1, 2]},
400,
{"ok": False, "errors": ["pks must be a list of strings"]},
),
),
)
async def test_create_table(ds_write, input, expected_status, expected_response):