创建 TestGetMenu.py

app/tests/TestGetMenu.py

@@ -0,0 +1,44 @@
+import unittest
+
+from app.views import *
+from flask import session
+from unittest.mock import patch  # 如果需要模拟数据库方法
+
+
+class MenuAPITestCase(unittest.TestCase):
+
+    def setUp(self):
+        app.testing = True  # 设置 Flask 应用为测试模式
+        self.client = app.test_client()  # 创建一个测试客户端
+        self.expected_menu = [
+            {"name": "课程类别", "path": "/course-category"},
+            {"name": "课程信息", "path": "/course-info"},
+            {"name": "课程签到", "path": "/attendance-teacher"},
+            {"name": "签到提醒", "path": "/attendance-reminder"}
+        ]
+
+    def test_menu_no_login(self):
+        # 测试未登录情况下请求API
+        response = self.client.get('/api/menu')
+        self.assertEqual(response.status_code, 401)
+        self.assertEqual(response.get_json(), [])
+
+    def test_login(self):
+        with self.client:
+            self.client.post('/login', data=dict(
+                number='G0001',
+                password='1'
+            ))
+
+            # 确认登录后session中有数据
+            self.assertIn('role', session)
+
+            # 执行logout
+            response = self.client.get('/api/menu')
+            print(response.get_json())
+            # 确认响应是重定向到登录页面
+            self.assertEqual(response.status_code, 200)
+            self.assertEqual(self.expected_menu,response.get_json())
+
+if __name__ == '__main__':
+    unittest.main()