test: fix test_awaitable_relation_field

2jun0 · Mar 28, 2024 · 13eca1c · 13eca1c
1 parent 7897857
commit 13eca1c
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/tests/test_awaitable_field.py b/tests/test_awaitable_field.py
@@ -65,9 +65,7 @@ class Hero(AsyncSQLModel, table=True):
         await session.commit()
 
     async with AsyncSession(engine) as session:
-        hero = (
-            await session.exec(select(Hero).where(Hero.id == hero_rusty_man.id))
-        ).one()
+        hero = (await session.exec(select(Hero).where(Hero.name == "Rusty-Man"))).one()
 
         # loading lazy loading attribute will raise MissingGreenlet error
         with pytest.raises(MissingGreenlet):