[test] Stricter locator matching on workflows sidebar elements (#1905)

* [test] Stricter locator matching on workflows sidebar elements

* nit
This commit is contained in:
Chenlei Hu
2024-12-13 14:13:56 -08:00
committed by GitHub
parent a99f63784d
commit 618b451f1e
2 changed files with 13 additions and 9 deletions

View File

@@ -85,32 +85,36 @@ export class WorkflowsSidebarTab extends SidebarTab {
super(page, 'workflows')
}
get root() {
return this.page.locator('.workflows-sidebar-tab')
}
get browseGalleryButton() {
return this.page.locator('.browse-templates-button')
return this.root.locator('.browse-templates-button')
}
get newBlankWorkflowButton() {
return this.page.locator('.new-blank-workflow-button')
return this.root.locator('.new-blank-workflow-button')
}
get openWorkflowButton() {
return this.page.locator('.open-workflow-button')
return this.root.locator('.open-workflow-button')
}
async getOpenedWorkflowNames() {
return await this.page
return await this.root
.locator('.comfyui-workflows-open .node-label')
.allInnerTexts()
}
async getActiveWorkflowName() {
return await this.page
return await this.root
.locator('.comfyui-workflows-open .p-tree-node-selected .node-label')
.innerText()
}
async getTopLevelSavedWorkflowNames() {
return await this.page
return await this.root
.locator('.comfyui-workflows-browse .node-label')
.allInnerTexts()
}
@@ -122,13 +126,13 @@ export class WorkflowsSidebarTab extends SidebarTab {
}
getOpenedItem(name: string) {
return this.page.locator('.comfyui-workflows-open .node-label', {
return this.root.locator('.comfyui-workflows-open .node-label', {
hasText: name
})
}
getPersistedItem(name: string) {
return this.page.locator('.comfyui-workflows-browse .node-label', {
return this.root.locator('.comfyui-workflows-browse .node-label', {
hasText: name
})
}

View File

@@ -1,7 +1,7 @@
<template>
<SidebarTabTemplate
:title="$t('sideToolbar.workflows')"
class="bg-[var(--p-tree-background)]"
class="workflows-sidebar-tab bg-[var(--p-tree-background)]"
>
<template #tool-buttons>
<Button