diff --git a/src/frontend/tests/end-to-end/group.spec.ts b/src/frontend/tests/end-to-end/group.spec.ts index f563bc1fe..4e6b9c081 100644 --- a/src/frontend/tests/end-to-end/group.spec.ts +++ b/src/frontend/tests/end-to-end/group.spec.ts @@ -1,80 +1,25 @@ import { expect, test } from "@playwright/test"; -import { readFileSync } from "fs"; -test.beforeEach(async ({ page }) => { - // await page.waitForTimeout(7000); - // test.setTimeout(120000); -}); + test.describe("group node test", () => { /// test("group and ungroup updating values", async ({ page }) => { await page.goto("/"); await page.locator('//*[@id="new-project-btn"]').click(); - await page.getByTestId("blank-flow").click(); + await page.getByRole('heading', { name: 'Data Ingestion' }).click(); await page.waitForTimeout(2000); - - // Read your file into a buffer. - const jsonContent = readFileSync( - "tests/end-to-end/assets/flow_group_test.json", - "utf-8" - ); - - // Create the DataTransfer and File - const dataTransfer = await page.evaluateHandle((data) => { - const dt = new DataTransfer(); - // Convert the buffer to a hex array - const file = new File([data], "flow_group_test.json", { - type: "application/json", - }); - dt.items.add(file); - return dt; - }, jsonContent); - - await page.waitForTimeout(2000); - - // Now dispatch - await page.dispatchEvent( - "//*[@id='react-flow-id']/div[1]/div[1]/div", - "drop", - { - dataTransfer, - } - ); - - const genericNoda = page.getByTestId("div-generic-node"); - const elementCount = await genericNoda.count(); - if (elementCount > 0) { - expect(true).toBeTruthy(); - } - await page - .locator('//*[@id="react-flow-id"]/div[1]/div[2]/button[3]') - .click(); - - await page.getByTestId("title-Agent Initializer").click({ - modifiers: ["Control"], - }); - - await page.getByTestId("title-PythonFunctionTool").click({ - modifiers: ["Control"], - }); - await page.getByTestId("title-ChatOpenAI").click({ - modifiers: ["Control"], - }); - - await page.getByRole("button", { name: "Group" }).click(); - - const textArea = page.getByTestId("div-textarea-description"); - const elementCountText = await textArea.count(); - if (elementCountText > 0) { - expect(true).toBeTruthy(); - } - - const groupNode = page.getByTestId("title-Group"); - const elementGroup = await groupNode.count(); - if (elementGroup > 0) { - expect(true).toBeTruthy(); - } - - // Now dispatch - }); + await page.getByLabel('fit view').click(); + await page.keyboard.down('Control'); + await page.getByTestId("title-OpenAIEmbeddings").click({modifiers: ['Control']}); + await page.getByTestId("title-URL").click({modifiers: ['Control']}); + await page.getByTestId("title-Recursive Character Text Splitter").click({modifiers: ['Control']}); + await page.keyboard.up('Control'); + await page.getByRole('button', { name: 'Group' }).click(); + await page.getByTestId(/input-collection_name_Chroma-.*/).click(); + await page.getByTestId(/input-collection_name_Chroma-.*/).fill('test'); + await page.getByTestId("title-Group").click(); + await page.keyboard.press("Control+g") + const value = await page.getByTestId('input-collection_name').inputValue() + expect(value).toBe('test'); + }) });