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

migrate db.get_all to pure SQLAlchemy

Browse source
This commit is contained in:
ducklet 2023-03-19 22:36:33 +01:00
parent 5015815097
commit 1dd7bab4aa
2 changed files with 45 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

36
tests/test_db.py
View file

@ -5,6 +5,42 @@ import pytest
from unwind import db, models, web_models
@pytest.mark.asyncio
async def test_get_all(shared_conn: db.Database):
async with shared_conn.transaction(force_rollback=True):
m1 = models.Movie(
title="test movie",
release_year=2013,
media_type="Movie",
imdb_id="tt0000000",
genres={"genre-1"},
)
await db.add(m1)
m2 = models.Movie(
title="test movie",
release_year=2013,
media_type="Movie",
imdb_id="tt0000001",
genres={"genre-1"},
)
await db.add(m2)
m3 = models.Movie(
title="test movie",
release_year=2014,
media_type="Movie",
imdb_id="tt0000002",
genres={"genre-1"},
)
await db.add(m3)
assert [] == list(await db.get_all(models.Movie, id="blerp"))
assert [m1] == list(await db.get_all(models.Movie, id=str(m1.id)))
assert [m1, m2] == list(await db.get_all(models.Movie, release_year=2013))
assert [m1, m2, m3] == list(await db.get_all(models.Movie))
@pytest.mark.asyncio
async def test_add_and_get(shared_conn: db.Database):
async with shared_conn.transaction(force_rollback=True):