Genarrative/apps/ai-game-creator-shell/tests/ErrorReportNotice.test.tsx

/** @vitest-environment jsdom */

import { cleanup, render, screen, waitFor } from '@testing-library/react';
import { afterEach, describe, expect, it, vi } from 'vitest';

const mockState = vi.hoisted(() => ({
  getPending: vi.fn(),
  listeners: [] as Array<() => void>,
}));

vi.mock('../src/services/errorReporting', () => ({
  getPendingClientErrorEvents: mockState.getPending,
  subscribeClientErrorEvents: (listener: () => void) => {
    mockState.listeners.push(listener);
    return () => {
      mockState.listeners = mockState.listeners.filter(
        (item) => item !== listener,
      );
    };
  },
}));

vi.mock('../src/components/error-report/ErrorReportDialog', () => ({
  ErrorReportDialog: () => null,
}));

import { ErrorReportNotice } from '../src/components/error-report/ErrorReportNotice';

const event = {
  eventId: 'rust-error-1',
  fingerprint: 'fingerprint-1',
  source: 'test',
  message: '错过 emit 的错误',
  occurredAt: '1',
  count: 1,
};

describe('ErrorReportNotice', () => {
  afterEach(() => {
    cleanup();
    mockState.getPending.mockReset();
    mockState.listeners = [];
  });

  it('挂载时查询快照，即使错过唤醒事件也能显示通知', async () => {
    mockState.getPending.mockResolvedValue([event]);

    render(<ErrorReportNotice />);

    expect(await screen.findByText('发现 1 个问题')).toBeTruthy();
    expect(mockState.getPending).toHaveBeenCalledTimes(1);
  });

  it('收到唤醒后重新查询权威快照', async () => {
    mockState.getPending
      .mockResolvedValueOnce([])
      .mockResolvedValueOnce([event]);

    render(<ErrorReportNotice />);
    await waitFor(() => expect(mockState.getPending).toHaveBeenCalledTimes(1));

    mockState.listeners[0]?.();

    expect(await screen.findByText('发现 1 个问题')).toBeTruthy();
    expect(mockState.getPending).toHaveBeenCalledTimes(2);
  });
});