fix: align cmhub image timeout display

This commit is contained in:
chengma
2026-07-07 19:13:00 +08:00
parent 95b180e9cf
commit 2928624019
13 changed files with 67 additions and 29 deletions
+3 -1
View File
@@ -408,8 +408,9 @@ class AITests(TempDirMixin, unittest.TestCase):
self.assertEqual("1K", payload["resolution"])
self.assertEqual("1:1", payload["aspect_ratio"])
self.assertTrue(payload["image_base64"].startswith("data:image/jpeg;base64,"))
self.assertEqual((3, 650), calls[0][2]["timeout"])
self.assertEqual("https://cdn.example.com/generated.png", downloads[0][0])
self.assertEqual((3, 240), downloads[0][1]["timeout"])
self.assertEqual((3, 650), downloads[0][1]["timeout"])
self.assertEqual(91, events[0]["metadata"]["points_balance"])
with Image.open(output) as saved:
self.assertEqual((1024, 1024), saved.size)
@@ -487,6 +488,7 @@ class AITests(TempDirMixin, unittest.TestCase):
self.assertEqual("read_timeout", raised.exception.code)
self.assertEqual(1, len(calls))
self.assertEqual((3, 650), calls[0][2]["timeout"])
self.assert_removed(temp_dir)