@ploykit/ui 0.6.0 → 0.6.2

package/src/components/__tests__/ErrorBoundary.test.tsx

@@ -0,0 +1,61 @@
+import '@testing-library/jest-dom/vitest'
+import { render, screen, fireEvent } from '@testing-library/react'
+import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest'
+import { ErrorBoundary } from '../ErrorBoundary'
+
+
+let shouldThrow = false
+function Bomb() {
+  if (shouldThrow) throw new Error('炸弹爆炸')
+  return <div>正常内容</div>
+}
+
+describe('ErrorBoundary', () => {
+  beforeEach(() => {
+    shouldThrow = false
+    vi.spyOn(console, 'error').mockImplementation(() => {})
+  })
+  afterEach(() => {
+    vi.restoreAllMocks()
+  })
+
+  it('子组件 throw 时渲染 fallback 文案', () => {
+    shouldThrow = true
+    render(
+      <ErrorBoundary fallback={<div>自定义兜底</div>}>
+        <Bomb />
+      </ErrorBoundary>,
+    )
+    expect(screen.getByText('自定义兜底')).toBeInTheDocument()
+    expect(screen.queryByText('正常内容')).not.toBeInTheDocument()
+  })
+
+  it('函数式 fallback 的 reset() 后恢复渲染 children', () => {
+    shouldThrow = true
+    render(
+      <ErrorBoundary fallback={(_error, reset) => <button onClick={reset}>重试</button>}>
+        <Bomb />
+      </ErrorBoundary>,
+    )
+    shouldThrow = false 
+    fireEvent.click(screen.getByRole('button', { name: '重试' }))
+    expect(screen.getByText('正常内容')).toBeInTheDocument()
+  })
+
+  it('无 fallback 时显示默认错误态（含"出错了"与重试按钮）', () => {
+    shouldThrow = true
+    render(<ErrorBoundary><Bomb /></ErrorBoundary>)
+    expect(screen.getByText('出错了')).toBeInTheDocument()
+    expect(screen.getByRole('button', { name: '重试' })).toBeInTheDocument()
+  })
+
+  it('componentDidCatch 将错误记入 console.error', () => {
+    shouldThrow = true
+    render(
+      <ErrorBoundary fallback={<div>ok</div>}>
+        <Bomb />
+      </ErrorBoundary>,
+    )
+    expect(console.error).toHaveBeenCalled()
+  })
+})

package/src/components/__tests__/ImpersonationBanner.test.tsx

@@ -0,0 +1,59 @@
+import '@testing-library/jest-dom/vitest'
+import { render, screen } from '@testing-library/react'
+import { describe, it, expect, vi, beforeEach } from 'vitest'
+import { ImpersonationBanner } from '../ImpersonationBanner'
+import type { PKUser } from '../../provider/PloykitProvider'
+
+
+
+const mocks = vi.hoisted(() => ({ impersonatedBy: undefined as string | undefined, user: null as PKUser | null }))
+vi.mock('../../../../hooks/src/hooks/useAuth', () => ({
+  useAuth: () => ({
+    user: mocks.user,
+    workspaces: [],
+    loading: false,
+    refresh: vi.fn(),
+    logout: vi.fn(async () => {}),
+  }),
+}))
+
+function impersonatedUser(impersonatedBy?: string): PKUser {
+  return {
+    email: 'bob@example.com',
+    display_name: 'Bob',
+    impersonated_by: impersonatedBy,
+  } as PKUser
+}
+
+describe('ImpersonationBanner（ADR 0008 模拟会话横幅）', () => {
+  beforeEach(() => {
+    mocks.impersonatedBy = undefined
+    mocks.user = null
+  })
+
+  it('模拟会话（impersonated_by 非空）：显示"您正处于管理员模拟会话"警示条', () => {
+    mocks.user = impersonatedUser('11111111-2222-3333-4444-555555555555')
+    render(<ImpersonationBanner />)
+    expect(screen.getByText('您正处于管理员模拟会话')).toBeInTheDocument()
+    expect(screen.getByText('您正处于管理员模拟会话')).toBeVisible()
+  })
+
+  it('普通会话（无 impersonated_by 字段）：不渲染横幅、不留占位', () => {
+    mocks.user = impersonatedUser(undefined)
+    const { container } = render(<ImpersonationBanner />)
+    expect(screen.queryByText('您正处于管理员模拟会话')).not.toBeInTheDocument()
+    expect(container.querySelector('[data-slot="impersonation-banner"]')).not.toBeInTheDocument()
+  })
+
+  it('未登录（user=null）：不渲染横幅', () => {
+    mocks.user = null
+    render(<ImpersonationBanner />)
+    expect(screen.queryByText('您正处于管理员模拟会话')).not.toBeInTheDocument()
+  })
+
+  it('横幅不提供退出按钮：退出由发起管理员控制（ADR 0008 裁决）', () => {
+    mocks.user = impersonatedUser('11111111-2222-3333-4444-555555555555')
+    render(<ImpersonationBanner />)
+    expect(screen.queryByRole('button')).not.toBeInTheDocument()
+  })
+})

package/src/components/__tests__/NotificationBell.test.tsx

@@ -0,0 +1,207 @@
+import '@testing-library/jest-dom/vitest'
+import { fireEvent, render, screen, waitFor } from '@testing-library/react'
+import { QueryClient, QueryClientProvider } from '@tanstack/react-query'
+import { describe, it, expect, vi, beforeEach } from 'vitest'
+import { NotificationBell } from '../NotificationBell'
+
+
+
+
+
+
+
+const mockedGet = vi.hoisted(() => vi.fn())
+const mockedPost = vi.hoisted(() => vi.fn())
+vi.mock('@ploykit/client', () => ({
+  api: { get: mockedGet, post: mockedPost },
+  setWorkspaceIdProvider: vi.fn(),
+}))
+
+type N = {
+  id: string; type: string; title: string; body: string
+  link?: string | null; count: number; read_at?: string | null; created_at: string
+}
+
+const now = '2026-10-06T08:00:00Z'
+const unreadItem: N = {
+  id: 'n-1', type: 'quota_near_limit', title: '用量接近上限',
+  body: '本月任务数已用 90%', link: '/settings/workspace/usage', count: 1,
+  read_at: null, created_at: now,
+}
+const readItem: N = {
+  id: 'n-2', type: 'task_comment', title: '已读的历史通知', body: '',
+  count: 2, read_at: '2026-10-05T08:00:00Z', created_at: '2026-10-05T07:00:00Z',
+}
+
+
+function dispatchGet(badgeCount: number, list: N[]) {
+  return (url: string) => {
+    if (url === '/api/notifications/badge') return Promise.resolve({ count: badgeCount })
+    if (url.startsWith('/api/notifications?')) return Promise.resolve(list)
+    return Promise.reject(new Error(`unexpected GET ${url}`))
+  }
+}
+
+function countCalls(urlPrefix: string) {
+  return mockedGet.mock.calls.filter(([u]) => (u as string).startsWith(urlPrefix)).length
+}
+
+function renderBell(onNavigate?: (link: string) => void) {
+  const qc = new QueryClient({ defaultOptions: { queries: { retry: false } } })
+  return render(
+    <QueryClientProvider client={qc}>
+      <NotificationBell onNavigate={onNavigate} />
+    </QueryClientProvider>,
+  )
+}
+
+beforeEach(() => {
+  mockedGet.mockReset()
+  mockedPost.mockReset()
+  mockedPost.mockResolvedValue(undefined)
+})
+
+describe('NotificationBell', () => {
+  it('未读角标：badge 查询渲染服务端未读数；0 时无角标', async () => {
+    mockedGet.mockImplementation(dispatchGet(3, []))
+    const { unmount } = renderBell()
+    await waitFor(() => expect(screen.getByText('3')).toBeInTheDocument())
+    unmount()
+
+    mockedGet.mockImplementation(dispatchGet(0, []))
+    renderBell()
+    
+    await waitFor(() => expect(countCalls('/api/notifications/badge')).toBe(2))
+    
+    expect(screen.getByRole('button', { name: '通知' })).toBeInTheDocument()
+    expect(screen.queryByText('0')).not.toBeInTheDocument()
+  })
+
+  it('未读数超 99 截断显示 99+', async () => {
+    mockedGet.mockImplementation(dispatchGet(120, []))
+    renderBell()
+    await waitFor(() => expect(screen.getByText('99+')).toBeInTheDocument())
+  })
+
+  it('打开下拉加载最近通知（limit=8）；关闭态不拉列表', async () => {
+    const get = dispatchGet(1, [unreadItem, readItem])
+    mockedGet.mockImplementation(get)
+    renderBell()
+
+    
+    await waitFor(() => expect(countCalls('/api/notifications?')).toBe(0))
+
+    fireEvent.click(screen.getByRole('button', { name: '通知' }))
+    expect(await screen.findByText('用量接近上限')).toBeInTheDocument()
+    expect(screen.getByText('已读的历史通知')).toBeInTheDocument()
+    expect(mockedGet).toHaveBeenCalledWith('/api/notifications?limit=8')
+  })
+
+  it('单条已读：POST /:id/read 成功后按前缀失效 badge（立即重拉）与 list（重开面板时重拉）', async () => {
+    mockedGet.mockImplementation(dispatchGet(2, [unreadItem]))
+    renderBell()
+    fireEvent.click(screen.getByRole('button', { name: '通知' }))
+    await screen.findByText('用量接近上限')
+    const badgeCallsBefore = countCalls('/api/notifications/badge')
+    const listCallsBefore = countCalls('/api/notifications?')
+
+    
+    
+    fireEvent.click(screen.getByText('用量接近上限'))
+
+    await waitFor(() => expect(mockedPost).toHaveBeenCalledWith('/api/notifications/n-1/read'))
+    await waitFor(() => {
+      expect(countCalls('/api/notifications/badge')).toBeGreaterThan(badgeCallsBefore)
+    }, { timeout: 2000 })
+    expect(mockedPost).not.toHaveBeenCalledWith('/api/notifications/read-all')
+
+    
+    fireEvent.click(screen.getByRole('button', { name: '通知' }))
+    await waitFor(() => {
+      expect(countCalls('/api/notifications?')).toBeGreaterThan(listCallsBefore)
+    }, { timeout: 2000 })
+  })
+
+  it('全部已读：POST /read-all 并失效查询', async () => {
+    mockedGet.mockImplementation(dispatchGet(2, [unreadItem, readItem]))
+    renderBell()
+    fireEvent.click(screen.getByRole('button', { name: '通知' }))
+    fireEvent.click(await screen.findByText('全部已读'))
+
+    await waitFor(() => expect(mockedPost).toHaveBeenCalledWith('/api/notifications/read-all'))
+    await waitFor(() => {
+      expect(countCalls('/api/notifications/badge')).toBeGreaterThanOrEqual(2)
+    }, { timeout: 2000 })
+  })
+
+  it('外点关闭：面板外 mousedown 关闭下拉', async () => {
+    mockedGet.mockImplementation(dispatchGet(1, [unreadItem]))
+    renderBell()
+    fireEvent.click(screen.getByRole('button', { name: '通知' }))
+    await screen.findByText('用量接近上限')
+
+    fireEvent.mouseDown(document.body)
+    expect(screen.queryByText('用量接近上限')).not.toBeInTheDocument()
+
+    
+    fireEvent.click(screen.getByRole('button', { name: '通知' }))
+    await screen.findByText('用量接近上限')
+    fireEvent.mouseDown(screen.getByText('用量接近上限'))
+    expect(screen.getByText('用量接近上限')).toBeInTheDocument()
+  })
+
+  it('点击带 link 的通知：onNavigate(link) 回调触发且面板关闭', async () => {
+    mockedGet.mockImplementation(dispatchGet(1, [unreadItem]))
+    const onNavigate = vi.fn()
+    renderBell(onNavigate)
+    fireEvent.click(screen.getByRole('button', { name: '通知' }))
+    fireEvent.click(await screen.findByText('用量接近上限'))
+
+    await waitFor(() => expect(onNavigate).toHaveBeenCalledWith('/settings/workspace/usage'))
+    expect(screen.queryByText('用量接近上限')).not.toBeInTheDocument()
+  })
+
+  it('点击无 link 的通知：不触发 onNavigate，仅走已读', async () => {
+    mockedGet.mockImplementation(dispatchGet(1, [{ ...unreadItem, link: null }]))
+    const onNavigate = vi.fn()
+    renderBell(onNavigate)
+    fireEvent.click(screen.getByRole('button', { name: '通知' }))
+    fireEvent.click(await screen.findByText('用量接近上限'))
+
+    await waitFor(() => expect(mockedPost).toHaveBeenCalledWith('/api/notifications/n-1/read'))
+    expect(onNavigate).not.toHaveBeenCalled()
+  })
+})
+
+describe('P2-7/P2-19：加载态与键盘化', () => {
+  it('P2-7：列表加载中显示"加载中…"而非"暂无通知"空态', async () => {
+    let resolveList!: (v: N[]) => void
+    mockedGet.mockImplementation((url: string) => {
+      if (url === '/api/notifications/badge') return Promise.resolve({ count: 0 })
+      if (url.startsWith('/api/notifications?')) return new Promise<N[]>((res) => { resolveList = res })
+      return Promise.reject(new Error(`unexpected GET ${url}`))
+    })
+    renderBell()
+    fireEvent.click(screen.getByRole('button', { name: '通知' }))
+    expect(await screen.findByText('加载中…')).toBeInTheDocument()
+    expect(screen.queryByText('暂无通知')).not.toBeInTheDocument()
+
+    resolveList([])
+    await waitFor(() => expect(screen.getByText('暂无通知')).toBeInTheDocument())
+  })
+
+  it('P2-19：条目为 button（Tab/Enter 可操作）；触发钮带 aria-haspopup/expanded', async () => {
+    mockedGet.mockImplementation(dispatchGet(1, [unreadItem]))
+    renderBell()
+    const trigger = screen.getByRole('button', { name: '通知' })
+    expect(trigger).toHaveAttribute('aria-haspopup', 'true')
+    expect(trigger).toHaveAttribute('aria-expanded', 'false')
+
+    fireEvent.click(trigger)
+    expect(trigger).toHaveAttribute('aria-expanded', 'true')
+    
+    const item = await screen.findByRole('button', { name: /用量接近上限/ })
+    fireEvent.click(item)
+    await waitFor(() => expect(mockedPost).toHaveBeenCalledWith('/api/notifications/n-1/read'))
+  })
+})

package/src/components/__tests__/OnboardingChecklist.test.tsx

@@ -0,0 +1,97 @@
+import '@testing-library/jest-dom/vitest'
+import { fireEvent, render, screen } from '@testing-library/react'
+import { renderToString } from 'react-dom/server'
+import { describe, it, expect, vi, beforeEach } from 'vitest'
+import { OnboardingChecklist } from '../OnboardingChecklist'
+
+vi.mock('../../../../hooks/src/hooks/useAuth', () => ({
+  useAuth: () => ({
+    user: { email: 'alice@example.com', display_name: 'Alice' },
+    workspaces: [{ id: 'ws-1', name: 'Acme' }],
+    loading: false,
+    refresh: vi.fn(),
+    logout: vi.fn(),
+  }),
+}))
+
+describe('OnboardingChecklist', () => {
+  beforeEach(() => localStorage.clear())
+
+  it('按 items 渲染步骤与进度值', () => {
+    render(
+      <OnboardingChecklist
+        items={[
+          { key: 'ws', label: '创建工作区', done: true },
+          { key: 'invite', label: '邀请成员', done: false, onClick: () => {} },
+        ]}
+      />,
+    )
+    expect(screen.getByText('创建工作区')).toBeInTheDocument()
+    expect(screen.getByText('邀请成员')).toBeInTheDocument()
+    expect(screen.getByText('50%')).toBeInTheDocument()
+    expect(screen.getByText('已完成 1/2 项')).toBeInTheDocument()
+  })
+
+  it('点击未完成且带 onClick 的步骤触发回调；已完成项不触发', () => {
+    const onClick = vi.fn()
+    render(
+      <OnboardingChecklist
+        items={[
+          { key: 'ws', label: '创建工作区', done: true },
+          { key: 'invite', label: '邀请成员', done: false, onClick },
+        ]}
+      />,
+    )
+    fireEvent.click(screen.getByRole('button', { name: /邀请成员/ }))
+    expect(onClick).toHaveBeenCalledTimes(1)
+
+    fireEvent.click(screen.getByRole('button', { name: /创建工作区/ }))
+    expect(screen.getByRole('button', { name: /创建工作区/ })).toBeDisabled()
+  })
+
+  it('关闭后写入 localStorage 并不再渲染', () => {
+    const { rerender } = render(<OnboardingChecklist items={[{ key: 'a', label: 'A', done: false }]} />)
+    fireEvent.click(screen.getByRole('button', { name: '关闭引导清单' }))
+    expect(localStorage.getItem('pk_onboarding_dismissed')).toBe('1')
+    rerender(<OnboardingChecklist items={[{ key: 'a', label: 'A', done: false }]} />)
+    expect(screen.queryByText('A')).not.toBeInTheDocument()
+  })
+
+  it('不传 items 时用上下文默认项（已有工作区 → 完成）', () => {
+    render(<OnboardingChecklist />)
+    expect(screen.getByText('创建第一个工作区')).toBeInTheDocument()
+    expect(screen.getByText('100%')).toBeInTheDocument()
+  })
+
+  
+
+  it('localStorage 已有关闭标记：渲染期输出与标记无关（双端一致），mount 后同步为关闭', () => {
+    localStorage.setItem('pk_onboarding_dismissed', '1')
+    
+    const html = renderToString(
+      <OnboardingChecklist items={[{ key: 'a', label: '引导项A', done: false }]} />,
+    )
+    expect(html).toContain('引导项A')
+    
+    render(<OnboardingChecklist items={[{ key: 'a', label: '引导项A', done: false }]} />)
+    expect(screen.queryByText('引导项A')).not.toBeInTheDocument()
+  })
+
+  it('localStorage 不可用：渲染与关闭都不炸（effect/事件期兜底），按未关闭处理', () => {
+    const desc = Object.getOwnPropertyDescriptor(globalThis, 'localStorage')
+    Object.defineProperty(globalThis, 'localStorage', {
+      configurable: true,
+      get() {
+        throw new Error('blocked')
+      },
+    })
+    try {
+      render(<OnboardingChecklist items={[{ key: 'a', label: 'A', done: false }]} />)
+      expect(screen.getByText('A')).toBeInTheDocument()
+      fireEvent.click(screen.getByRole('button', { name: '关闭引导清单' }))
+      expect(screen.queryByText('A')).not.toBeInTheDocument()
+    } finally {
+      Object.defineProperty(globalThis, 'localStorage', desc!)
+    }
+  })
+})

package/src/components/__tests__/SecretModal.test.tsx

@@ -0,0 +1,112 @@
+import '@testing-library/jest-dom/vitest'
+import { fireEvent, render, screen, waitFor } from '@testing-library/react'
+import { describe, it, expect, vi, beforeEach } from 'vitest'
+import { SecretModal } from '../SecretModal'
+
+
+
+
+describe('SecretModal', () => {
+  beforeEach(() => {
+    vi.stubGlobal('navigator', {
+      clipboard: { writeText: vi.fn().mockResolvedValue(undefined) },
+    })
+  })
+
+  function setup(secretValue: string, open: boolean, onOpenChange: (o: boolean) => void) {
+    return render(
+      <SecretModal
+        open={open}
+        onOpenChange={onOpenChange}
+        secretName="个人访问令牌"
+        secretValue={secretValue}
+      />,
+    )
+  }
+
+  it('轮换双密钥：确认第一个后展示第二个，复选框与关闭钮必须回到初始禁用态', async () => {
+    const onOpenChange = vi.fn()
+    const { rerender } = setup('tk_first', true, onOpenChange)
+
+    
+    fireEvent.click(screen.getByRole('checkbox'))
+    expect(screen.getByRole('button', { name: '我已保存' })).toBeEnabled()
+
+    
+    rerender(
+      <SecretModal
+        open
+        onOpenChange={onOpenChange}
+        secretName="个人访问令牌"
+        secretValue="tk_second"
+      />,
+    )
+
+    
+    expect(await screen.findByText('tk_second')).toBeInTheDocument()
+    expect(screen.getByRole('checkbox')).not.toBeChecked()
+    expect(screen.getByRole('button', { name: '请先确认已保存' })).toBeDisabled()
+    expect(screen.queryByText('tk_first')).not.toBeInTheDocument()
+  })
+
+  it('关闭再打开同一组件：confirmed 同样重置', async () => {
+    const onOpenChange = vi.fn()
+    const { rerender } = setup('tk_first', true, onOpenChange)
+
+    fireEvent.click(screen.getByRole('checkbox'))
+    expect(screen.getByRole('button', { name: '我已保存' })).toBeEnabled()
+
+    rerender(
+      <SecretModal
+        open={false}
+        onOpenChange={onOpenChange}
+        secretName="个人访问令牌"
+        secretValue="tk_first"
+      />,
+    )
+    rerender(
+      <SecretModal
+        open
+        onOpenChange={onOpenChange}
+        secretName="个人访问令牌"
+        secretValue="tk_first"
+      />,
+    )
+
+    expect(await screen.findByText('tk_first')).toBeInTheDocument()
+    expect(screen.getByRole('checkbox')).not.toBeChecked()
+    expect(screen.getByRole('button', { name: '请先确认已保存' })).toBeDisabled()
+  })
+
+  it('复制后 copied 状态随 secretValue 变化重置', async () => {
+    const onOpenChange = vi.fn()
+    const { rerender } = setup('tk_first', true, onOpenChange)
+
+    fireEvent.click(screen.getByRole('button', { name: '复制' }))
+    await waitFor(() => expect(screen.getByRole('button', { name: '已复制' })).toBeInTheDocument())
+
+    rerender(
+      <SecretModal
+        open
+        onOpenChange={onOpenChange}
+        secretName="个人访问令牌"
+        secretValue="tk_second"
+      />,
+    )
+    expect(await screen.findByRole('button', { name: '复制' })).toBeInTheDocument()
+  })
+
+  
+  it('复制失败如实提示手动复制，不显示"已复制"', async () => {
+    vi.stubGlobal('navigator', {
+      clipboard: { writeText: vi.fn().mockRejectedValue(new Error('denied')) },
+    })
+    setup('tk_first', true, vi.fn())
+
+    fireEvent.click(screen.getByRole('button', { name: '复制' }))
+    await waitFor(() =>
+      expect(screen.getByRole('button', { name: '复制失败，请手动复制' })).toBeInTheDocument(),
+    )
+    expect(screen.queryByRole('button', { name: '已复制' })).not.toBeInTheDocument()
+  })
+})

package/src/components/__tests__/SiteBannerLayer.test.tsx

@@ -0,0 +1,44 @@
+import '@testing-library/jest-dom/vitest'
+import { render, screen } from '@testing-library/react'
+import { describe, it, expect, vi } from 'vitest'
+import { SiteBannerLayer } from '../SiteBannerLayer'
+
+
+const mocked = vi.hoisted(() => ({ data: undefined as Record<string, string> | undefined }))
+vi.mock('../../../../hooks/src/hooks/useSiteConfig', () => ({
+  useSiteConfig: () => ({ data: mocked.data }),
+}))
+
+describe('SiteBannerLayer', () => {
+  it('无数据（/config 拉取失败/未返回）整层不渲染', () => {
+    mocked.data = undefined
+    const { container } = render(<SiteBannerLayer />)
+    expect(container.firstChild).toBeNull()
+  })
+
+  it('维护模式开 → warn 条 + 公告条（kind 跟随 banner_kind）', () => {
+    mocked.data = {
+      maintenance_mode: '1',
+      banner_text: '今晚 22:00 例行维护',
+      banner_kind: 'info',
+    }
+    render(<SiteBannerLayer />)
+    expect(screen.getByText(/维护中：系统维护中/)).toBeInTheDocument()
+    expect(screen.getByText('今晚 22:00 例行维护')).toBeInTheDocument()
+  })
+
+  it('公告为空 → 只渲染维护条；maintenanceText 可换文案', () => {
+    mocked.data = { maintenance_mode: '1', banner_text: '', banner_kind: 'info' }
+    render(<SiteBannerLayer maintenanceText="MyProduct 维护窗口中" />)
+    expect(screen.getByText('MyProduct 维护窗口中')).toBeInTheDocument()
+    
+    expect(screen.queryAllByRole('status')).toHaveLength(1)
+  })
+
+  it('维护关 + 公告 warn → 单公告条，无维护条', () => {
+    mocked.data = { maintenance_mode: '0', banner_text: '紧急公告', banner_kind: 'warn' }
+    render(<SiteBannerLayer />)
+    expect(screen.getByText('紧急公告')).toBeInTheDocument()
+    expect(screen.queryByText(/维护中/)).not.toBeInTheDocument()
+  })
+})

package/src/components/__tests__/UserMenu.test.tsx

@@ -0,0 +1,98 @@
+import '@testing-library/jest-dom/vitest'
+import { fireEvent, render, screen, waitFor } from '@testing-library/react'
+import { MemoryRouter, Routes, Route } from 'react-router'
+import { describe, it, expect, vi, beforeEach } from 'vitest'
+import { UserMenu } from '../UserMenu'
+import { Toaster } from '../toast'
+import type { PKUser } from '../../provider/PloykitProvider'
+
+
+const logout = vi.fn(async () => {})
+const mocks = vi.hoisted(() => ({ isPlatformAdmin: false }))
+vi.mock('../../../../hooks/src/hooks/useAuth', () => ({
+  useAuth: () => ({
+    user: {
+      email: 'alice@example.com',
+      display_name: 'Alice',
+      avatar_url: undefined,
+      is_platform_admin: mocks.isPlatformAdmin,
+    } as PKUser,
+    workspaces: [],
+    loading: false,
+    refresh: vi.fn(),
+    logout,
+  }),
+}))
+
+function renderAt(path: string) {
+  return render(
+    <MemoryRouter initialEntries={[path]}>
+      {/* Toaster 在路由外常驻：logout 失败兜底用例断言 toast 文案实际可见
+          （若随切换路由卸载，跳转后 toast 不再渲染） */}
+      <Toaster />
+      <Routes>
+        <Route path="/login" element={<div>login-page</div>} />
+        <Route path="/account/profile" element={<div>account-profile-page</div>} />
+        <Route path="/account/notifications" element={<div>notifications-page</div>} />
+        <Route path="/admin" element={<div>admin-page</div>} />
+        <Route path="*" element={<UserMenu />} />
+      </Routes>
+    </MemoryRouter>,
+  )
+}
+
+describe('UserMenu', () => {
+  beforeEach(() => {
+    logout.mockClear()
+    mocks.isPlatformAdmin = false
+  })
+
+  it('渲染头像 initials fallback', () => {
+    renderAt('/app')
+    expect(screen.getByRole('button', { name: '用户菜单' })).toHaveTextContent('A')
+  })
+
+  it('默认项：账户设置（→/account/profile）；普通用户无管理控制台', async () => {
+    renderAt('/app')
+    fireEvent.click(screen.getByRole('button', { name: '用户菜单' }))
+    expect(await screen.findByRole('menu')).toBeInTheDocument()
+    expect(screen.getByRole('menuitem', { name: '账户设置' })).toBeVisible()
+    expect(screen.queryByRole('menuitem', { name: '管理控制台' })).not.toBeInTheDocument()
+    expect(screen.queryByRole('menuitem', { name: '个人设置' })).not.toBeInTheDocument()
+    expect(screen.queryByRole('menuitem', { name: '通知偏好' })).not.toBeInTheDocument()
+    expect(screen.getByRole('menuitem', { name: '退出登录' })).toBeVisible()
+  })
+
+  it('账户设置点击跳 /account/profile', async () => {
+    renderAt('/app')
+    fireEvent.click(screen.getByRole('button', { name: '用户菜单' }))
+    fireEvent.click(await screen.findByRole('menuitem', { name: '账户设置' }))
+    await waitFor(() => expect(screen.getByText('account-profile-page')).toBeInTheDocument())
+  })
+
+  it('is_platform_admin：出现管理控制台（→/admin）', async () => {
+    mocks.isPlatformAdmin = true
+    renderAt('/app')
+    fireEvent.click(screen.getByRole('button', { name: '用户菜单' }))
+    fireEvent.click(await screen.findByRole('menuitem', { name: '管理控制台' }))
+    await waitFor(() => expect(screen.getByText('admin-page')).toBeInTheDocument())
+  })
+
+  it('退出登录调用 logout 并跳 /login', async () => {
+    renderAt('/app')
+    fireEvent.click(screen.getByRole('button', { name: '用户菜单' }))
+    fireEvent.click(await screen.findByRole('menuitem', { name: '退出登录' }))
+    await waitFor(() => expect(logout).toHaveBeenCalledTimes(1))
+    await waitFor(() => expect(screen.getByText('login-page')).toBeInTheDocument())
+  })
+
+  it('P3-11：logout 抛错不困死用户——toast 提示失败仍照常跳转登录页', async () => {
+    logout.mockRejectedValueOnce(new Error('network down'))
+    renderAt('/app')
+    fireEvent.click(screen.getByRole('button', { name: '用户菜单' }))
+    fireEvent.click(await screen.findByRole('menuitem', { name: '退出登录' }))
+    
+    expect(await screen.findByText('退出登录失败，请稍后在登录页重试')).toBeInTheDocument()
+    await waitFor(() => expect(screen.getByText('login-page')).toBeInTheDocument())
+  })
+})