fix test_default_admin tests

moeflow-com · Jun 9, 2024 · f6fe9a1 · f6fe9a1
1 parent b7b915e
commit f6fe9a1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/app/factory.py b/app/factory.py
@@ -79,7 +79,7 @@ def create_celery(app: Flask) -> Celery:
     return created
 
 
-def create_or_override_default_admin(app):
+def create_or_override_default_admin(app: Flask):
     """创建或覆盖默认管理员"""
     from app.models.user import User
 

diff --git a/tests/base/test_default_admin.py b/tests/base/test_default_admin.py
@@ -1,4 +1,5 @@
 from app import create_app
+from app.factory import init_db
 from app.models.user import User
 from tests import MoeTestCase
 
@@ -20,7 +21,7 @@ def test_reset_default_admin(self):
         admin_user.save()
         admin_user.reload()
         self.assertEqual(admin_user.admin, False)
-        create_app()
+        init_db(create_app())
         admin_user.reload()
         self.assertEqual(admin_user.admin, True)
         # 测试其他用户权限不受影响
@@ -34,7 +35,7 @@ def test_reset_default_admin_when_true(self):
         self.assertEqual(admin_user.email, self.app.config["ADMIN_EMAIL"])
         self.assertEqual(admin_user.admin, True)
         self.assertEqual(user.admin, False)
-        create_app()
+        init_db(create_app())
         admin_user.reload()
         self.assertEqual(admin_user.admin, True)
         # 测试其他用户权限不受影响