diff --git a/cypress/composables/ndv.ts b/cypress/composables/ndv.ts index 22eb423b9a..516a2e8aa2 100644 --- a/cypress/composables/ndv.ts +++ b/cypress/composables/ndv.ts @@ -73,7 +73,7 @@ export function getInputTbodyCell(row: number, col: number) { } export function getInputRunSelector() { - return getInputPanel().findChildByTestId('run-selector'); + return cy.get('[data-test-id="ndv-input-panel"] [data-test-id="run-selector"]'); } export function getInputPanelItemsCount() { @@ -105,7 +105,7 @@ export function getOutputTbodyCell(row: number, col: number) { } export function getOutputRunSelector() { - return getOutputPanel().findChildByTestId('run-selector'); + return cy.get('[data-test-id="output-panel"] [data-test-id="run-selector"]'); } export function getOutputRunSelectorInput() { diff --git a/cypress/e2e/30-langchain.cy.ts b/cypress/e2e/30-langchain.cy.ts index 9b6daf2a19..4aca21e3f6 100644 --- a/cypress/e2e/30-langchain.cy.ts +++ b/cypress/e2e/30-langchain.cy.ts @@ -381,7 +381,7 @@ describe('Langchain Integration', () => { workflowPage.actions.deselectAll(); workflowPage.actions.executeNode('Populate VS'); - workflow.getNodesWithSpinner().should('not.exist'); + workflow.waitForSuccessBannerToAppear(); const assertInputOutputText = (text: string, assertion: 'exist' | 'not.exist') => { ndv.getOutputPanel().contains(text).should(assertion); @@ -390,9 +390,6 @@ describe('Langchain Integration', () => { workflowPage.actions.openNode('Character Text Splitter'); - // Wait for the input panel to switch to Debugging mode - ndv.getInputPanelItemsCount().should('not.exist'); - ndv.getOutputRunSelector().should('exist'); ndv.getInputRunSelector().should('exist'); ndv.getInputRunSelector().find('input').should('include.value', '3 of 3');