test: 全面迁移 pytest 并加入格式门禁

tests/test_user_deletion.py

@@ -1,4 +1,4 @@
-import unittest
+import pytest
 from types import SimpleNamespace
 
 from sqlalchemy import create_engine, event
@@ -10,8 +10,8 @@ from web.backend.routers.users import delete_user, list_users, rename_user
 from web.backend.schemas import UserRename
 
 
-class UserDeletionTests(unittest.TestCase):
-    def setUp(self):
+class TestUserDeletion:
+    def setup_method(self):
         self.engine = create_engine("sqlite://")
 
         @event.listens_for(self.engine, "connect")
@@ -24,23 +24,27 @@ class UserDeletionTests(unittest.TestCase):
         self.user = User(username="support", password_hash="hash", role="support")
         self.session.add_all([self.admin, self.user])
         self.session.commit()
-        self.session.add(Account(
-            username="douyu-user",
-            password="password",
-            email="mail@example.com",
-            email_password="mail-password",
-            assigned_to=self.user.id,
-        ))
+        self.session.add(
+            Account(
+                username="douyu-user",
+                password="password",
+                email="mail@example.com",
+                email_password="mail-password",
+                assigned_to=self.user.id,
+            )
+        )
         self.session.commit()
         account = self.session.query(Account).one()
-        self.session.add(LoginTask(
-            batch_id="batch-1",
-            account_id=account.id,
-            created_by=self.user.id,
-        ))
+        self.session.add(
+            LoginTask(
+                batch_id="batch-1",
+                account_id=account.id,
+                created_by=self.user.id,
+            )
+        )
         self.session.commit()
 
-    def tearDown(self):
+    def teardown_method(self):
         self.session.close()
         Base.metadata.drop_all(self.engine)
         self.engine.dispose()
@@ -54,19 +58,19 @@ class UserDeletionTests(unittest.TestCase):
 
         deleted = self.session.get(User, self.user.id)
         account = self.session.query(Account).one()
-        self.assertTrue(result["success"])
-        self.assertEqual(result["released_accounts"], 1)
-        self.assertFalse(deleted.is_active)
-        self.assertIsNotNone(deleted.deleted_at)
-        self.assertEqual(deleted.deleted_username, "support")
-        self.assertTrue(deleted.username.startswith(f"deleted-{deleted.id}-"))
-        self.assertIsNone(account.assigned_to)
-        self.assertEqual(self.session.query(AuditLog).one().target, "support")
-        self.assertEqual(self.session.query(LoginTask).one().created_by, deleted.id)
-        self.assertEqual([item.id for item in list_users(db=self.session)], [self.admin.id])
+        assert result["success"]
+        assert result["released_accounts"] == 1
+        assert not deleted.is_active
+        assert deleted.deleted_at is not None
+        assert deleted.deleted_username == "support"
+        assert deleted.username.startswith(f"deleted-{deleted.id}-")
+        assert account.assigned_to is None
+        assert self.session.query(AuditLog).one().target == "support"
+        assert self.session.query(LoginTask).one().created_by == deleted.id
+        assert [item.id for item in list_users(db=self.session)] == [self.admin.id]
 
     def test_cannot_delete_current_user(self):
-        with self.assertRaisesRegex(Exception, "不能删除当前登录用户"):
+        with pytest.raises(Exception, match="不能删除当前登录用户"):
             delete_user(
                 self.user.id,
                 db=self.session,
@@ -84,12 +88,8 @@ class UserDeletionTests(unittest.TestCase):
         renamed = self.session.get(User, self.user.id)
         account = self.session.query(Account).one()
         task = self.session.query(LoginTask).one()
-        self.assertEqual(result.username, "renamed-support")
-        self.assertEqual(renamed.id, self.user.id)
-        self.assertEqual(account.assigned_to, self.user.id)
-        self.assertEqual(task.created_by, self.user.id)
-        self.assertEqual(self.session.query(AuditLog).one().target, "support -> renamed-support")
-
-
-if __name__ == "__main__":
-    unittest.main()
+        assert result.username == "renamed-support"
+        assert renamed.id == self.user.id
+        assert account.assigned_to == self.user.id
+        assert task.created_by == self.user.id
+        assert self.session.query(AuditLog).one().target == "support -> renamed-support"