Genarrative/packages/shared/src/components/PlatformProfileWalletLedgerModal/index.test.tsx

/* @vitest-environment jsdom */

import { render, screen, within } from '@testing-library/react';
import userEvent from '@testing-library/user-event';
import { describe, expect, test, vi } from 'vitest';

import type { ProfileWalletLedgerEntry } from '../../contracts/runtime';
import { PlatformProfileWalletLedgerModal } from './index';
import {
  buildWalletLedgerPresentation,
  formatWalletLedgerAmount,
  formatWalletLedgerDate,
  getWalletLedgerSourceLabel,
} from './model';

function buildLedgerEntry(
  overrides: Partial<ProfileWalletLedgerEntry> = {},
): ProfileWalletLedgerEntry {
  return {
    id: 'ledger-1',
    amountDelta: 12,
    balanceAfter: 88,
    sourceType: 'daily_task_reward',
    createdAt: '2026-06-10T08:00:00.000Z',
    ...overrides,
  };
}

describe('PlatformProfileWalletLedgerModal', () => {
  test('renders ledger entries with the latest balance and UTC date', () => {
    render(
      <PlatformProfileWalletLedgerModal
        ledger={{ entries: [buildLedgerEntry()] }}
        fallbackBalance={40}
        isLoading={false}
        error={null}
        onClose={vi.fn()}
        onRetry={vi.fn()}
      />,
    );

    const dialog = screen.getByRole('dialog', { name: '泥点账单' });
    expect(within(dialog).getByText('88泥点')).toBeTruthy();
    expect(within(dialog).getByText('每日任务奖励')).toBeTruthy();
    expect(within(dialog).getByText('2026-06-10')).toBeTruthy();
    expect(within(dialog).getByText('+12')).toBeTruthy();
    expect(within(dialog).getByText('余额 88')).toBeTruthy();
  });

  test('supports close, loading, empty, and error retry states', async () => {
    const user = userEvent.setup();
    const onClose = vi.fn();
    const onRetry = vi.fn();
    const { rerender } = render(
      <PlatformProfileWalletLedgerModal
        ledger={null}
        fallbackBalance={40}
        isLoading
        error={null}
        onClose={onClose}
        onRetry={onRetry}
      />,
    );

    expect(screen.getByRole('status', { name: '泥点账单加载中' })).toBeTruthy();
    await user.click(screen.getByRole('button', { name: '关闭泥点账单' }));
    expect(onClose).toHaveBeenCalledTimes(1);

    rerender(
      <PlatformProfileWalletLedgerModal
        ledger={{ entries: [] }}
        fallbackBalance={40}
        isLoading={false}
        error={null}
        onClose={onClose}
        onRetry={onRetry}
      />,
    );
    expect(screen.getByText('暂无账单记录')).toBeTruthy();
    expect(screen.getByText('40泥点')).toBeTruthy();

    rerender(
      <PlatformProfileWalletLedgerModal
        ledger={null}
        fallbackBalance={40}
        isLoading={false}
        error="账单加载失败"
        onClose={onClose}
        onRetry={onRetry}
      />,
    );
    await user.click(screen.getByRole('button', { name: '重新加载' }));
    expect(onRetry).toHaveBeenCalledTimes(1);
  });
});

test('builds wallet ledger presentation with stable source fallbacks', () => {
  expect(formatWalletLedgerAmount(-1)).toBe('-1');
  expect(formatWalletLedgerAmount(0)).toBe('0');
  expect(formatWalletLedgerAmount(30)).toBe('+30');
  expect(getWalletLedgerSourceLabel('asset_operation_consume')).toBe(
    '资产操作消耗',
  );
  expect(getWalletLedgerSourceLabel('future_source')).toBe('future_source');
  expect(getWalletLedgerSourceLabel('llm_router_consume')).toBe('LLM 调用消耗');
  expect(getWalletLedgerSourceLabel('')).toBe('未知来源');
  expect(formatWalletLedgerDate('not-a-date')).toBe('not-a-date');

  expect(
    buildWalletLedgerPresentation(
      {
        entries: [
          buildLedgerEntry({
            amountDelta: -1,
            sourceType: 'asset_operation_consume',
          }),
        ],
      },
      12,
    ),
  ).toMatchObject({
    balance: 88,
    balanceLabel: '88泥点',
    entries: [
      {
        amountLabel: '-1',
        balanceLabel: '余额 88',
        createdAtLabel: '2026-06-10',
        isIncome: false,
        sourceLabel: '资产操作消耗',
      },
    ],
  });

  for (const reason of [
    '生成美术素材',
    '编辑美术素材',
    '提取美术素材',
    '生成角色动画',
    '生成视频素材',
    '生成音效素材',
    '生成背景音乐',
  ]) {
    expect(
      buildWalletLedgerPresentation(
        {
          entries: [
            buildLedgerEntry({
              amountDelta: -12,
              sourceType: 'asset_operation_consume',
              reason,
            }),
          ],
        },
        12,
      ).entries[0]?.sourceLabel,
      `unexpected source label for ${reason}`,
    ).toBe(reason);
  }

  expect(
    buildWalletLedgerPresentation(
      {
        entries: [
          buildLedgerEntry({
            amountDelta: -12,
            sourceType: 'asset_operation_consume',
            reason: '   ',
          }),
        ],
      },
      12,
    ).entries[0]?.sourceLabel,
  ).toBe('资产操作消耗');
});