first commit

packages/testing/playwright/tests/e2e/ai/assistant-code-help.spec.ts

@@ -0,0 +1,101 @@
+import {
+	codeDiffSuggestionResponse,
+	applyCodeDiffResponse,
+	nodeExecutionSucceededResponse,
+	aiEnabledWorkflowBaseRequirements,
+	aiEnabledWithCodeDiffRequirements,
+} from '../../../config/ai-assistant-fixtures';
+import { test, expect } from '../../../fixtures/base';
+
+test.describe('AI Assistant::enabled', {
+	annotation: [
+		{ type: 'owner', description: 'AI' },
+	],
+}, () => {
+	test.describe('Code Node Error Help', () => {
+		test('should apply code diff to code node', async ({ n8n, setupRequirements }) => {
+			await setupRequirements(aiEnabledWithCodeDiffRequirements);
+
+			let applySuggestionCalls = 0;
+			await n8n.page.route('**/rest/ai/chat/apply-suggestion', async (route) => {
+				applySuggestionCalls += 1;
+				await route.fulfill({
+					status: 200,
+					contentType: 'application/json',
+					body: JSON.stringify(applyCodeDiffResponse),
+				});
+			});
+
+			await n8n.canvas.openNode('Code');
+
+			await n8n.ndv.execute();
+
+			await n8n.aiAssistant.getNodeErrorViewAssistantButton().click();
+
+			await expect(n8n.aiAssistant.getChatMessagesAll()).toHaveCount(2);
+			await expect(n8n.aiAssistant.getCodeDiffs()).toHaveCount(1);
+			await expect(n8n.aiAssistant.getApplyCodeDiffButtons()).toHaveCount(1);
+
+			await n8n.aiAssistant.getApplyCodeDiffButtons().first().click();
+
+			await expect(n8n.aiAssistant.getApplyCodeDiffButtons()).toHaveCount(0);
+			await expect(n8n.aiAssistant.getUndoReplaceCodeButtons()).toHaveCount(1);
+			await expect(n8n.aiAssistant.getCodeReplacedMessage()).toBeVisible();
+			await expect(n8n.ndv.getCodeEditor()).toContainText('item.json.myNewField = 1');
+
+			await n8n.aiAssistant.getUndoReplaceCodeButtons().first().click();
+
+			await expect(n8n.aiAssistant.getApplyCodeDiffButtons()).toHaveCount(1);
+			await expect(n8n.aiAssistant.getCodeReplacedMessage()).toHaveCount(0);
+			expect(applySuggestionCalls).toBe(1);
+			await expect(n8n.ndv.getCodeEditor()).toContainText('item.json.myNewField = 1aaa');
+
+			await n8n.aiAssistant.getApplyCodeDiffButtons().first().click();
+
+			await expect(n8n.ndv.getCodeEditor()).toContainText('item.json.myNewField = 1');
+		});
+
+		test('should ignore node execution success and error messages after the node run successfully once', async ({
+			n8n,
+			setupRequirements,
+		}) => {
+			await setupRequirements(aiEnabledWorkflowBaseRequirements);
+
+			let chatRequestCount = 0;
+			await n8n.page.route('**/rest/ai/chat', async (route) => {
+				chatRequestCount += 1;
+				const response =
+					chatRequestCount === 1 ? codeDiffSuggestionResponse : nodeExecutionSucceededResponse;
+				await route.fulfill({
+					status: 200,
+					contentType: 'application/json',
+					body: JSON.stringify(response),
+				});
+			});
+
+			await n8n.canvas.openNode('Code');
+			await n8n.ndv.execute();
+			await n8n.aiAssistant.getNodeErrorViewAssistantButton().click();
+
+			await n8n.ndv
+				.getCodeEditor()
+				.fill(
+					"// Loop over input items and add a new field called 'myNewField' to the JSON of each one\nfor (const item of $input.all()) {\n  item.json.myNewField = 1;\n}\n\nreturn $input.all();",
+				);
+
+			await n8n.ndv.execute();
+
+			await n8n.ndv
+				.getCodeEditor()
+				.fill(
+					"// Loop over input items and add a new field called 'myNewField' to the JSON of each one\nfor (const item of $input.all()) {\n  item.json.myNewField = 1aaaa!;\n}\n\nreturn $input.all();",
+				);
+
+			await n8n.ndv.execute();
+
+			await expect(n8n.aiAssistant.getChatMessagesAssistant().nth(2)).toContainText(
+				'Code node ran successfully, did my solution help resolve your issue?',
+			);
+		});
+	});
+});