test: fix test after changes on LLMs components (#2538)

* ✅ (tests): remove redundant test assertions in filterEdge-shard-0.spec.ts ✅ (tests): add new test assertions and remove redundant ones in filterEdge-shard-1.spec.ts * ✅ (filterSidebar.spec.ts): add test for visibility of helpersID Generator ♻️ (filterSidebar.spec.ts): remove redundant visibility test for textsplittersCharacterTextSplitter --------- Co-authored-by: Gabriel Luiz Freitas Almeida <[email protected]>
langflow-ai · Jul 4, 2024 · 2b8d314 · 2b8d314
1 parent a933139
commit 2b8d314
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 13 deletions.
diff --git a/src/frontend/tests/end-to-end/filterEdge-shard-0.spec.ts b/src/frontend/tests/end-to-end/filterEdge-shard-0.spec.ts
@@ -55,9 +55,6 @@ test("LLMChain - Tooltip", async ({ page }) => {
     await expect(
       page.getByTestId("available-output-chains").first(),
     ).toBeVisible();
-    await expect(
-      page.getByTestId("available-output-textsplitters").first(),
-    ).toBeVisible();
     await expect(
       page.getByTestId("available-output-retrievers").first(),
     ).toBeVisible();
@@ -73,9 +70,6 @@ test("LLMChain - Tooltip", async ({ page }) => {
     await expect(
       page.getByTestId("available-output-toolkits").first(),
     ).toBeVisible();
-    await expect(
-      page.getByTestId("available-output-chains").first(),
-    ).toBeVisible();
     await expect(
       page.getByTestId("available-output-agents").first(),
     ).toBeVisible();

diff --git a/src/frontend/tests/end-to-end/filterEdge-shard-1.spec.ts b/src/frontend/tests/end-to-end/filterEdge-shard-1.spec.ts
@@ -69,10 +69,13 @@ test("LLMChain - Filter", async ({ page }) => {
   await expect(page.getByTestId("disclosure-memories")).toBeVisible();
   await expect(page.getByTestId("disclosure-prototypes")).toBeVisible();
   await expect(page.getByTestId("disclosure-retrievers")).toBeVisible();
-  await expect(page.getByTestId("disclosure-text splitters")).toBeVisible();
   await expect(page.getByTestId("disclosure-toolkits")).toBeVisible();
   await expect(page.getByTestId("disclosure-tools")).toBeVisible();
 
+  await expect(page.getByTestId("helpersID Generator").first()).toBeVisible();
+
+  await expect(page.getByTestId("agentsCSVAgent").first()).toBeVisible();
+
   await expect(page.getByTestId("chainsLLMChain").first()).toBeVisible();
   await expect(
     page.getByTestId("langchain_utilitiesSearchApi").first(),
@@ -86,9 +89,7 @@ test("LLMChain - Filter", async ({ page }) => {
   await expect(
     page.getByTestId("retrieversAmazon Kendra Retriever").first(),
   ).toBeVisible();
-  await expect(
-    page.getByTestId("textsplittersCharacterTextSplitter").first(),
-  ).toBeVisible();
+
   await expect(
     page.getByTestId("toolkitsVectorStoreInfo").first(),
   ).toBeVisible();

diff --git a/src/frontend/tests/end-to-end/filterSidebar.spec.ts b/src/frontend/tests/end-to-end/filterSidebar.spec.ts
@@ -50,6 +50,7 @@ test("LLMChain - Filter", async ({ page }) => {
     )
     .click();
 
+  await expect(page.getByTestId("helpersID Generator")).toBeVisible();
   await expect(page.getByTestId("disclosure-agents")).toBeVisible();
 
   await expect(page.getByTestId("chainsLLMChain").first()).toBeVisible();
@@ -65,9 +66,7 @@ test("LLMChain - Filter", async ({ page }) => {
   await expect(
     page.getByTestId("retrieversAmazon Kendra Retriever").first(),
   ).toBeVisible();
-  await expect(
-    page.getByTestId("textsplittersCharacterTextSplitter").first(),
-  ).toBeVisible();
+
   await expect(
     page.getByTestId("toolkitsVectorStoreInfo").first(),
   ).toBeVisible();