diff --git a/tests/test_get.py b/tests/test_get.py new file mode 100644 index 0000000..0b2d714 --- /dev/null +++ b/tests/test_get.py @@ -0,0 +1,22 @@ +from sqlite_utils.db import Database, RowNotFound +import pytest + + +def test_get_single_pk(fresh_db): + cleo = {"id": 1, "name": "Cleo", "age": 4} + table = fresh_db["dogs"].insert(cleo, pk="id") + with pytest.raises(RowNotFound): + table.get(2) + with pytest.raises(RowNotFound): + table.get(None) + assert cleo == table.get(1) + + +def test_get_compound_pk(fresh_db): + cleo = {"id1": 1, "id2": 1, "name": "Cleo", "age": 4} + table = fresh_db["dogs"].insert(cleo, pk=("id1", "id2")) + with pytest.raises(RowNotFound): + table.get(2) + with pytest.raises(RowNotFound): + table.get([2, 1]) + assert cleo == table.get([1, 1]) diff --git a/tests/test_update.py b/tests/test_update.py new file mode 100644 index 0000000..45018da --- /dev/null +++ b/tests/test_update.py @@ -0,0 +1,9 @@ +from sqlite_utils.db import Database + + +def test_update(fresh_db): + cleo = {"id": 1, "name": "Cleo", "age": 4} + fresh_db["dogs"].insert(cleo, pk="id") + assert [cleo] == list(fresh_db["dogs"].rows) + fresh_db["dogs"].update(1, {"age": 5}) + assert [{"id": 1, "name": "Cleo", "age": 5}] == list(fresh_db["dogs"].rows)