first commit

packages/testing/playwright/tests/e2e/ai/assistant-support-chat.spec.ts

@@ -0,0 +1,118 @@
+import {
+	simpleAssistantResponse,
+	aiEnabledWithCodeSnippetRequirements,
+	aiEnabledWithHttpWorkflowRequirements,
+} from '../../../config/ai-assistant-fixtures';
+import { HTTP_REQUEST_NODE_NAME } from '../../../config/constants';
+import { test, expect } from '../../../fixtures/base';
+
+type ChatRequestBody = {
+	payload?: {
+		type?: string;
+		text?: string;
+		question?: string;
+		context?: Record<string, unknown>;
+	};
+};
+
+test.describe('AI Assistant::enabled', {
+	annotation: [
+		{ type: 'owner', description: 'AI' },
+	],
+}, () => {
+	test.describe('Support Chat', () => {
+		test('assistant returns code snippet', async ({ n8n, setupRequirements }) => {
+			await setupRequirements(aiEnabledWithCodeSnippetRequirements);
+			await n8n.page.goto('/workflow/new');
+
+			await expect(n8n.aiAssistant.getAskAssistantCanvasActionButton()).toBeVisible();
+			await n8n.aiAssistant.getAskAssistantCanvasActionButton().click();
+
+			await expect(n8n.aiAssistant.getAskAssistantChat()).toBeVisible();
+
+			await n8n.aiAssistant.sendMessage('Show me an expression');
+
+			await expect(n8n.aiAssistant.getChatMessagesAll()).toHaveCount(3);
+			await expect(n8n.aiAssistant.getChatMessagesUser().first()).toContainText(
+				'Show me an expression',
+			);
+			await expect(n8n.aiAssistant.getChatMessagesAssistant().first()).toContainText(
+				'To use expressions in n8n, follow these steps:',
+			);
+			await expect(n8n.aiAssistant.getChatMessagesAssistant().first()).toContainText('New York');
+			await expect(n8n.aiAssistant.getCodeSnippet()).toHaveText('{{$json.body.city}}');
+		});
+
+		test('should send current context to support chat', async ({ n8n, setupRequirements }) => {
+			await setupRequirements(aiEnabledWithHttpWorkflowRequirements);
+
+			const chatRequests: ChatRequestBody[] = [];
+			await n8n.page.route('**/rest/ai/chat', async (route) => {
+				const body = route.request().postDataJSON() as ChatRequestBody;
+				chatRequests.push(body);
+				await route.fulfill({
+					status: 200,
+					contentType: 'application/json',
+					body: JSON.stringify(simpleAssistantResponse),
+				});
+			});
+
+			await n8n.aiAssistant.getAskAssistantCanvasActionButton().click();
+			await n8n.aiAssistant.sendMessage('What is wrong with this workflow?');
+
+			const supportRequest = chatRequests.find(
+				(request) => request.payload?.question === 'What is wrong with this workflow?',
+			);
+			expect(supportRequest).toBeDefined();
+			const supportContext = supportRequest?.payload?.context;
+			expect(supportContext).toBeDefined();
+			expect(supportContext?.currentView).toBeDefined();
+			expect(supportContext?.currentWorkflow).toBeDefined();
+		});
+
+		test('should not send workflow context if nothing changed', async ({
+			n8n,
+			setupRequirements,
+		}) => {
+			await setupRequirements(aiEnabledWithHttpWorkflowRequirements);
+
+			const chatRequests: ChatRequestBody[] = [];
+			await n8n.page.route('**/rest/ai/chat', async (route) => {
+				const body = route.request().postDataJSON() as ChatRequestBody;
+				chatRequests.push(body);
+				await route.fulfill({
+					status: 200,
+					contentType: 'application/json',
+					body: JSON.stringify(simpleAssistantResponse),
+				});
+			});
+
+			await n8n.aiAssistant.getAskAssistantCanvasActionButton().click();
+
+			await n8n.aiAssistant.sendMessage('What is wrong with this workflow?', 'enter-key');
+			// Wait for message to be processed
+			await expect(n8n.aiAssistant.getChatMessagesAssistant()).toHaveCount(1);
+
+			await n8n.aiAssistant.sendMessage('And now?', 'enter-key');
+			await expect(n8n.aiAssistant.getChatMessagesAssistant()).toHaveCount(2);
+			const secondRequest = chatRequests.find((request) => request.payload?.text === 'And now?');
+			const secondContext = secondRequest?.payload?.context;
+			expect(secondContext?.currentWorkflow).toBeUndefined();
+
+			await n8n.canvas.openNode(HTTP_REQUEST_NODE_NAME);
+			await n8n.ndv.setParameterInputValue('url', 'https://example.com');
+			await n8n.ndv.close();
+			await n8n.canvas.clickExecuteWorkflowButton();
+
+			await n8n.aiAssistant.sendMessage('What about now?', 'enter-key');
+			await expect(n8n.aiAssistant.getChatMessagesAssistant()).toHaveCount(3);
+
+			const thirdRequest = chatRequests.find(
+				(request) => request.payload?.text === 'What about now?',
+			);
+			const thirdContext = thirdRequest?.payload?.context;
+			expect(thirdContext?.currentWorkflow).toBeTruthy();
+			expect(thirdContext?.executionData).toBeTruthy();
+		});
+	});
+});