diff options
Diffstat (limited to '')
| -rw-r--r-- | tests/api/test_db.py | 23 |
1 files changed, 12 insertions, 11 deletions
diff --git a/tests/api/test_db.py b/tests/api/test_db.py index f53b90f..b030035 100644 --- a/tests/api/test_db.py +++ b/tests/api/test_db.py @@ -1,10 +1,16 @@ from datetime import datetime, timedelta, timezone +import pytest +from conftest import DB +from sqlalchemy.exc import StatementError +from sqlalchemy.orm import ( + Mapped, + mapped_column, +) + import hircine.db as database import hircine.db.models as models import hircine.db.ops as ops -import pytest -from conftest import DB from hircine.db.models import ( Artist, Base, @@ -16,11 +22,6 @@ from hircine.db.models import ( Tag, TagNamespaces, ) -from sqlalchemy.exc import StatementError -from sqlalchemy.orm import ( - Mapped, - mapped_column, -) class Date(MixinID, Base): @@ -66,8 +67,8 @@ async def test_models_retained_when_clearing_association( comic = await DB.add(comic) async with database.session() as s: - object = await s.get(Comic, comic.id) - setattr(object, key, []) + obj = await s.get(Comic, comic.id) + setattr(obj, key, []) await s.commit() assert await DB.get(assoccls, (comic.id, model.id)) is None @@ -86,8 +87,8 @@ async def test_models_retained_when_clearing_comictag(empty_comic): await DB.add(ct) async with database.session() as s: - object = await s.get(Comic, comic.id) - object.tags = [] + obj = await s.get(Comic, comic.id) + obj.tags = [] await s.commit() assert await DB.get(ComicTag, (comic.id, ct.namespace_id, ct.tag_id)) is None |