From f8cc2c0d677f55e81f21d0da9f45e773739a9745 Mon Sep 17 00:00:00 2001
From: huchenlei <huchenlei@proton.me>
Date: Fri, 22 Nov 2024 22:07:46 -0500
Subject: [PATCH] Update test expectations

---
 browser_tests/changeTracker.spec.ts | 12 +++++-------
 browser_tests/menu.spec.ts          |  8 ++++----
 2 files changed, 9 insertions(+), 11 deletions(-)

diff --git a/browser_tests/changeTracker.spec.ts b/browser_tests/changeTracker.spec.ts
index a76c5f6c9..188526540 100644
--- a/browser_tests/changeTracker.spec.ts
+++ b/browser_tests/changeTracker.spec.ts
@@ -56,9 +56,7 @@ test.describe('Change Tracker', () => {
       expect(await comfyPage.getToastErrorCount()).toBe(0)
       expect(await isModified()).toBe(false)
 
-      // TODO(huchenlei): Investigate why saving the workflow is causing the
-      // undo queue to be triggered.
-      expect(await getUndoQueueSize()).toBe(1)
+      expect(await getUndoQueueSize()).toBe(0)
       expect(await getRedoQueueSize()).toBe(0)
 
       const node = (await comfyPage.getFirstNodeRef())!
@@ -66,25 +64,25 @@ test.describe('Change Tracker', () => {
       await node.click('collapse')
       await expect(node).toBeCollapsed()
       expect(await isModified()).toBe(true)
-      expect(await getUndoQueueSize()).toBe(2)
+      expect(await getUndoQueueSize()).toBe(1)
       expect(await getRedoQueueSize()).toBe(0)
 
       await comfyPage.ctrlB()
       await expect(node).toBeBypassed()
       expect(await isModified()).toBe(true)
-      expect(await getUndoQueueSize()).toBe(3)
+      expect(await getUndoQueueSize()).toBe(2)
       expect(await getRedoQueueSize()).toBe(0)
 
       await comfyPage.ctrlZ()
       await expect(node).not.toBeBypassed()
       expect(await isModified()).toBe(true)
-      expect(await getUndoQueueSize()).toBe(2)
+      expect(await getUndoQueueSize()).toBe(1)
       expect(await getRedoQueueSize()).toBe(1)
 
       await comfyPage.ctrlZ()
       await expect(node).not.toBeCollapsed()
       expect(await isModified()).toBe(false)
-      expect(await getUndoQueueSize()).toBe(1)
+      expect(await getUndoQueueSize()).toBe(0)
       expect(await getRedoQueueSize()).toBe(2)
     })
   })
diff --git a/browser_tests/menu.spec.ts b/browser_tests/menu.spec.ts
index 3b956a04f..198a856d5 100644
--- a/browser_tests/menu.spec.ts
+++ b/browser_tests/menu.spec.ts
@@ -393,7 +393,7 @@ test.describe('Menu', () => {
 
       await tab.newBlankWorkflowButton.click()
       expect(await tab.getOpenedWorkflowNames()).toEqual([
-        '*Unsaved Workflow.json',
+        'Unsaved Workflow.json',
         'Unsaved Workflow (2).json'
       ])
     })
@@ -434,7 +434,7 @@ test.describe('Menu', () => {
       const openedWorkflow = tab.getOpenedItem('foo/bar')
       await tab.renameWorkflow(openedWorkflow, 'foo/baz')
       expect(await tab.getOpenedWorkflowNames()).toEqual([
-        '*Unsaved Workflow.json',
+        'Unsaved Workflow.json',
         'foo/baz.json'
       ])
     })
@@ -444,12 +444,12 @@ test.describe('Menu', () => {
       await comfyPage.menu.topbar.saveWorkflowAs('workflow3.json')
       expect(
         await comfyPage.menu.workflowsTab.getOpenedWorkflowNames()
-      ).toEqual(['*Unsaved Workflow.json', 'workflow3.json'])
+      ).toEqual(['Unsaved Workflow.json', 'workflow3.json'])
 
       await comfyPage.menu.topbar.saveWorkflowAs('workflow4.json')
       expect(
         await comfyPage.menu.workflowsTab.getOpenedWorkflowNames()
-      ).toEqual(['*Unsaved Workflow.json', 'workflow3.json', 'workflow4.json'])
+      ).toEqual(['Unsaved Workflow.json', 'workflow3.json', 'workflow4.json'])
     })
 
     test('Can save workflow as with same name', async ({ comfyPage }) => {