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

Add pk column if missing from insert

Browse source
This commit is contained in:
Simon Willison 2019-07-23 13:12:48 +02:00
commit e9d983e561
2 changed files with 7 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

5
tests/test_create.py
View file

@ -734,3 +734,8 @@ def test_cannot_provide_both_filename_and_memory():
AssertionError, match="Either specify a filename_or_conn or pass memory=True"
):
Database("/tmp/foo.db", memory=True)
def test_creates_id_column(fresh_db):
last_pk = fresh_db.table("cats", pk="id").insert({"name": "barry"}).last_pk
assert [{"name": "barry", "id": last_pk}] == list(fresh_db["cats"].rows)