解决已发现问题
This commit is contained in:
@@ -403,9 +403,10 @@ class AdminTaskMonitorTests(TestCase):
|
||||
]
|
||||
self.assertTrue(task_selects)
|
||||
for sql in task_selects:
|
||||
self.assertNotIn("request_payload", sql)
|
||||
# 仅允许从 JSON 中提取 feature 做「全能创作」分类,不能把整份 Prompt/响应/报错读进列表页。
|
||||
self.assertNotIn("response_payload", sql)
|
||||
self.assertNotIn("error_message", sql)
|
||||
self.assertNotIn("x" * 100_000, str(response.data))
|
||||
|
||||
detail = self.ac.get(f"/api/admin/tasks/{self.t_ok.id}/")
|
||||
self.assertEqual(detail.status_code, 200)
|
||||
@@ -420,6 +421,26 @@ class AdminTaskMonitorTests(TestCase):
|
||||
self.assertIn(str(self.t_anom.id), ids)
|
||||
self.assertNotIn(str(self.t_ok.id), ids)
|
||||
|
||||
def test_omni_create_tasks_have_separate_category_and_filter(self):
|
||||
task = self.AITask.objects.create(
|
||||
team=self.team,
|
||||
model_config=self.mc,
|
||||
task_type=self.AITask.Type.FREE_VIDEO,
|
||||
status=self.AITask.Status.SUCCEEDED,
|
||||
estimated_cost="1.0",
|
||||
actual_cost="1.0",
|
||||
idempotency_key="k-omni-create",
|
||||
request_payload={"feature": "omni_create", "prompt": "完整出片指令"},
|
||||
)
|
||||
|
||||
all_rows = self.ac.get("/api/admin/tasks/").data["results"]
|
||||
row = next(item for item in all_rows if item["id"] == str(task.id))
|
||||
self.assertEqual(row["task_type"], "free_video")
|
||||
self.assertEqual(row["task_category"], "omni_create")
|
||||
|
||||
category_rows = self.ac.get("/api/admin/tasks/?category=omni_create").data["results"]
|
||||
self.assertEqual([item["id"] for item in category_rows], [str(task.id)])
|
||||
|
||||
def test_cost_anomaly_flag(self):
|
||||
self.assertTrue(self.ac.get(f"/api/admin/tasks/{self.t_anom.id}/").data["cost_anomaly"])
|
||||
self.assertFalse(self.ac.get(f"/api/admin/tasks/{self.t_ok.id}/").data["cost_anomaly"])
|
||||
|
||||
Reference in New Issue
Block a user