chengma/cmshoppe
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: tune cmhub image timeouts

Browse Source
This commit is contained in:
chengma
2026-07-08 11:22:47 +08:00
parent 2970aaa943
commit 46afae4b98
7 changed files with 79 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_ai.py
+13 -10
View File
@@ -490,9 +490,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((3, ai.CMHUB_IMAGE_READ_TIMEOUT_SECONDS), calls[0][2]["timeout"])
self.assertEqual("https://cdn.example.com/generated.png", downloads[0][0])
self.assertEqual((3, 650), downloads[0][1]["timeout"])
self.assertEqual((3, ai.CMHUB_IMAGE_READ_TIMEOUT_SECONDS), downloads[0][1]["timeout"])
self.assertEqual(91, events[0]["metadata"]["points_balance"])
timed_steps = [
event for event in steps
@@ -668,7 +668,7 @@ class AITests(TempDirMixin, unittest.TestCase):
image_bytes, elapsed = ai._download_cmhub_image_with_retry(
"https://cdn.example.com/generated.png",
connect_timeout=3,
read_timeout=650,
read_timeout=ai.CMHUB_IMAGE_READ_TIMEOUT_SECONDS,
on_step=steps.append,
slow_threshold=0,
)
@@ -716,7 +716,10 @@ class AITests(TempDirMixin, unittest.TestCase):
self.assertIn("--connect-timeout", args)
self.assertEqual("3", args[args.index("--connect-timeout") + 1])
self.assertIn("--max-time", args)
self.assertEqual("650", args[args.index("--max-time") + 1])
self.assertEqual(
str(ai.CMHUB_IMAGE_READ_TIMEOUT_SECONDS),
args[args.index("--max-time") + 1],
)
self.assertIn("--max-filesize", args)
self.assertEqual(str(ai.CMHUB_IMAGE_MAX_BYTES), args[args.index("--max-filesize") + 1])
self.assertIn("--noproxy", args)
@@ -733,7 +736,7 @@ class AITests(TempDirMixin, unittest.TestCase):
image_bytes = ai._download_cmhub_image(
url,
connect_timeout=3,
read_timeout=650,
read_timeout=ai.CMHUB_IMAGE_READ_TIMEOUT_SECONDS,
use_system_proxy=False,
download_with_curl="true",
)
@@ -748,7 +751,7 @@ class AITests(TempDirMixin, unittest.TestCase):
ai._download_cmhub_image(
"http://127.0.0.1/a.png",
connect_timeout=3,
read_timeout=650,
read_timeout=ai.CMHUB_IMAGE_READ_TIMEOUT_SECONDS,
download_with_curl="true",
)
run.assert_not_called()
@@ -772,7 +775,7 @@ class AITests(TempDirMixin, unittest.TestCase):
image_bytes = ai._download_cmhub_image(
"https://cdn.example.com/generated.png",
connect_timeout=3,
read_timeout=650,
read_timeout=ai.CMHUB_IMAGE_READ_TIMEOUT_SECONDS,
download_with_curl="true",
)
@@ -797,7 +800,7 @@ class AITests(TempDirMixin, unittest.TestCase):
image_bytes = ai._download_cmhub_image(
"https://cdn.example.com/generated.png",
connect_timeout=3,
read_timeout=650,
read_timeout=ai.CMHUB_IMAGE_READ_TIMEOUT_SECONDS,
download_with_curl="auto",
)
@@ -821,7 +824,7 @@ class AITests(TempDirMixin, unittest.TestCase):
image_bytes = ai._download_cmhub_image(
"https://cdn.example.com/generated.png",
connect_timeout=3,
read_timeout=650,
read_timeout=ai.CMHUB_IMAGE_READ_TIMEOUT_SECONDS,
download_with_curl="auto",
)
@@ -886,7 +889,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.assertEqual((3, ai.CMHUB_IMAGE_READ_TIMEOUT_SECONDS), calls[0][2]["timeout"])
self.assert_removed(temp_dir)