mirror of
https://github.com/linshenkx/prompt-optimizer.git
synced 2026-05-09 06:58:40 +08:00
94 lines
3.2 KiB
TypeScript
94 lines
3.2 KiB
TypeScript
import { test, expect } from '../fixtures'
|
||
|
||
/**
|
||
* Pro Variable 模式 - Session 持久化测试
|
||
*
|
||
* 注意：当前 session 使用 PreferenceService（IndexedDB 等）持久化，
|
||
* 不再依赖 localStorage，因此测试应以 UI 状态为准。
|
||
*/
|
||
test.describe('Pro Variable - Session Persistence', () => {
|
||
const normalizeText = (text: string | null | undefined) =>
|
||
String(text || '').replace(/\s+/g, ' ').trim()
|
||
|
||
const gotoMode = async (page: any, route: string) => {
|
||
await page.goto('/')
|
||
await page.waitForLoadState('networkidle')
|
||
await page.goto(route)
|
||
await page.waitForLoadState('networkidle')
|
||
await page.waitForTimeout(1500)
|
||
}
|
||
|
||
const getSelectByLabel = async (page: any, label: RegExp) => {
|
||
const labelEl = page.getByText(label).first()
|
||
await expect(labelEl).toBeVisible({ timeout: 20000 })
|
||
|
||
const container = labelEl.locator(
|
||
'xpath=ancestor::*[.//div[contains(@class,"n-base-selection")]][1]'
|
||
)
|
||
const select = container.locator('.n-base-selection').first()
|
||
await expect(select).toBeVisible({ timeout: 20000 })
|
||
return select
|
||
}
|
||
|
||
test('切换优化模型后刷新页面，选择应该保留', async ({ page }) => {
|
||
await gotoMode(page, '/#/pro/variable')
|
||
|
||
const select = await getSelectByLabel(page, /优化模型|Optimization Model/i)
|
||
await select.click()
|
||
|
||
const optionLocator = page.locator('.n-base-select-option')
|
||
await expect(optionLocator.first()).toBeVisible({ timeout: 20000 })
|
||
|
||
const count = await optionLocator.count()
|
||
expect(count).toBeGreaterThan(0)
|
||
if (count < 2) test.skip(true, 'only one model option')
|
||
|
||
const target = normalizeText(await optionLocator.nth(1).textContent())
|
||
await optionLocator.nth(1).click()
|
||
|
||
await expect
|
||
.poll(async () => normalizeText(await select.textContent()), { timeout: 20000 })
|
||
.toBe(target)
|
||
|
||
await page.reload()
|
||
await page.waitForLoadState('networkidle')
|
||
await page.waitForTimeout(1500)
|
||
|
||
const selectAfter = await getSelectByLabel(page, /优化模型|Optimization Model/i)
|
||
await expect
|
||
.poll(async () => normalizeText(await selectAfter.textContent()), { timeout: 20000 })
|
||
.toBe(target)
|
||
})
|
||
|
||
test('切换模板后刷新页面，选择应该保留', async ({ page }) => {
|
||
await gotoMode(page, '/#/pro/variable')
|
||
|
||
const select = await getSelectByLabel(page, /优化提示词模板|Optimization Template/i)
|
||
await select.click()
|
||
|
||
const optionLocator = page.locator('.n-base-select-option')
|
||
await expect(optionLocator.first()).toBeVisible({ timeout: 20000 })
|
||
|
||
const count = await optionLocator.count()
|
||
expect(count).toBeGreaterThan(0)
|
||
if (count < 2) test.skip(true, 'only one template option')
|
||
|
||
const target = normalizeText(await optionLocator.nth(1).textContent())
|
||
await optionLocator.nth(1).click()
|
||
|
||
await expect
|
||
.poll(async () => normalizeText(await select.textContent()), { timeout: 20000 })
|
||
.toBe(target)
|
||
|
||
await page.reload()
|
||
await page.waitForLoadState('networkidle')
|
||
await page.waitForTimeout(1500)
|
||
|
||
const selectAfter = await getSelectByLabel(page, /优化提示词模板|Optimization Template/i)
|
||
await expect
|
||
.poll(async () => normalizeText(await selectAfter.textContent()), { timeout: 20000 })
|
||
.toBe(target)
|
||
})
|
||
})
|
||
|