test(subscription): suspend client access checks
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user