mirror of
https://github.com/linshenkx/prompt-optimizer.git
synced 2026-05-07 14:06:53 +08:00
183 lines
7.3 KiB
TypeScript
183 lines
7.3 KiB
TypeScript
import { test, expect } from '../fixtures'
|
||
import { navigateToMode } from '../helpers/common'
|
||
import {
|
||
fillOriginalPrompt,
|
||
clickAnalyzeButton,
|
||
getEvaluationScore,
|
||
closeEvaluationPanelIfOpen,
|
||
verifyAnalyzeButtonDisabledWhenEmpty,
|
||
getWorkspace,
|
||
} from '../helpers/analysis'
|
||
import {
|
||
closeEvaluationDrawer,
|
||
expectPromptVersionTagVisible,
|
||
openEvaluationDrawerFromBadge,
|
||
} from '../helpers/evaluation'
|
||
import {
|
||
clickOptimizeButton,
|
||
expectOptimizedResultNotEmpty,
|
||
expectOutputByTestIdNotEmpty,
|
||
readOutputByTestIdText,
|
||
} from '../helpers/optimize'
|
||
|
||
/**
|
||
* Basic User 模式 - 提示词分析测试
|
||
*
|
||
* ✨ 最佳范式示例：
|
||
* - 使用 data-testid 精确定位，不依赖文本内容
|
||
* - 容器隔离：通过 data-mode 区分不同工作区
|
||
* - 类型安全：使用 TypeScript 类型定义
|
||
* - 代码复用：与 basic-system 使用相同的辅助函数
|
||
*
|
||
* 功能：分析用户提示词并显示评估分数
|
||
*
|
||
* 前提：
|
||
* - .env.local 已配置 API keys
|
||
* - 实际调用 LLM API（会产生费用）
|
||
*
|
||
* 测试流程：
|
||
* 1. 导航到 basic-user 工作区
|
||
* 2. 填写提示词
|
||
* 3. 点击"分析"按钮
|
||
* 4. 等待 LLM 响应
|
||
* 5. 验证评估结果和分数显示
|
||
*/
|
||
|
||
const MODE = 'basic-user' as const
|
||
const STALE_COMPARE_MESSAGE =
|
||
/测试或工作区已变更，建议重新对比。|The test setup or workspace has changed\. Re-run the comparison if needed\./i
|
||
|
||
test.describe('Basic User - 提示词分析', () => {
|
||
test('分析提示词并显示评估结果', async ({ page }) => {
|
||
test.setTimeout(180000) // 3分钟超时
|
||
|
||
// 1. 导航到 basic-user 工作区
|
||
await navigateToMode(page, 'basic', 'user')
|
||
|
||
// 2. 填写提示词（使用 data-testid 定位）
|
||
const testPrompt = '帮我写一封邮件，关于项目进度汇报'
|
||
await fillOriginalPrompt(page, MODE, testPrompt)
|
||
|
||
// 3. 点击分析按钮（使用 data-testid 定位）
|
||
await clickAnalyzeButton(page, MODE)
|
||
|
||
// 4. 验证评估分数（使用 data-testid 定位）
|
||
const score = await getEvaluationScore(page, MODE, 'prompt-only')
|
||
expect(score).toBeGreaterThan(0)
|
||
expect(score).toBeLessThanOrEqual(100)
|
||
|
||
const workspace = getWorkspace(page, MODE)
|
||
const analysisBadge = workspace.locator('[data-testid="score-badge-prompt-only"]')
|
||
const drawer = await openEvaluationDrawerFromBadge(analysisBadge)
|
||
await expect(drawer.getByTestId('evaluation-panel-rewrite-from-evaluation')).toBeVisible()
|
||
await expect(drawer.getByTestId('evaluation-panel-rewrite-from-evaluation')).toBeEnabled()
|
||
await closeEvaluationDrawer(drawer)
|
||
})
|
||
|
||
test('验证分析按钮在没有提示词时禁用', async ({ page }) => {
|
||
await navigateToMode(page, 'basic', 'user')
|
||
|
||
// 分析按钮应该在没有输入时禁用
|
||
await verifyAnalyzeButtonDisabledWhenEmpty(page, MODE)
|
||
})
|
||
|
||
test('分析后右侧 workspace 测试应切换到新的 V0 而不是继续沿用旧链', async ({ page }) => {
|
||
test.setTimeout(240000)
|
||
|
||
const oldToken = 'OLDQ7'
|
||
const newToken = 'NEWV0'
|
||
|
||
await navigateToMode(page, 'basic', 'user')
|
||
|
||
await fillOriginalPrompt(page, MODE, `请只输出 ${oldToken}`)
|
||
await clickOptimizeButton(page, MODE)
|
||
await expectOptimizedResultNotEmpty(page, MODE)
|
||
|
||
await fillOriginalPrompt(page, MODE, `请只输出 ${newToken}`)
|
||
await clickAnalyzeButton(page, MODE)
|
||
await getEvaluationScore(page, MODE, 'prompt-only')
|
||
await closeEvaluationPanelIfOpen(page)
|
||
|
||
const workspace = getWorkspace(page, MODE)
|
||
await workspace.locator('[data-testid="basic-user-test-run-b"]').click()
|
||
await expectOutputByTestIdNotEmpty(page, 'basic-user-test-optimized-output')
|
||
|
||
const output = await readOutputByTestIdText(page, 'basic-user-test-optimized-output')
|
||
expect(output).toContain(newToken)
|
||
expect(output).not.toContain(oldToken)
|
||
})
|
||
|
||
test('对比评估在工作区内容变更后应保留并标记为过期', async ({ page }) => {
|
||
test.setTimeout(300000)
|
||
|
||
await navigateToMode(page, 'basic', 'user')
|
||
|
||
await fillOriginalPrompt(page, MODE, '你是一个中文文案助手。请将用户输入改写成更正式、更清晰的一句话。')
|
||
await clickAnalyzeButton(page, MODE)
|
||
await getEvaluationScore(page, MODE, 'prompt-only')
|
||
await closeEvaluationPanelIfOpen(page)
|
||
|
||
const workspace = getWorkspace(page, MODE)
|
||
await workspace.locator('[data-testid="basic-user-test-run-all"]').click()
|
||
await expectOutputByTestIdNotEmpty(page, 'basic-user-test-original-output')
|
||
await expectOutputByTestIdNotEmpty(page, 'basic-user-test-optimized-output')
|
||
|
||
await workspace.locator('.test-area-top').first().getByTestId('focus-analyze-main').click()
|
||
await getEvaluationScore(page, MODE, 'compare')
|
||
|
||
const workspaceOutput = workspace.locator('[data-testid="basic-user-output"]').first()
|
||
const workspaceEditor = workspaceOutput.locator('textarea')
|
||
if ((await workspaceEditor.count()) > 0) {
|
||
await workspaceEditor.first().fill(
|
||
'你是一个中文文案助手。请将用户输入改写成更正式、更清晰的一句话。\n补充要求：务必保持原意，不要扩写。'
|
||
)
|
||
} else {
|
||
const cmContent = workspaceOutput.locator('.cm-content').first()
|
||
await cmContent.click()
|
||
await page.keyboard.press(process.platform === 'darwin' ? 'Meta+A' : 'Control+A')
|
||
await page.keyboard.type(
|
||
'你是一个中文文案助手。请将用户输入改写成更正式、更清晰的一句话。\n补充要求：务必保持原意，不要扩写。'
|
||
)
|
||
}
|
||
|
||
const compareBadge = workspace.locator('[data-testid="score-badge-compare"]')
|
||
await expect(compareBadge).toBeVisible()
|
||
await expect(compareBadge).toHaveClass(/evaluation-score-badge-btn--stale/)
|
||
|
||
await compareBadge.click()
|
||
await expect(page.locator('.evaluation-hover-card:visible').getByText(STALE_COMPARE_MESSAGE)).toBeVisible()
|
||
})
|
||
|
||
test('点击 V0 后重新分析应回到 prompt-only，而不是沿用旧版本的 iterate 状态', async ({ page }) => {
|
||
test.setTimeout(360000)
|
||
|
||
await navigateToMode(page, 'basic', 'user')
|
||
|
||
await fillOriginalPrompt(page, MODE, '请把用户输入改写成更礼貌、更简洁的一句话。')
|
||
await clickOptimizeButton(page, MODE)
|
||
await expectOptimizedResultNotEmpty(page, MODE)
|
||
|
||
const workspace = getWorkspace(page, MODE)
|
||
await workspace.getByTestId('prompt-panel-continue-optimize').click()
|
||
const iterateModal = page.getByTestId('prompt-panel-iterate-modal')
|
||
await expect(iterateModal).toBeVisible({ timeout: 15000 })
|
||
await iterateModal.getByTestId('prompt-panel-iterate-input').locator('textarea').fill(
|
||
'请进一步统一输出口吻，并补充明确的改写边界。'
|
||
)
|
||
await iterateModal.getByTestId('prompt-panel-iterate-submit').click()
|
||
|
||
await expectPromptVersionTagVisible(page, 2)
|
||
await expectPromptVersionTagVisible(page, 1)
|
||
const focusAnalyzeGroup = workspace.locator('.evaluation-entry .focus-analyze-group')
|
||
await expect(focusAnalyzeGroup).toBeVisible({ timeout: 15000 })
|
||
await expect(focusAnalyzeGroup).toHaveAttribute('data-evaluation-type', 'prompt-iterate')
|
||
|
||
await workspace.locator('[data-testid="prompt-panel-version-tag-v0"]').click()
|
||
await page.mouse.move(10, 10)
|
||
|
||
await expect(focusAnalyzeGroup).toHaveAttribute('data-evaluation-type', 'prompt-only', {
|
||
timeout: 15000,
|
||
})
|
||
})
|
||
})
|