add async db

MorvanZhou · Nov 18, 2023 · bff53f0 · bff53f0
1 parent c1dd00c
commit bff53f0
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/tests/test_auth.py b/tests/test_auth.py
@@ -8,7 +8,7 @@
 from . import utils
 
 
-class AuthTest(unittest.TestCase):
+class AuthTest(unittest.IsolatedAsyncioTestCase):
     @classmethod
     def setUpClass(cls):
         utils.set_env(".env.test.local")
@@ -17,9 +17,12 @@ def setUpClass(cls):
 
     @classmethod
     def tearDownClass(cls) -> None:
-        database.drop_all()
         utils.drop_env(".env.test.local")
 
+    async def asyncSetUp(self) -> None:
+        await database.drop_all()
+        await database.init()
+
     def test_verify(self):
         password = "123abc"
         email = "[email protected]"