Add yield1

2021-03-12 18:22:59 +07:00
parent ff9d412508
commit 1c63b1c80f
1 changed files with 4 additions and 2 deletions
--- a/tests/test_default_relation_order.py
+++ b/tests/test_default_relation_order.py
@ -72,8 +72,10 @@ def create_test_database():

@pytest.fixture(autouse=True, scope="function")
 async def cleanup():
-    await Book.objects.delete(each=True)
-    await Author.objects.delete(each=True)
+    yield
+    async with database:
+        await Book.objects.delete(each=True)
+        await Author.objects.delete(each=True)


@pytest.mark.asyncio