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

New upsert implementation, refs #66

Browse source
This commit is contained in:
Simon Willison 2019-12-29 21:03:43 -08:00
commit 84bcabd093
4 changed files with 156 additions and 48 deletions
Download patch file Download diff file Expand all files Collapse all files

16
tests/test_upsert.py Normal file
View file

@ -0,0 +1,16 @@
def test_upsert(fresh_db):
table = fresh_db["table"]
table.insert({"id": 1, "name": "Cleo"}, pk="id")
table.upsert({"id": 1, "age": 5}, pk="id", alter=True)
assert [{"id": 1, "name": "Cleo", "age": 5}] == list(table.rows)
def test_upsert_all(fresh_db):
table = fresh_db["table"]
table.upsert_all([{"id": 1, "name": "Cleo"}, {"id": 2, "name": "Nixie"}], pk="id")
table.upsert_all([{"id": 1, "age": 5}, {"id": 2, "age": 5}], pk="id", alter=True)
assert [
{"id": 1, "name": "Cleo", "age": 5},
{"id": 2, "name": "Nixie", "age": 5},
] == list(table.rows)
assert 2 == table.last_pk