remote/sqlite-utils
1
0
Fork 0
mirror of https://github.com/simonw/sqlite-utils.git synced 2026-09-22 00:14:15 +02:00
Code Issues Projects Releases Packages Wiki Activity

Test transform rowids against different table types, refs #592

Browse source
This commit is contained in:
Simon Willison 2023-09-08 17:43:19 -07:00
commit b86374f705
1 changed files with 10 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

12
tests/test_transform.py
View file

@ -500,14 +500,22 @@ def test_transform_replace_foreign_keys(fresh_db, foreign_keys):
) )
def test_transform_preserves_rowids(fresh_db): @pytest.mark.parametrize("table_type", ("id_pk", "rowid", "compound_pk"))
def test_transform_preserves_rowids(fresh_db, table_type):
pk = None
if table_type == "id_pk":
pk = "id"
elif table_type == "compound_pk":
pk = ("id", "name")
elif table_type == "rowid":
pk = None
fresh_db["places"].insert_all( fresh_db["places"].insert_all(
[ [
{"id": "1", "name": "Paris", "country": "France"}, {"id": "1", "name": "Paris", "country": "France"},
{"id": "2", "name": "London", "country": "UK"}, {"id": "2", "name": "London", "country": "UK"},
{"id": "3", "name": "New York", "country": "USA"}, {"id": "3", "name": "New York", "country": "USA"},
], ],
pk="id", pk=pk,
) )
# Now delete and insert a row to mix up the `rowid` sequence # Now delete and insert a row to mix up the `rowid` sequence
fresh_db["places"].delete_where("id = ?", ["2"]) fresh_db["places"].delete_where("id = ?", ["2"])