ducklet/unwind
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

improve strict typing

Browse source
This commit is contained in:
ducklet 2023-07-22 19:37:01 +02:00
parent 86c3030e31
commit 25f31db756
4 changed files with 34 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

16
tests/test_db.py
View file

@ -38,21 +38,25 @@ async def test_get(shared_conn: db.Database):
m2 = a_movie(release_year=m1.release_year + 1)
await db.add(m2)
assert None == await db.get(models.Movie)
assert None == await db.get(models.Movie, id="blerp")
assert None is await db.get(models.Movie)
assert None is await db.get(models.Movie, id="blerp")
assert m1 == await db.get(models.Movie, id=str(m1.id))
assert m2 == await db.get(models.Movie, release_year=m2.release_year)
assert None == await db.get(
assert None is await db.get(
models.Movie, id=str(m1.id), release_year=m2.release_year
)
assert m2 == await db.get(
models.Movie, id=str(m2.id), release_year=m2.release_year
)
assert m1 == await db.get(
models.Movie, media_type=m1.media_type, order_by=("release_year", "asc")
models.Movie,
media_type=m1.media_type,
order_by=(models.movies.c.release_year, "asc"),
)
assert m2 == await db.get(
models.Movie, media_type=m1.media_type, order_by=("release_year", "desc")
models.Movie,
media_type=m1.media_type,
order_by=(models.movies.c.release_year, "desc"),
)
@ -136,7 +140,7 @@ async def test_remove(shared_conn: db.Database):
assert m1 == await db.get(models.Movie, id=str(m1.id))
await db.remove(m1)
assert None == await db.get(models.Movie, id=str(m1.id))
assert None is await db.get(models.Movie, id=str(m1.id))
@pytest.mark.asyncio