From 0cfaf017474ff599314ce40f633e17c92fdf934b Mon Sep 17 00:00:00 2001
From: magicwang1111 <alimagicwang@gmail.com>
Date: Wed, 28 Oct 2026 11:43:51 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E6=96=AD=E7=BA=BF=E6=8F=90?=
 =?UTF-8?q?=E7=A4=BA=E5=BC=80=E5=85=B3=E5=86=99=E5=85=A5=E9=94=99=E8=AF=AF?=
 =?UTF-8?q?=E5=AF=BC=E8=87=B4=E8=8B=B1=E6=96=87=E6=8F=90=E7=A4=BA=E4=BB=8D?=
 =?UTF-8?q?=E7=84=B6=E5=87=BA=E7=8E=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 tests/auto_refresh.test.cjs       | 4 +++-
 tests/browser_frontend_reload.cjs | 5 ++++-
 web/js/autoRefresh.js             | 4 ++--
 3 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/tests/auto_refresh.test.cjs b/tests/auto_refresh.test.cjs
index 278f024..f37a5fc 100644
--- a/tests/auto_refresh.test.cjs
+++ b/tests/auto_refresh.test.cjs
@@ -12,6 +12,7 @@ function harness() {
     };
     const app = {
         isGraphReady: true,
+        ui: { settings: { settingsValues: {} } },
         graphToPrompt: async () => ({ workflow: { nodes: [] } }),
         extensionManager: {
             workflow: { openWorkflows: [] },
@@ -98,7 +99,8 @@ test('disconnect shows one waiting notification and reconnect removes it', async
     h.events.get('reconnecting')();
     assert.equal(h.toasts.length, 1);
     assert.equal(h.toasts[0].summary, '服务器连接已断开，正在等待恢复');
-    assert.equal(h.app.extensionManager.setting.settings['Comfy.Toast.DisableReconnectingToast'], true);
+    assert.equal(h.app.ui.settings.settingsValues['Comfy.Toast.DisableReconnectingToast'], true);
+    assert.equal(h.app.extensionManager.setting.settings['Comfy.Toast.DisableReconnectingToast'], undefined);
     await h.events.get('reconnected')();
     assert.equal(h.toasts.length, 0);
     h.events.get('reconnecting')();
diff --git a/tests/browser_frontend_reload.cjs b/tests/browser_frontend_reload.cjs
index c610d14..14af61f 100644
--- a/tests/browser_frontend_reload.cjs
+++ b/tests/browser_frontend_reload.cjs
@@ -81,6 +81,8 @@ async function draftImages(page) {
     const browser = await chromium.launch({ headless: true, executablePath: process.env.PLAYWRIGHT_CHROMIUM_EXECUTABLE || undefined });
     try {
         const connection = await harness(browser);
+        assert.equal(await connection.page.evaluate(() =>
+            window.comfyAPI.app.app.extensionManager.setting.get('Comfy.Toast.DisableReconnectingToast')), true);
         const connectionBefore = connection.navigations();
         const waiting = connection.page.getByText('服务器连接已断开，正在等待恢复', { exact: true });
         for (let cycle = 0; cycle < 2; cycle++) {
@@ -97,8 +99,9 @@ async function draftImages(page) {
         connection.state.disconnected = true;
         await connection.sockets.at(-1).close({ code: 1001, reason: 'Test server shutdown' });
         await waiting.waitFor();
-        await connection.page.waitForTimeout(1500);
+        await connection.page.waitForTimeout(6000);
         assert.equal(await waiting.count(), 1);
+        assert.equal(await connection.page.getByText('Reconnecting', { exact: true }).count(), 0);
         assert.equal(await connection.page.getByText('自动刷新已暂停', { exact: true }).count(), 0);
         connection.state.disconnected = false;
         await waiting.waitFor({ state: 'hidden' });
diff --git a/web/js/autoRefresh.js b/web/js/autoRefresh.js
index a41b4f3..38d34c1 100644
--- a/web/js/autoRefresh.js
+++ b/web/js/autoRefresh.js
@@ -142,7 +142,7 @@ async function checkForUpdate() {
         }
         reloading = true;
         // Change only the in-memory preference; the saved setting returns after reload.
-        app.extensionManager.setting.settings["Comfy.Window.UnloadConfirmation"] = false;
+        app.ui.settings.settingsValues["Comfy.Window.UnloadConfirmation"] = false;
         window.location.reload();
     } finally {
         busy = false;
@@ -158,7 +158,7 @@ app.registerExtension({
     name: "ComfyUI.AccountManager.AutoRefresh",
     setup() {
         // Replace the built-in reconnect toast for this page only (do not save the setting).
-        app.extensionManager.setting.settings["Comfy.Toast.DisableReconnectingToast"] = true;
+        app.ui.settings.settingsValues["Comfy.Toast.DisableReconnectingToast"] = true;
         const onDisconnect = () => {
             if (!disconnected) {
                 connectionToast = {