mirror of
https://github.com/simonw/datasette.git
synced 2026-11-03 08:57:39 +01:00
fix: return inserted rows from WITHOUT ROWID tables
Look up returned rows by explicit primary keys when SQLite tables have no rowid. Cover compound keys and STRICT table options. Refs #1873.
This commit is contained in:
parent
cec5e6b2ef
commit
35dc60041a
3 changed files with 65 additions and 0 deletions
|
|
@ -438,6 +438,55 @@ async def test_insert_row(ds_write, content_type):
|
|||
assert not event.replace
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
@pytest.mark.parametrize(
|
||||
"table_options",
|
||||
(
|
||||
"without rowid",
|
||||
pytest.param(
|
||||
"without rowid, strict",
|
||||
marks=pytest.mark.skipif(
|
||||
sqlite3.sqlite_version_info < (3, 37, 0),
|
||||
reason="STRICT tables require SQLite 3.37",
|
||||
),
|
||||
),
|
||||
),
|
||||
)
|
||||
async def test_insert_row_into_without_rowid_table(ds_write, table_options):
|
||||
db = ds_write.get_database("data")
|
||||
await db.execute_write(
|
||||
"create table compound_without_rowid (a text, b text, "
|
||||
f"primary key (a, b)) {table_options}"
|
||||
)
|
||||
response = await ds_write.client.post(
|
||||
"/data/compound_without_rowid/-/insert",
|
||||
json={"row": {"a": "first", "b": "second"}},
|
||||
headers=_headers(write_token(ds_write)),
|
||||
)
|
||||
assert response.status_code == 201, response.json()
|
||||
assert response.json()["rows"] == [{"a": "first", "b": "second"}]
|
||||
assert (await db.execute("select a, b from compound_without_rowid")).dicts() == [
|
||||
{"a": "first", "b": "second"}
|
||||
]
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_insert_rows_into_without_rowid_table_preserves_return_order(ds_write):
|
||||
db = ds_write.get_database("data")
|
||||
await db.execute_write(
|
||||
"create table compound_without_rowid (a text, b text, "
|
||||
"primary key (a, b)) without rowid"
|
||||
)
|
||||
rows = [{"a": "z", "b": "first"}, {"a": "a", "b": "second"}]
|
||||
response = await ds_write.client.post(
|
||||
"/data/compound_without_rowid/-/insert",
|
||||
json={"rows": rows, "return": True},
|
||||
headers=_headers(write_token(ds_write)),
|
||||
)
|
||||
assert response.status_code == 201, response.json()
|
||||
assert response.json()["rows"] == rows
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_insert_row_alter(ds_write):
|
||||
token = write_token(ds_write)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue