mirror of
https://github.com/toeverything/AFFiNE.git
synced 2026-02-15 05:37:32 +00:00
a4680d236d
<!-- This is an auto-generated comment: release notes by coderabbit.ai --> ## Summary by CodeRabbit * **Refactor** * Replaced the boolean flag for selecting AI workflow endpoints with a clear and flexible enum, enhancing clarity and maintainability for AI-powered features. * **Tests** * Simplified example text in AI action tests to improve consistency. <!-- end of auto-generated comment: release notes by coderabbit.ai --> Co-authored-by: DarkSky <25152247+darkskygit@users.noreply.github.com>
90 lines
2.9 KiB
TypeScript
90 lines
2.9 KiB
TypeScript
import { expect } from '@playwright/test';
|
|
|
|
import { test } from '../base/base-test';
|
|
|
|
test.describe('AIAction/MakeItReal', () => {
|
|
test.beforeEach(async ({ loggedInPage: page, utils }) => {
|
|
await utils.testUtils.setupTestEnvironment(page);
|
|
await utils.chatPanel.openChatPanel(page);
|
|
});
|
|
|
|
test('should support making the selected content to real', async ({
|
|
loggedInPage: page,
|
|
utils,
|
|
}) => {
|
|
const { makeItReal } = await utils.editor.askAIWithText(page, 'Hello');
|
|
const { answer, responses } = await makeItReal();
|
|
await expect(answer.locator('iframe')).toBeVisible({ timeout: 30000 });
|
|
expect(responses).toEqual(new Set(['insert-below']));
|
|
});
|
|
|
|
test('should support making the selected text block to real in edgeless', async ({
|
|
loggedInPage: page,
|
|
utils,
|
|
}) => {
|
|
const { makeItReal } = await utils.editor.askAIWithEdgeless(
|
|
page,
|
|
async () => {
|
|
await utils.editor.createEdgelessText(page, 'Hello');
|
|
}
|
|
);
|
|
const { answer, responses } = await makeItReal();
|
|
await expect(answer.locator('iframe')).toBeVisible({ timeout: 30000 });
|
|
expect(responses).toEqual(new Set(['insert-below']));
|
|
});
|
|
|
|
test('should support making the selected note block to real in edgeless', async ({
|
|
loggedInPage: page,
|
|
utils,
|
|
}) => {
|
|
const { makeItReal } = await utils.editor.askAIWithEdgeless(
|
|
page,
|
|
async () => {
|
|
await utils.editor.createEdgelessNote(page, 'Hello');
|
|
}
|
|
);
|
|
const { answer, responses } = await makeItReal();
|
|
await expect(answer.locator('iframe')).toBeVisible({ timeout: 30000 });
|
|
expect(responses).toEqual(new Set(['insert-below']));
|
|
});
|
|
|
|
test('should support making the selected element to real in edgeless', async ({
|
|
loggedInPage: page,
|
|
utils,
|
|
}) => {
|
|
const { makeItReal } = await utils.editor.askAIWithEdgeless(
|
|
page,
|
|
async () => {
|
|
await utils.editor.createShape(page, 'HelloWorld');
|
|
}
|
|
);
|
|
|
|
const { answer, responses } = await makeItReal();
|
|
await expect(answer.locator('iframe')).toBeVisible({ timeout: 30000 });
|
|
expect(responses).toEqual(new Set(['insert-below']));
|
|
});
|
|
|
|
test('should show chat history in chat panel', async ({
|
|
loggedInPage: page,
|
|
utils,
|
|
}) => {
|
|
const { makeItReal } = await utils.editor.askAIWithText(page, 'Hello');
|
|
const { answer } = await makeItReal();
|
|
const insert = answer.getByTestId('answer-insert-below');
|
|
await insert.click();
|
|
await utils.chatPanel.waitForHistory(page, [
|
|
{
|
|
role: 'action',
|
|
},
|
|
]);
|
|
const {
|
|
answer: panelAnswer,
|
|
prompt,
|
|
actionName,
|
|
} = await utils.chatPanel.getLatestAIActionMessage(page);
|
|
await expect(panelAnswer.locator('affine-code')).toBeVisible();
|
|
await expect(prompt).toHaveText(/Write a web page of follow text/);
|
|
await expect(actionName).toHaveText(/Make it real with text/);
|
|
});
|
|
});
|