test(subscription): suspend client access checks

This commit is contained in:
chengma
2026-07-22 15:55:46 +08:00
parent 953d98314b
commit 9163f6b371
6 changed files with 35 additions and 4 deletions
+17
View File
@@ -74,6 +74,7 @@ from app.gui import (
)
from app.gui import file_manager
import app.gui.workers as gui_workers
import app.gui.main_window as main_window
from app.gui.tabs.collect import _format_collect_elapsed
from app.gui.tabs.generate import CoverGalleryDialog, OriginalImageDialog
from app.gui.tabs.settings import SettingsSpinBox
@@ -11185,6 +11186,22 @@ class GuiTests(TempDirMixin, unittest.TestCase):
self.assertEqual(title == "设置", window.tabs.isTabEnabled(index))
self.assertEqual(TAB_TITLES.index("设置"), window.tabs.currentIndex())
def test_main_window_temporarily_bypasses_subscription_checks(self):
self.assertFalse(main_window.SUBSCRIPTION_CHECK_ENABLED)
with self.make_temp_dir() as temp_dir:
window = MainWindow(config=self.make_config(temp_dir))
self.addCleanup(window.close)
with mock.patch("app.gui.main_window.SubscriptionCheckWorker") as worker:
window.begin_subscription_check()
worker.assert_not_called()
self.assertEqual("会员:订阅检测已暂停", window._subscription_label.text())
self.assertTrue(
all(window.tabs.isTabEnabled(index) for index in range(window.tabs.count()))
)
self.assertTrue(window.ensure_subscription_for_new_submit("开始 AI 生成"))
def test_generate_and_suite_preflight_can_stop_new_submissions(self):
with self.make_temp_dir() as temp_dir:
config = self.make_config(temp_dir)