tests: Update tool mode test to use URL component instead of Astra DB (#5329)

fix: update tool mode test to use URL component instead of Astra DB
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-12-18 09:58:47 -03:00 • committed by GitHub
commit 1cda13d833
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -7,25 +7,29 @@ test(
async ({ page }) => { async ({ page }) => {
await awaitBootstrapTest(page); await awaitBootstrapTest(page);
await page.getByTestId("blank-flow").click(); await page.getByTestId("blank-flow").click();
await page.waitForSelector('[data-testid="disclosure-vector stores"]', { await page.waitForSelector('[data-testid="disclosure-data"]', {
timeout: 3000, timeout: 3000,
state: "visible", state: "visible",
}); });
await page.getByTestId("disclosure-vector stores").click(); await page.getByTestId("disclosure-data").click();
await page.waitForSelector('[data-testid="vectorstoresAstra DB"]', { await page.waitForSelector('[data-testid="dataURL"]', {
timeout: 3000, timeout: 3000,
state: "visible", state: "visible",
}); });
await page await page
.getByTestId("vectorstoresAstra DB") .getByTestId("dataURL")
.hover() .hover()
.then(async () => { .then(async () => {
await page.getByTestId("add-component-button-astra-db").click(); await page.getByTestId("add-component-button-url").click();
}); });
await page.getByTestId("generic-node-title-arrangement").click(); await page.getByTestId("generic-node-title-arrangement").click();
await page
.getByTestId("generic-node-title-arrangement")
.dragTo(page.locator('//*[@id="react-flow-id"]'));
await page.keyboard.press("ControlOrMeta+Shift+m"); await page.keyboard.press("ControlOrMeta+Shift+m");
await page.waitForSelector("text=toolset", { await page.waitForSelector("text=toolset", {
@ -105,7 +109,7 @@ test(
state: "visible", state: "visible",
}); });
await page.getByTestId("disclosure-vector stores").click(); await page.getByTestId("disclosure-data").click();
await page.getByTestId("disclosure-agents").click(); await page.getByTestId("disclosure-agents").click();
@ -120,10 +124,9 @@ test(
await page.getByTestId("add-component-button-agent").click(); await page.getByTestId("add-component-button-agent").click();
}); });
await page // Move the Agent node a bit
.getByTestId("handle-astradb-shownode-toolset-right")
.first() await page.getByTestId("handle-url-shownode-toolset-right").first().click();
.click();
await page.getByTestId("handle-agent-shownode-tools-left").first().click(); await page.getByTestId("handle-agent-shownode-tools-left").first().click();