alighasami
3d5eaf9445
Security: Sync from Public / sync-from-public (push) Has been cancelled
Test: Benchmark Nightly / build (push) Has been cancelled
Test: Benchmark Nightly / Notify Cats on failure (push) Has been cancelled
CI: Python / Checks (push) Has been cancelled
Test: Evals Python / Workflow Comparison Python (push) Has been cancelled
Util: Check Docs URLs / check-docs-urls (push) Has been cancelled
Test: Visual Storybook / Cloudflare Pages (push) Has been cancelled
Test: E2E Performance / build-and-test-performance (push) Has been cancelled
Test: Workflows Nightly / Run Workflow Tests (push) Has been cancelled
Util: Cleanup CI Docker Images / Delete stale CI images (push) Has been cancelled
Test: Benchmark Destroy Env / build (push) Has been cancelled
Util: Update Node Popularity / update-popularity (push) Has been cancelled
Test: E2E Coverage Weekly / Coverage Tests (push) Has been cancelled
106 lines
4.0 KiB
TypeScript
106 lines
4.0 KiB
TypeScript
import fs from 'fs/promises';
|
|
import os from 'os';
|
|
import path from 'path';
|
|
|
|
import { test, expect, chatHubTestConfig } from './fixtures';
|
|
import { ChatHubChatPage } from '../../../pages/ChatHubChatPage';
|
|
|
|
test.use(chatHubTestConfig);
|
|
|
|
test.describe('File attachment @capability:proxy', {
|
|
annotation: [
|
|
{ type: 'owner', description: 'Chat' },
|
|
],
|
|
}, () => {
|
|
let tmpDir: string;
|
|
let testImagePath: string;
|
|
let testTextPath: string;
|
|
|
|
test.beforeEach(async () => {
|
|
tmpDir = await fs.mkdtemp(path.join(os.tmpdir(), 'chat-hub-test-'));
|
|
testImagePath = path.join(tmpDir, 'test-image.png');
|
|
testTextPath = path.join(tmpDir, 'test-file.txt');
|
|
|
|
// 100x100 solid red square PNG
|
|
const pngBuffer = Buffer.from(
|
|
'iVBORw0KGgoAAAANSUhEUgAAAGQAAABkCAIAAAD/gAIDAAAAkElEQVR42u3QMQ0AAAjAsPk3DRb4eJpUQZviSIEsWbJkyUKBLFmyZMlCgSxZsmTJQoEsWbJkyUKBLFmyZMlCgSxZsmTJQoEsWbJkyUKBLFmyZMlCgSxZsmTJQoEsWbJkyUKBLFmyZMlCgSxZsmTJQoEsWbJkyUKBLFmyZMlCgSxZsmTJQoEsWbJkyUKBLFnvFp4t6yugc3LNAAAAAElFTkSuQmCC',
|
|
'base64',
|
|
);
|
|
await fs.writeFile(testImagePath, pngBuffer);
|
|
|
|
// Simple text file
|
|
await fs.writeFile(testTextPath, 'I am a file');
|
|
});
|
|
|
|
test.afterEach(async () => {
|
|
if (tmpDir) {
|
|
await fs.rm(tmpDir, { recursive: true, force: true });
|
|
}
|
|
});
|
|
|
|
test('image attachment', async ({ n8n, anthropicCredential: _ }) => {
|
|
const page = new ChatHubChatPage(n8n.page);
|
|
|
|
await n8n.navigate.toChatHub();
|
|
await page.dismissWelcomeScreen();
|
|
await expect(page.getModelSelectorButton()).toContainText(/claude/i); // auto-select a model
|
|
|
|
await page.getFileInput().setInputFiles(testImagePath);
|
|
await page.getChatInput().fill('What color is this image? Reply with just the color name.');
|
|
await page.getSendButton().click();
|
|
|
|
await expect(page.getChatMessages().nth(0)).toContainText('What color is this image?');
|
|
await expect(page.getAttachmentsAt(0)).toHaveCount(1);
|
|
await expect(page.getAttachmentsAt(0).nth(0)).toBeVisible();
|
|
await expect(page.getChatMessages().nth(1)).toContainText(/red/i);
|
|
|
|
// Verify image is persisted and can be opened in new tab
|
|
await n8n.page.reload();
|
|
const newPage = await page.openAttachmentAt(0, 0);
|
|
|
|
await expect(newPage.locator('img')).toBeVisible();
|
|
await expect(newPage.locator('img')).toHaveJSProperty('naturalWidth', 100);
|
|
await expect(newPage.locator('img')).toHaveJSProperty('naturalHeight', 100);
|
|
await newPage.close();
|
|
});
|
|
|
|
test('text file attachment', async ({ n8n, anthropicCredential: _ }) => {
|
|
const page = new ChatHubChatPage(n8n.page);
|
|
|
|
await n8n.navigate.toChatHub();
|
|
await page.dismissWelcomeScreen();
|
|
await expect(page.getModelSelectorButton()).toContainText(/claude/i);
|
|
|
|
await page.getFileInput().setInputFiles(testTextPath);
|
|
await page.getChatInput().fill('What is the exact content of this file?');
|
|
await page.getSendButton().click();
|
|
|
|
await expect(page.getChatMessages().nth(0)).toContainText('What is the exact content');
|
|
await expect(page.getAttachmentsAt(0)).toHaveCount(1);
|
|
await expect(page.getChatMessages().nth(1)).toContainText('I am a file');
|
|
});
|
|
|
|
test('reference attachment in subsequent message', async ({ n8n, anthropicCredential: _ }) => {
|
|
const page = new ChatHubChatPage(n8n.page);
|
|
|
|
await n8n.navigate.toChatHub();
|
|
await page.dismissWelcomeScreen();
|
|
await expect(page.getModelSelectorButton()).toContainText(/claude/i);
|
|
|
|
// Send initial message with attachment
|
|
await page.getFileInput().setInputFiles(testImagePath);
|
|
await page.getChatInput().fill('What color is this image? Reply with just the color name.');
|
|
await page.getSendButton().click();
|
|
|
|
await expect(page.getChatMessages().nth(0)).toContainText('What color is this image?');
|
|
await expect(page.getChatMessages().nth(1)).toContainText(/red/i);
|
|
|
|
// Send follow-up question referencing the image
|
|
await page.getChatInput().fill('Does the image include text? Reply just yes or no.');
|
|
await page.getSendButton().click();
|
|
|
|
await expect(page.getChatMessages().nth(2)).toContainText('Does the image include text?');
|
|
await expect(page.getChatMessages().nth(3)).toContainText(/no/i);
|
|
});
|
|
});
|