baodan/tests/generation_routes_test.py

37 lines
1.1 KiB
Python
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

"""统一任务接口响应结构回归测试。"""
from pathlib import Path
import sys
from flask import Flask
sys.path.insert(0, str(Path(__file__).resolve().parents[1] / "api"))
from insurance.generation import task_service
from insurance.generation.routes import workspace_bp
def test_task_detail_returns_task_as_direct_data(monkeypatch):
"""任务详情不能重复包裹 code/data否则前端轮询读不到 status。"""
monkeypatch.setattr(
task_service,
"get_task",
lambda task_id, user_id: {
"code": 0,
"data": {"id": task_id, "userId": user_id, "status": "running"},
},
)
app = Flask(__name__)
app.config["GUEST_MODE"] = True
app.register_blueprint(workspace_bp, url_prefix="/insurance/workspace")
response = app.test_client().get(
"/insurance/workspace/tasks/task-1",
headers={"Authorization": "Bearer guest_test"},
)
assert response.status_code == 200
body = response.get_json()
assert body["data"]["status"] == "running"
assert "data" not in body["data"]