feat: tune cmhub image timeouts

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
+18
View File
@@ -227,6 +227,10 @@ class AppConfigTests(TempDirMixin, unittest.TestCase):
self.assertEqual("cmhub", ai["backend"])
self.assertEqual("cmhub", appconfig.ai_backend(config))
self.assertEqual("", appconfig.cmhub_config(config)["base_url"])
self.assertEqual(
appconfig.CMHUB_CONNECT_TIMEOUT_DEFAULT,
appconfig.cmhub_config(config)["connect_timeout"],
)
self.assertFalse(os.path.exists(cmhub_path))
self.assertEqual({"api_key": ""}, appconfig.load_cmhub_config(cmhub_path))
@@ -244,6 +248,20 @@ class AppConfigTests(TempDirMixin, unittest.TestCase):
self.assertEqual("cmhub", appconfig.ai_config(migrated)["backend"])
self.assertEqual(180, appconfig.response_timeout(migrated))
with open(config_path, "w", encoding="utf-8") as fh:
json.dump({"ai": {"cmhub": {"connect_timeout": 10}}}, fh)
migrated = appconfig.load_config(config_path)
self.assertEqual(
appconfig.CMHUB_CONNECT_TIMEOUT_DEFAULT,
appconfig.cmhub_config(migrated)["connect_timeout"],
)
saved = appconfig.save_config(
{"ai": {"cmhub": {"connect_timeout": 10}}},
path=config_path,
)
self.assertEqual(10, appconfig.cmhub_config(saved)["connect_timeout"])
direct_cfg = appconfig.default_config()
direct_cfg["ai"]["backend"] = "direct"
self.assertEqual("direct", appconfig.ai_backend(direct_cfg))