T-529 default to cmhub gateway
This commit is contained in:
@@ -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
|
||||
|
||||
@@ -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
@@ -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())
|
||||
|
||||
Reference in New Issue
Block a user