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

migrate db.add, db.update, db.remove to SQLA

Browse source
This commit is contained in:
ducklet 2023-03-23 22:49:17 +01:00
parent 6f6354cfac
commit b91fcd3f55
2 changed files with 34 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

25
tests/test_db.py
View file

@ -105,6 +105,31 @@ async def test_add_and_get(shared_conn: db.Database):
assert m2 == await db.get(models.Movie, id=str(m2.id))
@pytest.mark.asyncio
async def test_update(shared_conn: db.Database):
async with shared_conn.transaction(force_rollback=True):
m = a_movie()
await db.add(m)
assert m == await db.get(models.Movie, id=str(m.id))
m.title += "something else"
assert m != await db.get(models.Movie, id=str(m.id))
await db.update(m)
assert m == await db.get(models.Movie, id=str(m.id))
@pytest.mark.asyncio
async def test_remove(shared_conn: db.Database):
async with shared_conn.transaction(force_rollback=True):
m1 = a_movie()
await db.add(m1)
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))
@pytest.mark.asyncio
async def test_find_ratings(shared_conn: db.Database):
async with shared_conn.transaction(force_rollback=True):