lingxi/apps/web/src/api/client.spec.ts

import { beforeEach, describe, expect, it, vi } from 'vitest';

import { useAuthStore } from '@/stores/auth-store';

import { ApiError, request, setUnauthorizedHandler } from './client';

const jsonResponse = (body: unknown, status = 200): Response =>
  new Response(JSON.stringify(body), {
    status,
    headers: { 'Content-Type': 'application/json' },
  });

const emptyResponse = (status: number): Response => new Response(null, { status });

function signIn(accessToken = 'token-1', refreshToken = 'refresh-1'): void {
  useAuthStore.getState().setSession({
    accessToken,
    refreshToken,
    user: { id: '01J0TEST000000000000000000', name: '测试用户', account: 'test@lingxi.local' },
    permissions: [],
  });
}

function authorizationOf(init: RequestInit | undefined): string | null {
  return init?.headers instanceof Headers ? init.headers.get('Authorization') : null;
}

function toUrl(input: string | URL | Request | undefined): string {
  if (typeof input === 'string') {
    return input;
  }
  if (input instanceof URL) {
    return input.href;
  }
  return input?.url ?? '';
}

describe('请求层', () => {
  beforeEach(() => {
    vi.restoreAllMocks();
    window.localStorage.clear();
    useAuthStore.getState().clearSession();
    setUnauthorizedHandler(null);
  });

  it('注入 Bearer token 并解析 JSON', async () => {
    signIn();
    const fetchMock = vi
      .spyOn(globalThis, 'fetch')
      .mockResolvedValue(jsonResponse({ status: 'ok' }));

    await expect(request<{ status: string }>('/health')).resolves.toEqual({ status: 'ok' });
    const init = fetchMock.mock.calls[0]?.[1];
    expect(authorizationOf(init)).toBe('Bearer token-1');
    expect(toUrl(fetchMock.mock.calls[0]?.[0])).toBe('http://127.0.0.1:3000/api/v1/health');
  });

  it('auth: false 时不注入 token', async () => {
    signIn();
    const fetchMock = vi.spyOn(globalThis, 'fetch').mockResolvedValue(jsonResponse({ ok: true }));
    await request('/auth/login', {
      method: 'POST',
      auth: false,
      body: { email: 'a', password: 'b' },
    });
    expect(authorizationOf(fetchMock.mock.calls[0]?.[1])).toBeNull();
  });

  it('204 返回 undefined', async () => {
    signIn();
    vi.spyOn(globalThis, 'fetch').mockResolvedValue(emptyResponse(204));
    await expect(request('/resource')).resolves.toBeUndefined();
  });

  it('统一错误体解析为 ApiError，保留 trace_id', async () => {
    signIn();
    vi.spyOn(globalThis, 'fetch').mockResolvedValue(
      jsonResponse(
        { code: 42201, message: '业务校验失败', details: { field: 'name' }, trace_id: 'trace-1' },
        422,
      ),
    );
    const error = await request('/resource').catch((caught: unknown) => caught);
    expect(error).toBeInstanceOf(ApiError);
    const apiError = error instanceof ApiError ? error : null;
    expect(apiError?.code).toBe(42201);
    expect(apiError?.status).toBe(422);
    expect(apiError?.traceId).toBe('trace-1');
    expect(apiError?.details).toEqual({ field: 'name' });
  });

  it('401 时单飞刷新并重放原请求', async () => {
    signIn('expired-token', 'refresh-1');
    let refreshCalls = 0;
    vi.spyOn(globalThis, 'fetch').mockImplementation((input, init) => {
      const url = toUrl(input);
      const authorization = authorizationOf(init);
      if (url.endsWith('/auth/refresh')) {
        refreshCalls += 1;
        return Promise.resolve(
          jsonResponse({ accessToken: 'new-token', refreshToken: 'refresh-2' }),
        );
      }
      if (authorization === 'Bearer expired-token') {
        return Promise.resolve(emptyResponse(401));
      }
      return Promise.resolve(jsonResponse({ ok: true }));
    });

    const [first, second] = await Promise.all([request('/a'), request('/b')]);
    expect(first).toEqual({ ok: true });
    expect(second).toEqual({ ok: true });
    expect(refreshCalls).toBe(1);
    expect(useAuthStore.getState().accessToken).toBe('new-token');
  });

  it('刷新失败时清空会话并触发未授权回调', async () => {
    signIn('expired-token', 'refresh-1');
    const onUnauthorized = vi.fn();
    setUnauthorizedHandler(onUnauthorized);
    vi.spyOn(globalThis, 'fetch').mockImplementation((input) => {
      const url = toUrl(input);
      return Promise.resolve(
        url.endsWith('/auth/refresh') ? emptyResponse(401) : emptyResponse(401),
      );
    });

    await expect(request('/secure')).rejects.toBeInstanceOf(ApiError);
    expect(onUnauthorized).toHaveBeenCalledTimes(1);
    expect(useAuthStore.getState().accessToken).toBeNull();
  });

  it('把请求体序列化为 JSON', async () => {
    signIn();
    const fetchMock = vi.spyOn(globalThis, 'fetch').mockResolvedValue(jsonResponse({ ok: true }));
    await request('/resource', { method: 'POST', body: { name: '素材' } });
    const init = fetchMock.mock.calls[0]?.[1];
    expect(init?.body).toBe(JSON.stringify({ name: '素材' }));
    expect(init?.method).toBe('POST');
  });
});