From 9e350d7ac209b5945184c2bb49dec039f8f74c9d Mon Sep 17 00:00:00 2001
From: huchenlei <chenlei.hu@mail.utoronto.ca>
Date: Tue, 27 Aug 2024 17:10:51 -0400
Subject: [PATCH] Adjust setting in test

---
 browser_tests/dialog.spec.ts | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/browser_tests/dialog.spec.ts b/browser_tests/dialog.spec.ts
index c372a788b..392ef0323 100644
--- a/browser_tests/dialog.spec.ts
+++ b/browser_tests/dialog.spec.ts
@@ -27,6 +27,15 @@ test.describe('Execution error', () => {
 })
 
 test.describe('Missing models warning', () => {
+  test.beforeEach(async ({ comfyPage }) => {
+    await comfyPage.setSetting('Comfy.Workflow.ModelDownload.AllowedSources', [
+      'http://localhost:8188'
+    ])
+    await comfyPage.setSetting('Comfy.Workflow.ModelDownload.AllowedSuffixes', [
+      '.safetensors'
+    ])
+  })
+
   test('Should display a warning when missing models are found', async ({
     comfyPage
   }) => {