Added table.delete(pk) method, refs #62

This commit is contained in:
Simon Willison 2019-11-04 08:07:44 -08:00
commit 19073d6d97
3 changed files with 39 additions and 0 deletions

14
tests/test_delete.py Normal file
View file

@ -0,0 +1,14 @@
def test_delete_rowid_table(fresh_db):
table = fresh_db["table"]
table.insert({"foo": 1}).last_pk
rowid = table.insert({"foo": 2}).last_pk
table.delete(rowid)
assert [{"foo": 1}] == list(table.rows)
def test_delete_pk_table(fresh_db):
table = fresh_db["table"]
table.insert({"id": 1}, pk="id")
table.insert({"id": 2}, pk="id")
table.delete(1)
assert [{"id": 2}] == list(table.rows)