T-576 增加生成和更新内容模式
This commit is contained in:
@@ -27,6 +27,9 @@ class AppConfigTests(TempDirMixin, unittest.TestCase):
|
||||
self.assertEqual(os.path.join(temp_dir, "config", "cmhub.json"), config["cmhub_config_path"])
|
||||
self.assertEqual(240, appconfig.response_timeout(config))
|
||||
self.assertFalse(appconfig.ai_config(config)["generate_cover"])
|
||||
self.assertEqual("title", appconfig.ai_generate_mode(config))
|
||||
self.assertEqual("title", appconfig.shopee_update_config(config)["update_mode"])
|
||||
self.assertFalse(appconfig.shopee_update_config(config)["allow_cover_update"])
|
||||
|
||||
updated = appconfig.update_config(
|
||||
{"ai": {"resolution": "2k"}},
|
||||
@@ -35,6 +38,30 @@ class AppConfigTests(TempDirMixin, unittest.TestCase):
|
||||
self.assertEqual(360, appconfig.response_timeout(updated))
|
||||
self.assertEqual((9222, 9260), appconfig.debug_port_range(updated))
|
||||
|
||||
legacy = appconfig.save_config(
|
||||
{
|
||||
"ai": {"generate_cover": True},
|
||||
"shopee_update": {"allow_cover_update": True},
|
||||
},
|
||||
path=config_path,
|
||||
)
|
||||
self.assertEqual("title_cover", appconfig.ai_generate_mode(legacy))
|
||||
self.assertTrue(appconfig.ai_config(legacy)["generate_cover"])
|
||||
self.assertEqual("title_cover", appconfig.shopee_update_config(legacy)["update_mode"])
|
||||
self.assertTrue(appconfig.shopee_update_config(legacy)["allow_cover_update"])
|
||||
|
||||
cover_only = appconfig.save_config(
|
||||
{
|
||||
"ai": {"generate_mode": "cover"},
|
||||
"shopee_update": {"update_mode": "cover"},
|
||||
},
|
||||
path=config_path,
|
||||
)
|
||||
self.assertEqual("cover", appconfig.ai_generate_mode(cover_only))
|
||||
self.assertTrue(appconfig.ai_config(cover_only)["generate_cover"])
|
||||
self.assertEqual("cover", appconfig.shopee_update_config(cover_only)["update_mode"])
|
||||
self.assertTrue(appconfig.shopee_update_config(cover_only)["allow_cover_update"])
|
||||
|
||||
self.assert_removed(temp_dir)
|
||||
|
||||
def test_config_save_does_not_persist_runtime_paths(self):
|
||||
|
||||
Reference in New Issue
Block a user