jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

refactor Improve Vector Store test (#5144)

Browse source 
* fix: Improve Vector Store test and use AsyncSession in memory
This commit is contained in:
anovazzi1 2024-12-06 22:26:00 -03:00 • committed by GitHub
commit 678834dae8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 12 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

23
src/frontend/tests/core/integrations/Vector Store.spec.ts
View file

@ -37,7 +37,9 @@ test(
} }
while (modalCount === 0) { while (modalCount === 0) {
await page.getByText("New Flow", { exact: true }).click(); await page.getByText("New Flow", { exact: true }).click();
await page.waitForTimeout(3000); await page.waitForSelector('[data-testid="modal-title"]', {
timeout: 3000,
});
modalCount = await page.getByTestId("modal-title")?.count(); modalCount = await page.getByTestId("modal-title")?.count();
} }
await page.getByTestId("side_nav_options_all-templates").click(); await page.getByTestId("side_nav_options_all-templates").click();
@ -57,22 +59,20 @@ test(
.count(); .count();
while (outdatedComponents > 0) { while (outdatedComponents > 0) {
await page.getByTestId("icon-AlertTriangle").first().click(); await page.getByTestId("icon-AlertTriangle").first().click();
await page.waitForTimeout(1000);
outdatedComponents = await page.getByTestId("icon-AlertTriangle").count(); outdatedComponents = await page.getByTestId("icon-AlertTriangle").count();
} }
let filledApiKey = await page.getByTestId("remove-icon-badge").count(); let filledApiKey = await page.getByTestId("remove-icon-badge").count();
while (filledApiKey > 0) { while (filledApiKey > 0) {
await page.getByTestId("remove-icon-badge").first().click(); await page.getByTestId("remove-icon-badge").first().click();
await page.waitForTimeout(1000);
filledApiKey = await page.getByTestId("remove-icon-badge").count(); filledApiKey = await page.getByTestId("remove-icon-badge").count();
} }
if (process?.env?.ASTRA_DB_API_ENDPOINT?.includes("astra-dev")) { if (process?.env?.ASTRA_DB_API_ENDPOINT?.includes("astra-dev")) {
const getUA = await page.evaluate(() => navigator.userAgent); const getUA = await page.evaluate(() => navigator.userAgent);
const userAgentInfo = uaParser(getUA);
await page.getByTestId("title-Astra DB").first().click(); await page.getByTestId("title-Astra DB").first().click();
await page.waitForTimeout(500);
await page.getByTestId("code-button-modal").click(); await page.getByTestId("code-button-modal").click();
await page.waitForTimeout(500); await page.waitForSelector("text=Edit Code", {
timeout: 3000,
});
let cleanCode = await extractAndCleanCode(page); let cleanCode = await extractAndCleanCode(page);
cleanCode = cleanCode!.replace( cleanCode = cleanCode!.replace(
'"pre_delete_collection": self.pre_delete_collection or False,', '"pre_delete_collection": self.pre_delete_collection or False,',
@ -82,16 +82,18 @@ test(
await page.keyboard.press("Backspace"); await page.keyboard.press("Backspace");
await page.locator("textarea").last().fill(cleanCode); await page.locator("textarea").last().fill(cleanCode);
await page.locator('//*[@id="checkAndSaveBtn"]').click(); await page.locator('//*[@id="checkAndSaveBtn"]').click();
await page.waitForTimeout(500); await page.waitForSelector('[data-testid="title-Astra DB"]', {
timeout: 3000,
});
await page.getByTestId("title-Astra DB").last().click(); await page.getByTestId("title-Astra DB").last().click();
await page.waitForTimeout(500);
await page.getByTestId("code-button-modal").click(); await page.getByTestId("code-button-modal").click();
await page.waitForTimeout(500); await page.waitForSelector("text=Edit Code", {
timeout: 3000,
});
await page.locator("textarea").last().press(`ControlOrMeta+a`); await page.locator("textarea").last().press(`ControlOrMeta+a`);
await page.keyboard.press("Backspace"); await page.keyboard.press("Backspace");
await page.locator("textarea").last().fill(cleanCode); await page.locator("textarea").last().fill(cleanCode);
await page.locator('//*[@id="checkAndSaveBtn"]').click(); await page.locator('//*[@id="checkAndSaveBtn"]').click();
await page.waitForTimeout(500);
} }
const apiKeyInput = page.getByTestId("popover-anchor-input-api_key"); const apiKeyInput = page.getByTestId("popover-anchor-input-api_key");
const isApiKeyInputVisible = await apiKeyInput.isVisible(); const isApiKeyInputVisible = await apiKeyInput.isVisible();
@ -136,7 +138,6 @@ test(
path.join(__dirname, "../../assets/test_file.txt"), path.join(__dirname, "../../assets/test_file.txt"),
); );
await page.getByText("test_file.txt").isVisible(); await page.getByText("test_file.txt").isVisible();
await page.waitForTimeout(1000);
await page.getByTestId("button_run_astra db").last().click(); await page.getByTestId("button_run_astra db").last().click();
await page.waitForSelector("text=built successfully", { await page.waitForSelector("text=built successfully", {
timeout: 60000 * 2, timeout: 60000 * 2,