T-570 调整 cmhub 生图 submit 超时

2026-07-09 14:30:53 +08:00
parent 4e98a86fb3
commit bd99a0d5b7
5 changed files with 60 additions and 4 deletions
@@ -1155,11 +1155,13 @@ class AITests(TempDirMixin, unittest.TestCase):
            ]
            poll_db_values = []
            headers_seen = []
+            submit_timeouts = []
            downloads = []

            def fake_request(method, url, **kwargs):
                if str(method).upper() == "POST":
                    headers_seen.append(dict(kwargs["headers"]))
+                    submit_timeouts.append(kwargs.get("timeout"))
                    return _RequestsResponse(
                        {
                            "task_id": "cmhub-task-1",
@@ -1200,6 +1202,7 @@ class AITests(TempDirMixin, unittest.TestCase):
            self.assertEqual(["cmhub-task-1"], poll_db_values)
            self.assertTrue(headers_seen[0]["Idempotency-Key"].startswith("cmshopee-task-"))
            self.assertTrue(headers_seen[0]["X-Client-Version"])
+            self.assertEqual([(3, 36)], submit_timeouts)
            self.assertEqual(
                "https://cmhub.example.com/generated/images/generated.png",
                downloads[0][0],