T-529 default to cmhub gateway

This commit is contained in:
chengma
2026-07-04 17:59:23 +08:00
parent 3dbcbab799
commit 60497be557
13 changed files with 106 additions and 112 deletions
+1
View File
@@ -75,6 +75,7 @@ class AITests(TempDirMixin, unittest.TestCase):
def _config(self):
cfg = appconfig.default_config()
cfg["ai"]["backend"] = "direct"
cfg["ai"]["default_text_model"] = "Text"
cfg["ai"]["default_image_model"] = "Image"
cfg["ai"]["retry"] = 1
+7 -3
View File
@@ -38,8 +38,8 @@ class AppConfigTests(TempDirMixin, unittest.TestCase):
config = appconfig.load_config(config_path)
ai = appconfig.ai_config(config)
self.assertEqual("direct", ai["backend"])
self.assertEqual("direct", appconfig.ai_backend(config))
self.assertEqual("cmhub", ai["backend"])
self.assertEqual("cmhub", appconfig.ai_backend(config))
self.assertEqual("", appconfig.cmhub_config(config)["base_url"])
self.assertFalse(os.path.exists(cmhub_path))
self.assertEqual({"api_key": ""}, appconfig.load_cmhub_config(cmhub_path))
@@ -55,9 +55,13 @@ class AppConfigTests(TempDirMixin, unittest.TestCase):
with open(config_path, "w", encoding="utf-8") as fh:
json.dump({"ai": {"resolution": "512"}}, fh)
migrated = appconfig.load_config(config_path)
self.assertEqual("direct", appconfig.ai_config(migrated)["backend"])
self.assertEqual("cmhub", appconfig.ai_config(migrated)["backend"])
self.assertEqual(180, appconfig.response_timeout(migrated))
direct_cfg = appconfig.default_config()
direct_cfg["ai"]["backend"] = "direct"
self.assertEqual("direct", appconfig.ai_backend(direct_cfg))
self.assert_removed(temp_dir)
def test_config_rejects_sensitive_fields(self):
+3 -1
View File
@@ -814,7 +814,9 @@ class GuiTests(TempDirMixin, unittest.TestCase):
self.addCleanup(tab.close)
self.assertEqual("cmhub", tab.backend_combo.currentData())
self.assertTrue(tab.backend_combo.isHidden())
self.assertTrue(tab.model_picker_panel.isHidden())
self.assertTrue(tab.direct_role_panel.isHidden())
self.assertFalse(tab.cmhub_panel.isHidden())
self.assertEqual("https://cmhub.old", tab.cmhub_base_url_edit.text())
self.assertEqual("sk-old-secret", tab.cmhub_api_key_edit.text())
@@ -964,7 +966,7 @@ class GuiTests(TempDirMixin, unittest.TestCase):
self.assertEqual("失败 0", tab.failed_progress_label.text())
self.assertEqual("generateCmhubBalanceLabel", tab.cmhub_balance_label.objectName())
self.assertEqual("cmhub余额:未获取", tab.cmhub_balance_label.text())
self.assertTrue(tab.cmhub_balance_label.isHidden())
self.assertFalse(tab.cmhub_balance_label.isHidden())
self.assertEqual(0, tab.title_progress_bar.value())
self.assertEqual(0, tab.cover_progress_bar.value())
self.assertEqual(1, tab.title_progress_bar.maximum())