yaoyao/tests/e2e/hermesMedia.spec.ts

import { expect, test, type APIRequestContext, type Locator } from '@playwright/test'

const mediaPaths = [
  '/Users/samien/.hermes/profiles/yaoer/cache/images/openai_codex_gpt-image-2-high_20260902_194820_1d225f08.png',
  '/Users/samien/.hermes/profiles/yaoer/cache/images/国漫三视图.png',
]
const mediaNames = mediaPaths.map(path => path.split('/').at(-1)!)
const mediaLabels = ['3D 半写实女性四视图', '国漫三视图']

async function signIn(request: APIRequestContext, origin: string) {
  const bootstrap = await (await request.get('/api/app/bootstrap')).json()
  if (bootstrap.setupRequired) {
    const setup = await request.post('/api/app/setup', {
      headers: { Origin: origin, 'X-CSRF-Token': bootstrap.csrfToken },
      data: { username: 'admin', password: 'e2e-password' },
    })
    expect(setup.ok()).toBe(true)
    return
  }
  const login = await request.post('/api/app/login', {
    headers: { Origin: origin, 'X-CSRF-Token': bootstrap.csrfToken },
    data: { username: 'admin', password: 'e2e-password' },
  })
  expect(login.ok()).toBe(true)
}

async function expectDecodedImage(image: Locator) {
  await expect(image).toBeVisible()
  await expect.poll(() => image.evaluate(element => {
    const img = element instanceof HTMLImageElement ? element : element.querySelector('img')
    return img?.naturalWidth ?? 0
  })).toBeGreaterThan(0)
}

test('loads Hermes profile cache images through the Web server and switches their full-size previews', async ({ page, baseURL }, testInfo) => {
  await signIn(page.request, new URL(baseURL!).origin)
  // No image interception: this exercises the real Web route, upstream account
  // authentication, attachment-to-inline response handling, and browser decode.
  const responses = mediaPaths.map(path => page.waitForResponse(response => (
    new URL(response.url()).pathname === '/api/files/download' && new URL(response.url()).searchParams.get('path') === path
  )))
  await page.goto('/chat/session-media?profile=yaoer')
  await expect(page).toHaveTitle('瑶儿生成图片验收 · 夭夭 AI')

  for (const [index, name] of mediaLabels.entries()) {
    const image = page.getByRole('button', { name: `预览图片 ${name}`, exact: true })
    await image.scrollIntoViewIfNeeded()
    await expectDecodedImage(image)
    const response = await responses[index]!
    expect(response.status()).toBe(200)
    expect(response.headers()['content-type']).toContain('image/png')
    expect(response.headers()['content-disposition'] ?? '').not.toMatch(/^attachment(?:;|$)/)
    expect(new URL(response.url()).origin).toBe(new URL(baseURL!).origin)
    expect(new URL(response.url()).searchParams.get('profile')).toBe('yaoer')
  }

  await page.getByRole('button', { name: `预览图片 ${mediaLabels[0]}`, exact: true }).click()
  const preview = page.getByRole('dialog')
  await expect(preview).toHaveAccessibleName(`预览 ${mediaNames[0]}`)
  await expectDecodedImage(preview.getByRole('img', { name: mediaNames[0], exact: true }))
  await expect(preview.getByRole('button', { name: '下一张媒体' })).toBeVisible()

  await preview.getByRole('button', { name: '下一张媒体' }).click()
  await expect(preview).toHaveAccessibleName(`预览 ${mediaNames[1]}`)
  const secondPreview = preview.getByRole('img', { name: mediaNames[1], exact: true })
  await expectDecodedImage(secondPreview)
  expect(new URL(await secondPreview.getAttribute('src') || '', baseURL).searchParams.get('path')).toBe(mediaPaths[1])
  await expect(preview.getByRole('button', { name: '上一张媒体' })).toBeVisible()
  await expect(preview.getByRole('button', { name: '下一张媒体' })).toBeHidden()

  await page.keyboard.press('ArrowLeft')
  await expect(preview).toHaveAccessibleName(`预览 ${mediaNames[0]}`)
  await expectDecodedImage(preview.getByRole('img', { name: mediaNames[0], exact: true }))
  await expect(preview.getByRole('button', { name: '下一张媒体' })).toBeVisible()
  await page.screenshot({ path: testInfo.outputPath('hermes-profile-cache-preview.png') })
  await page.keyboard.press('Escape')
  await expect(preview).toBeHidden()
})

test('restores nine user Profile uploads as images and opens the whole batch in the preview', async ({ page, baseURL }, testInfo) => {
  await signIn(page.request, new URL(baseURL!).origin)
  await page.goto('/chat/session-user-media?profile=yaoer')
  const message = page.locator('.message--user')
  await expect(message).toContainText('读取图片中的提示词')
  await expect(message).not.toContainText('@image:')
  await expect(message).not.toContainText('[用户附加图片')
  await expect(message).not.toContainText('[screenshot]')
  const images = message.locator('img')
  await expect(images).toHaveCount(9)
  for (const image of await images.all()) {
    await image.scrollIntoViewIfNeeded()
    await expectDecodedImage(image)
    const url = new URL(await image.getAttribute('src') || '', baseURL)
    expect(url.origin).toBe(new URL(baseURL!).origin)
    expect(url.pathname).toBe('/api/files/download')
    expect(url.searchParams.get('path')).toContain('/.hermes/profiles/yaoer/images/')
    expect(url.searchParams.get('profile')).toBe('yaoer')
  }
  await page.screenshot({ path: testInfo.outputPath('user-profile-images.png') })
  await images.first().click()
  const preview = page.locator('.image-preview-overlay')
  await expect(preview.locator('.image-preview-counter')).toHaveText('1 / 9')
  await expectDecodedImage(preview.getByRole('img', { name: '照片-1.png', exact: true }))
  await preview.getByRole('button', { name: '下一张媒体' }).click()
  await expect(preview.locator('.image-preview-counter')).toHaveText('2 / 9')
  await expectDecodedImage(preview.getByRole('img', { name: '照片-2.png', exact: true }))
  await page.screenshot({ path: testInfo.outputPath('user-profile-image-preview.png') })
})

test('server permissions reject a previously readable image and retry succeeds after authorization', async ({page,baseURL}) => {
  const origin=new URL(baseURL!).origin
  await signIn(page.request,origin)
  const capabilities=await(await page.request.get('/api/realtime/capabilities')).json()
  const headers={Origin:origin,'X-CSRF-Token':capabilities.csrfToken}
  const policy=async(mode:'folders'|'all')=>{
    const response=await page.request.put('/api/app/settings/file-access',{headers,data:{mode,folders:[]}})
    expect(response.ok(),await response.text()).toBe(true)
  }
  await policy('folders')
  try {
    await page.goto('/chat/session-media?profile=yaoer')
    const failure=page.getByRole('button',{name:`${mediaLabels[0]} · 加载失败，点击重试`,exact:true})
    await expect(failure).toBeVisible()
    const denied=await page.request.get('/api/files/download',{params:{path:mediaPaths[0]!,profile:'yaoer',preview:'1'}})
    expect(denied.status()).toBe(403)
    await policy('all')
    await failure.click()
    await expectDecodedImage(page.getByRole('button',{name:`预览图片 ${mediaLabels[0]}`,exact:true}))
    await expect(failure).toHaveCount(0)
  } finally {await policy('all')}
})