twitter-lite/e2e/integrations/research.test.ts

import type { ResearchRun } from '../../src/features/research/model'
import { expect, test } from '../fixtures'

test('pushes the current research snapshot on each SSE connection', async ({
  page,
}) => {
  await page.goto('/')
  const snapshots = await page.evaluate(async () => {
    const connect = () =>
      new Promise<unknown>((resolve, reject) => {
        const events = new EventSource('/api/research/events')
        const timer = setTimeout(() => {
          events.close()
          reject(new Error('SSE timed out'))
        }, 5000)
        events.onmessage = (event) => {
          clearTimeout(timer)
          events.close()
          resolve(JSON.parse(event.data))
        }
        events.onerror = () => {
          clearTimeout(timer)
          events.close()
          reject(new Error('SSE failed'))
        }
      })
    return [await connect(), await connect()]
  })
  expect(snapshots).toEqual([
    { configured: false, run: null },
    { configured: false, run: null },
  ])
})

test('restores the conversation and growing temporary deck from reconnect snapshots', async ({
  page,
}) => {
  await page.goto('/')
  await page.getByRole('button', { name: 'カラムを追加', exact: true }).click()
  const accounts = page.getByRole('combobox', {
    name: '接続プロファイル',
    exact: true,
  })
  await accounts.selectOption({ label: 'e2e' })
  const connectionId = await accounts.inputValue()
  await page.keyboard.press('Escape')
  const run: ResearchRun = {
    id: '36ad8cc7-318c-46ae-94fa-28d30336f027',
    threadId: 'conversation-test',
    topic: '既存の観点を使って調査して',
    status: 'running',
    startedAt: Date.now(),
    deckVersion: 1,
    message: '既存の検索条件を再利用しています。',
    messages: [
      { id: 'user-1', role: 'user', text: '既存の観点を使って調査して' },
      { id: 'tool-1', role: 'tool', text: 'list_decks: 完了' },
      {
        id: 'assistant-1',
        role: 'assistant',
        text: '既存の検索条件を再利用しています。',
      },
    ],
    deck: {
      id: 'generated-plan',
      title: '会話からの調査',
      columns: [
        {
          id: 'one',
          title: '最初の観点',
          connectionId,
          source: {
            platform: 'twitter',
            kind: 'search',
            query: 'WebMCP',
            product: 'Latest',
            following: false,
          },
        },
      ],
    },
  }
  await page.route('**/api/research/events', (route) =>
    route.fulfill({
      contentType: 'text/event-stream',
      body: `data: ${JSON.stringify({ configured: true, run })}\n\n`,
    }),
  )
  await page.reload()
  const conversation = page.getByRole('log', { name: '調査の会話' })
  await expect(conversation).toContainText('既存の観点を使って調査して')
  await expect(conversation).toContainText('list_decks: 完了')
  await expect(conversation).toContainText('既存の検索条件を再利用しています。')
  await expect(page.locator('.deck-column h2')).toHaveText(['最初の観点'])
  run.deckVersion = 2
  run.status = 'complete'
  run.messages.push({
    id: 'assistant-2',
    role: 'assistant',
    text: '比較する観点を追加しました。',
  })
  run.deck?.columns.push({
    id: 'two',
    title: '比較の観点',
    connectionId,
    source: {
      platform: 'twitter',
      kind: 'search',
      query: 'WebMCP lang:ja',
      product: 'Latest',
      following: false,
    },
  })
  await page.reload()
  await expect(conversation).toContainText('既存の観点を使って調査して')
  await expect(conversation).toContainText('比較する観点を追加しました。')
  await expect(page.locator('.deck-column h2')).toHaveText([
    '最初の観点',
    '比較の観点',
  ])
  await expect(
    page.getByRole('button', { name: 'デッキとして保存', exact: true }),
  ).toBeEnabled()
  await page.getByLabel('メッセージ', { exact: true }).fill('この観点を詳しく')
  await expect(
    page.getByRole('button', { name: '送信', exact: true }),
  ).toBeEnabled()
})

test('rejects a different tailnet owner at the SSE endpoint', async ({
  request,
}) => {
  const response = await request.get('/api/research/events', {
    headers: { 'Tailscale-User-Login': '[email protected]' },
  })
  expect(response.status()).toBe(403)
})

test('explains unconfigured Codex and prevents starting research without affecting the deck', async ({
  page,
  a11y,
}) => {
  await page.goto('/')
  const heading = page.getByRole('heading', { level: 1 })
  await expect(heading).toBeVisible()
  const originalTitle = await heading.textContent()
  const panel = page.getByRole('complementary', { name: '調査チャット' })
  await expect(panel).toBeVisible()
  await expect(
    panel.getByText('自宅のCodexへの接続が設定されていません。'),
  ).toBeVisible()
  await expect(
    panel.getByRole('button', { name: '送信', exact: true }),
  ).toBeDisabled()
  await expect(panel.getByLabel('メッセージ', { exact: true })).toBeDisabled()
  const accessibility = await a11y().analyze()
  expect(accessibility.violations).toEqual([])
  await expect(heading).toHaveText(originalTitle ?? '')
  await page.getByRole('button', { name: 'デッキを作成', exact: true }).click()
  await page.getByLabel('新しいデッキ名').fill('手動の調査')
  await page.getByRole('button', { name: '作成', exact: true }).click()
  await expect(heading).toHaveText('手動の調査')
  await expect(panel).toBeVisible()
})