diff --git a/test/test_login_reg.py b/test/test_login_reg.py
index bd689fe..516232f 100644
--- a/test/test_login_reg.py
+++ b/test/test_login_reg.py
@@ -69,7 +69,7 @@ class TestLoginReg(unittest.TestCase):
         # 发送登录请求
         response = client.post("/api/s1/login", json={"username": "testuser", "password": "testpassword"})
         self.assertEqual(response.status_code, 200)
-        self.assertEqual(response.json(), {"message": "Login successful"})
+        self.assertEqual(response.json(), {'message': 'Login successful', 'role': 1})
 
     def test_register(self):
         # 发送注册请求