mirror of
https://github.com/simonw/sqlite-utils.git
synced 2026-07-26 19:04:32 +02:00
69a1c0d960* Automated upgrades by Ruff
uvx --with 'ruff>=0.16.0' ruff check . --fix --unsafe-fixes
* Fix remaining Ruff errors with GPT-5.6 Sol high
https://gist.github.com/simonw/6da7906a9fea6e90da131c21a9055199
* Fix flake E501 long lines
* New Protocol for migrations to make ty happy
32 lines
931 B
Python
32 lines
931 B
Python
import pytest
|
|
|
|
from sqlite_utils.db import NotFoundError
|
|
|
|
|
|
def test_get_rowid(fresh_db):
|
|
dogs = fresh_db["dogs"]
|
|
cleo = {"name": "Cleo", "age": 4}
|
|
row_id = dogs.insert(cleo).last_rowid
|
|
assert cleo == dogs.get(row_id)
|
|
|
|
|
|
def test_get_primary_key(fresh_db):
|
|
dogs = fresh_db["dogs"]
|
|
cleo = {"name": "Cleo", "age": 4, "id": 5}
|
|
last_pk = dogs.insert(cleo, pk="id").last_pk
|
|
assert 5 == last_pk
|
|
assert cleo == dogs.get(5)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"argument,expected_msg",
|
|
[(100, None), (None, None), ((1, 2), "Need 1 primary key value"), ("2", None)],
|
|
)
|
|
def test_get_not_found(argument, expected_msg, fresh_db):
|
|
fresh_db["dogs"].insert(
|
|
{"id": 1, "name": "Cleo", "age": 4, "is_good": True}, pk="id"
|
|
)
|
|
with pytest.raises(NotFoundError) as excinfo:
|
|
fresh_db["dogs"].get(argument)
|
|
if expected_msg is not None:
|
|
assert expected_msg == excinfo.value.args[0]
|