fix: align cmhub image timeout display
This commit is contained in:
+3
-1
@@ -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)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user