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

Unit tests for .update()

Browse source 
Also now set .last_pk to lastrowid for rowid tables.
This commit is contained in:
Simon Willison 2019-07-28 15:30:28 +03:00
commit 455071f3c5
2 changed files with 25 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

24
tests/test_update.py Normal file
View file

@ -0,0 +1,24 @@
import pytest
def test_update_rowid_table(fresh_db):
table = fresh_db["table"]
rowid = table.insert({"foo": "bar"}).last_pk
table.update(rowid, {"foo": "baz"})
assert [{"foo": "baz"}] == list(table.rows)
def test_update_pk_table(fresh_db):
table = fresh_db["table"]
pk = table.insert({"foo": "bar", "id": 5}, pk="id").last_pk
assert 5 == pk
table.update(pk, {"foo": "baz"})
assert [{"id": 5, "foo": "baz"}] == list(table.rows)
def test_update_compound_pk_table(fresh_db):
table = fresh_db["table"]
pk = table.insert({"id1": 5, "id2": 3, "v": 1}, pk=("id1", "id2")).last_pk
assert (5, 3) == pk
table.update(pk, {"v": 2})
assert [{"id1": 5, "id2": 3, "v": 2}] == list(table.rows)