type: 收窄测试与响应模型类型

tests/test_user_deletion.py

@@ -1,6 +1,4 @@
 import pytest
-from types import SimpleNamespace
-
 from sqlalchemy import create_engine, event
 from sqlalchemy.orm import sessionmaker
 
@@ -53,11 +51,12 @@ class TestUserDeletion:
         result = delete_user(
             self.user.id,
             db=self.session,
-            current=SimpleNamespace(id=self.admin.id, username=self.admin.username),
+            current=self.admin,
         )
 
         deleted = self.session.get(User, self.user.id)
         account = self.session.query(Account).one()
+        assert deleted is not None
         assert result["success"]
         assert result["released_accounts"] == 1
         assert not deleted.is_active
@@ -74,7 +73,7 @@ class TestUserDeletion:
             delete_user(
                 self.user.id,
                 db=self.session,
-                current=SimpleNamespace(id=self.user.id, username=self.user.username),
+                current=self.user,
             )
 
     def test_rename_preserves_user_id_and_related_data(self):
@@ -82,12 +81,13 @@ class TestUserDeletion:
             self.user.id,
             UserRename(username="renamed-support"),
             db=self.session,
-            current=SimpleNamespace(id=self.admin.id, username=self.admin.username),
+            current=self.admin,
         )
 
         renamed = self.session.get(User, self.user.id)
         account = self.session.query(Account).one()
         task = self.session.query(LoginTask).one()
+        assert renamed is not None
         assert result.username == "renamed-support"
         assert renamed.id == self.user.id
         assert account.assigned_to == self.user.id