diff --git a/browser_tests/dialog.spec.ts b/browser_tests/dialog.spec.ts index 46ad6e31d..dcc20afd5 100644 --- a/browser_tests/dialog.spec.ts +++ b/browser_tests/dialog.spec.ts @@ -1,4 +1,4 @@ -import { expect } from '@playwright/test' +import { Locator, expect } from '@playwright/test' import { Keybinding } from '../src/types/keyBindingTypes' import { comfyPageFixture as test } from './fixtures/ComfyPage' @@ -140,6 +140,49 @@ test.describe('Missing models warning', () => { const download = await downloadPromise expect(download.suggestedFilename()).toBe('fake_model.safetensors') }) + + test.describe('Do not show again checkbox', () => { + let checkbox: Locator + let closeButton: Locator + + test.beforeEach(async ({ comfyPage }) => { + await comfyPage.setSetting( + 'Comfy.Workflow.ShowMissingModelsWarning', + true + ) + await comfyPage.loadWorkflow('missing_models') + + checkbox = comfyPage.page.getByLabel("Don't show this again") + closeButton = comfyPage.page.getByLabel('Close') + }) + + test('Should disable warning dialog when checkbox is checked', async ({ + comfyPage + }) => { + await checkbox.click() + const changeSettingPromise = comfyPage.page.waitForRequest( + '**/api/settings/Comfy.Workflow.ShowMissingModelsWarning' + ) + await closeButton.click() + await changeSettingPromise + + const settingValue = await comfyPage.getSetting( + 'Comfy.Workflow.ShowMissingModelsWarning' + ) + expect(settingValue).toBe(false) + }) + + test('Should keep warning dialog enabled when checkbox is unchecked', async ({ + comfyPage + }) => { + await closeButton.click() + + const settingValue = await comfyPage.getSetting( + 'Comfy.Workflow.ShowMissingModelsWarning' + ) + expect(settingValue).toBe(true) + }) + }) }) test.describe('Settings', () => { diff --git a/src/components/dialog/content/MissingModelsWarning.vue b/src/components/dialog/content/MissingModelsWarning.vue index 257746d11..7ca1f2b20 100644 --- a/src/components/dialog/content/MissingModelsWarning.vue +++ b/src/components/dialog/content/MissingModelsWarning.vue @@ -2,9 +2,15 @@ +
+ + +