保存今天的修改

tests/test_dev_integration.py

@@ -10,6 +10,7 @@ from unittest.mock import patch
 from openpyxl import load_workbook
 
 from logistics_chat import create_app, db
+from staff_fixture import login_staff
 
 
 class DevIntegrationTests(unittest.TestCase):
@@ -22,6 +23,7 @@ class DevIntegrationTests(unittest.TestCase):
             'DEEPSEEK_API_KEY': '', 'RATE_LIMIT_PER_MIN': 1000})
         self.client = self.app.test_client()
         self.client.post('/api/visitor/session', json={'key': '1' * 64})
+        login_staff(self.app, self.client)
 
     def session(self):
         cid = self.client.post('/api/new_chat').json['id']
@@ -130,6 +132,7 @@ class DevIntegrationTests(unittest.TestCase):
             connection.commit()
             connection.close()
             db.init_db()
+        login_staff(self.app, self.client, 'demo', seed=False)
         results = self.client.get('/api/search_chat?q=专线').json
         self.assertEqual(len(results), 1)
         self.assertIn('SUM', results[0]['content'])
@@ -160,7 +163,7 @@ class DevIntegrationTests(unittest.TestCase):
         state = self.client.get('/api/human/poll?sid=1').json
         self.assertEqual(state['messages'][0]['content'], '保留的回复')
         self.assertEqual(state['note'], '')
-        self.assertEqual(self.client.get('/api/agents/list').json[0]['username'], 'fixture-agent')
+        self.assertIn('fixture-agent', [row['username'] for row in self.client.get('/api/agents/list').json])
 
 
 if __name__ == '__main__':