Fix 30-langchain e2e spec

Signed-off-by: Oleg Ivaniv <me@olegivaniv.com>
n8n-io · Feb 1, 2024 · 1b6b7a7 · 1b6b7a7
1 parent 244e63b
commit 1b6b7a7
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 5 deletions.
diff --git a/cypress/composables/workflow.ts b/cypress/composables/workflow.ts
@@ -106,14 +106,19 @@ export function addSupplementalNodeToParent(
 	nodeName: string,
 	endpointType: EndpointType,
 	parentNodeName: string,
+	exactMatch = false,
 ) {
 	getAddInputEndpointByType(parentNodeName, endpointType).click({ force: true });
-	getNodeCreatorItems().contains(nodeName).click();
+	if (exactMatch) {
+		getNodeCreatorItems().contains(new RegExp("^" + nodeName + "$", "g")).click();
+	} else {
+		getNodeCreatorItems().contains(nodeName).click();
+	}
 	getConnectionBySourceAndTarget(parentNodeName, nodeName).should('exist');
 }
 
-export function addLanguageModelNodeToParent(nodeName: string, parentNodeName: string) {
-	addSupplementalNodeToParent(nodeName, 'ai_languageModel', parentNodeName);
+export function addLanguageModelNodeToParent(nodeName: string, parentNodeName: string, exactMatch = false) {
+	addSupplementalNodeToParent(nodeName, 'ai_languageModel', parentNodeName, exactMatch);
 }
 
 export function addMemoryNodeToParent(nodeName: string, parentNodeName: string) {

diff --git a/cypress/e2e/30-langchain.cy.ts b/cypress/e2e/30-langchain.cy.ts
@@ -83,6 +83,7 @@ describe('Langchain Integration', () => {
 		addLanguageModelNodeToParent(
 			AI_LANGUAGE_MODEL_OPENAI_CHAT_MODEL_NODE_NAME,
 			BASIC_LLM_CHAIN_NODE_NAME,
+			true
 		);
 
 		clickCreateNewCredential();
@@ -121,7 +122,7 @@ describe('Langchain Integration', () => {
 		addNodeToCanvas(MANUAL_CHAT_TRIGGER_NODE_NAME, true);
 		addNodeToCanvas(AGENT_NODE_NAME, true);
 
-		addLanguageModelNodeToParent(AI_LANGUAGE_MODEL_OPENAI_CHAT_MODEL_NODE_NAME, AGENT_NODE_NAME);
+		addLanguageModelNodeToParent(AI_LANGUAGE_MODEL_OPENAI_CHAT_MODEL_NODE_NAME, AGENT_NODE_NAME, true);
 
 		clickCreateNewCredential();
 		setCredentialValues({
@@ -159,7 +160,7 @@ describe('Langchain Integration', () => {
 		addNodeToCanvas(MANUAL_CHAT_TRIGGER_NODE_NAME, true);
 		addNodeToCanvas(AGENT_NODE_NAME, true);
 
-		addLanguageModelNodeToParent(AI_LANGUAGE_MODEL_OPENAI_CHAT_MODEL_NODE_NAME, AGENT_NODE_NAME);
+		addLanguageModelNodeToParent(AI_LANGUAGE_MODEL_OPENAI_CHAT_MODEL_NODE_NAME, AGENT_NODE_NAME, true);
 
 		clickCreateNewCredential();
 		setCredentialValues({