Restore YG rules generation flow
This commit is contained in:
66
tests/test_route_messages.py
Normal file
66
tests/test_route_messages.py
Normal file
@@ -0,0 +1,66 @@
|
||||
import unittest
|
||||
from unittest.mock import patch
|
||||
|
||||
from app import create_app
|
||||
|
||||
|
||||
class RouteMessageTest(unittest.TestCase):
|
||||
def setUp(self):
|
||||
self.app = create_app("testing")
|
||||
self.client = self.app.test_client()
|
||||
|
||||
def test_success_messages_include_business_action(self):
|
||||
response = self.client.get("/api/health")
|
||||
|
||||
self.assertEqual(response.status_code, 200)
|
||||
self.assertEqual(response.get_json()["message"], "健康检查成功")
|
||||
|
||||
def test_failure_messages_include_business_action(self):
|
||||
response = self.client.get("/api/rules/status")
|
||||
|
||||
self.assertEqual(response.status_code, 400)
|
||||
self.assertEqual(response.get_json()["message"], "查询失败:缺少 task_id")
|
||||
|
||||
def test_clear_domains_message_is_delete_success(self):
|
||||
with patch("app.routes.domain.DomainStorage") as domain_storage_cls:
|
||||
with patch("app.routes.domain.SchemaStorage") as schema_storage_cls:
|
||||
domain_storage_cls.return_value.clear_all.return_value = 3
|
||||
schema_storage_cls.return_value.delete_file.return_value = True
|
||||
|
||||
response = self.client.post("/api/domains/clear")
|
||||
|
||||
self.assertEqual(response.status_code, 200)
|
||||
self.assertEqual(response.get_json()["message"], "删除成功")
|
||||
self.assertEqual(response.get_json()["data"]["deleted_count"], 3)
|
||||
|
||||
def test_generate_rules_returns_compact_async_task_info(self):
|
||||
with patch("app.routes.rules.RuleGenerationService") as service_cls:
|
||||
service_cls.return_value.start.return_value = {
|
||||
"task_id": "task-1",
|
||||
"status": "running",
|
||||
"limit": 3,
|
||||
"generated_count": 0,
|
||||
"output_file": "/app/output/rules-task-1/rules-task-1.xlsx",
|
||||
"markdown_file": "/app/output/rules-task-1/rules-task-1.md",
|
||||
"files": {
|
||||
"excel": "/app/output/rules-task-1/rules-task-1.xlsx",
|
||||
"markdown": "/app/output/rules-task-1/rules-task-1.md",
|
||||
},
|
||||
}
|
||||
|
||||
response = self.client.post("/api/rules/generate", json={"limit": 3})
|
||||
|
||||
body = response.get_json()
|
||||
self.assertEqual(response.status_code, 200)
|
||||
self.assertEqual(body["message"], "规则生成任务创建成功")
|
||||
self.assertEqual(body["data"], {
|
||||
"task_id": "task-1",
|
||||
"status": "running",
|
||||
"async": True,
|
||||
"limit": 3,
|
||||
"status_url": "/api/rules/status?task_id=task-1",
|
||||
})
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
Reference in New Issue
Block a user