Test register tab

Author: huchenlei

browser_tests/ComfyPage.ts

@@ -38,9 +38,11 @@ class ComfyNodeSearchBox {
 }
 
 class ComfyMenu {
+  public readonly sideToolBar: Locator
   public readonly themeToggleButton: Locator
 
   constructor(public readonly page: Page) {
+    this.sideToolBar = page.locator('.side-tool-bar-container')
     this.themeToggleButton = page.locator('.comfy-vue-theme-toggle')
   }
 

browser_tests/menu.spec.ts

@@ -42,4 +42,29 @@ test.describe('Menu', () => {
 
     expect(await comfyPage.menu.getThemeId()).toBe('dark')
   })
+
+  test('Can register sidebar tab', async ({ comfyPage }) => {
+    const initialChildrenCount = await comfyPage.menu.sideToolBar.evaluate(
+      (el) => el.children.length
+    )
+
+    await comfyPage.page.evaluate(async () => {
+      window['app'].extensionManager.registerSidebarTab({
+        id: 'search',
+        icon: 'pi pi-search',
+        title: 'search',
+        tooltip: 'search',
+        type: 'custom',
+        render: (el) => {
+          el.innerHTML = '<div>Custom search tab</div>'
+        }
+      })
+    })
+    await comfyPage.nextFrame()
+
+    const newChildrenCount = await comfyPage.menu.sideToolBar.evaluate(
+      (el) => el.children.length
+    )
+    expect(newChildrenCount).toBe(initialChildrenCount + 1)
+  })
 })