deepseek-harness/packages/client/ui-openrouter-usage/tests/balance-badge.client.spec.tsx

// @vitest-environment jsdom
// BalanceBadge presentation: polls an injected snapshot callback and renders
// 余额 in both widths, hiding unknown figures behind the empty placeholder.

import { act, cleanup, render, screen } from '@testing-library/react'
import { afterEach, describe, expect, it, vi } from 'vitest'
import { makeTranslate } from '@deepseek-ai/dsh-client-test-runtime'
import { zh as commonZh } from '@deepseek-ai/dsh-client-locale/src/locales/zh.ts'
import type { OpenRouterBalance } from '@deepseek-ai/dsh-openrouter-usage/client'
import { BALANCE_POLL_MS, BalanceBadge, type BalanceBadgeProps } from '../src/client/BalanceBadge.tsx'
import { zh } from '../src/client/locales.ts'

afterEach(() => {
  vi.useRealTimers()
  cleanup()
})

const t = makeTranslate(zh, commonZh)

function makeBalance(over: Partial<OpenRouterBalance> = {}): OpenRouterBalance {
  return {
    balanceUsd: 10.5,
    label: 'prod-key',
    usageTokens: 1000,
    limitTokens: 100000,
    isFreeTier: false,
    updatedAt: Date.now(),
    currency: 'USD',
    ...over,
  }
}

function makeProps(over: Partial<BalanceBadgeProps> = {}) {
  return {
    wide: true,
    snapshot: vi.fn(async () => makeBalance()),
    t,
    ...over,
  } as unknown as Parameters<typeof BalanceBadge>[0]
}

describe('BalanceBadge', () => {
  it('renders the balance label with the USD figure in wide mode', async () => {
    render(<BalanceBadge {...makeProps()} />)
    expect(await screen.findByLabelText('余额 $10.50')).toBeTruthy()
    expect(screen.getByText('余额 $10.50')).toBeTruthy()
  })

  it('polls the snapshot callback on the balance interval', async () => {
    vi.useFakeTimers()
    const snapshot = vi.fn(async () => makeBalance())
    render(<BalanceBadge {...makeProps({ snapshot })} />)
    await act(async () => { await Promise.resolve() })
    expect(snapshot).toHaveBeenCalledTimes(1)
    await act(async () => { vi.advanceTimersByTime(BALANCE_POLL_MS) })
    await act(async () => { await Promise.resolve() })
    expect(snapshot).toHaveBeenCalledTimes(2)
  })

  it('renders only the rail marker in the collapsed sidebar', async () => {
    render(<BalanceBadge {...makeProps({ wide: false })} />)
    expect(await screen.findByLabelText('余额 $10.50')).toBeTruthy()
    expect(screen.queryByText('余额 $10.50')).toBeNull()
  })

  it('shows the empty placeholder when no balance is known yet', async () => {
    render(<BalanceBadge {...makeProps({ snapshot: vi.fn(async () => makeBalance({ balanceUsd: null })) })} />)
    expect(await screen.findByLabelText('余额 ——')).toBeTruthy()
  })
})