feat: tune cmhub image timeouts
This commit is contained in:
@@ -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))
|
||||
|
||||
Reference in New Issue
Block a user