test_upsert_compound_primary_key

This commit is contained in:
Simon Willison 2019-12-29 21:31:03 -08:00
commit 468d51314a

View file

@ -14,3 +14,13 @@ def test_upsert_all(fresh_db):
{"id": 2, "name": "Nixie", "age": 5},
] == list(table.rows)
assert 2 == table.last_pk
def test_upsert_compound_primary_key(fresh_db):
table = fresh_db["table"]
table.upsert_all([{"species": "dog", "id": 1, "name": "Cleo", "age": 4}, {"species": "cat", "id": 1, "name": "Catbag"}], pk=("species", "id"))
table.upsert_all([{"species": "dog", "id": 1, "age": 5}], pk=("species", "id"))
assert [
{"species": "dog", "id": 1, "name": "Cleo", "age": 5},
{"species": "cat", "id": 1, "name": "Catbag", "age": None},
] == list(table.rows)