mirror of
https://github.com/toeverything/AFFiNE.git
synced 2026-02-13 12:55:00 +00:00
317d3e7ea6
### TL;DR
Split and enhance copilot e2e tests.
### What Changed
#### Tests Structure
The e2e tests are organized into the following categories:
1. **Basic Tests (`/basic`)**: Tests for verifying core AI capabilities including feature onboarding, authorization workflows, and basic chat interactions.
2. **Chat Interaction Tests (`/chat-with`)**: Tests for verifying the AI's interaction with various object types, such as attachments, images, text content, Edgeless elements, etc.
3. **AI Action Tests (`/ai-action`)**: Tests for verifying the AI's actions, such as text translation, gramma correction, etc.
4. **Insertion Tests (`/insertion`)**: Tests for verifying answer insertion functionality.
#### Tests Writing
Writing a copilot test cases is easier and clear
e.g.
```ts
test('support chat with specified doc', async ({ page, utils }) => {
// Initialize the doc
await focusDocTitle(page);
await page.keyboard.insertText('Test Doc');
await page.keyboard.press('Enter');
await page.keyboard.insertText('EEee is a cute cat');
await utils.chatPanel.chatWithDoc(page, 'Test Doc');
await utils.chatPanel.makeChat(page, 'What is EEee?');
await utils.chatPanel.waitForHistory(page, [
{
role: 'user',
content: 'What is EEee?',
},
{
role: 'assistant',
status: 'success',
},
]);
const { content } = await utils.chatPanel.getLatestAssistantMessage(page);
expect(content).toMatch(/EEee/);
});
```
#### Summary
||Cases|
|------|----|
|Before|19||
|After|151||
> Close BS-2769
53 lines
1.7 KiB
TypeScript
53 lines
1.7 KiB
TypeScript
import { loginUser } from '@affine-test/kit/utils/cloud';
|
|
import type { EdgelessRootBlockComponent } from '@blocksuite/affine/blocks/root';
|
|
import type { GfxModel } from '@blocksuite/std/gfx';
|
|
import { expect } from '@playwright/test';
|
|
|
|
import { test } from '../base/base-test';
|
|
|
|
test.describe('AIChatWith/EdgelessMindMap', () => {
|
|
test.beforeEach(async ({ page, utils }) => {
|
|
const user = await utils.testUtils.getUser();
|
|
await loginUser(page, user);
|
|
await utils.testUtils.setupTestEnvironment(page);
|
|
await utils.chatPanel.openChatPanel(page);
|
|
});
|
|
|
|
test('should support replace mindmap with the regenerated one', async ({
|
|
page,
|
|
utils,
|
|
}) => {
|
|
let id: string;
|
|
const { regenerateMindMap } = await utils.editor.askAIWithEdgeless(
|
|
page,
|
|
async () => {
|
|
id = await utils.editor.createMindmap(page);
|
|
},
|
|
async () => {
|
|
const { id: rootId } = await utils.editor.getMindMapNode(page, id!, [
|
|
0,
|
|
]);
|
|
await utils.editor.selectElementInEdgeless(page, [rootId]);
|
|
}
|
|
);
|
|
|
|
const { answer } = await regenerateMindMap();
|
|
await expect(answer.locator('mini-mindmap-preview')).toBeVisible();
|
|
const replace = answer.getByTestId('answer-replace');
|
|
await replace.click();
|
|
|
|
// Expect original mindmap to be replaced
|
|
const mindmaps = await page.evaluate(() => {
|
|
const edgelessBlock = document.querySelector(
|
|
'affine-edgeless-root'
|
|
) as EdgelessRootBlockComponent;
|
|
const mindmaps = edgelessBlock?.gfx.gfxElements
|
|
.filter((el: GfxModel) => 'type' in el && el.type === 'mindmap')
|
|
.map((el: GfxModel) => el.id);
|
|
return mindmaps;
|
|
});
|
|
expect(mindmaps).toHaveLength(1);
|
|
expect(mindmaps?.[0]).not.toBe(id!);
|
|
});
|
|
});
|