add screenshot expectations to test

Comfy-Org · Sep 23, 2024 · 0cc48d3 · 0cc48d3
1 parent a701832
commit 0cc48d3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/browser_tests/dialog.spec.ts b/browser_tests/dialog.spec.ts
@@ -61,6 +61,7 @@ test.describe('Missing models warning', () => {
     // Wait for the element with the .comfy-missing-models selector to be visible
     const missingModelsWarning = comfyPage.page.locator('.comfy-missing-models')
     await expect(missingModelsWarning).toBeVisible()
+    await expect(comfyPage.canvas).toHaveScreenshot('missing_models_dialog_initial.png')
 
     // Click the download button
     const downloadButton = comfyPage.page.getByLabel('Download')
@@ -70,5 +71,6 @@ test.describe('Missing models warning', () => {
     // Wait for the element with the .download-complete selector to be visible
     const downloadComplete = comfyPage.page.locator('.download-complete')
     await expect(downloadComplete).toBeVisible()
+    await expect(comfyPage.canvas).toHaveScreenshot('missing_models_dialog_downloaded.png')
   })
 })