glchat/web/tests/queryRetry.test.ts

import { QueryClient } from '@tanstack/react-query';
import { describe, expect, it, vi } from 'vitest';

import { ApiError } from '@/api/client';
import {
  getQueryClient,
  queryRetryDelay,
  setQueryClient,
  shouldRetryQuery,
} from '@/lib/queryClient';

/** Ошибка лимита частоты в том виде, в каком её отдаёт `request()`. */
function rateLimited(retryAfterMs: number): ApiError {
  const error = new ApiError(429, 'rate_limited', 'too many requests');
  Object.defineProperty(error, 'apiError', {
    value: { code: 'rate_limited', message: 'too many requests', retry_after_ms: retryAfterMs },
    enumerable: false,
  });
  return error;
}

describe('повторы запросов при лимите частоты', () => {
  it('повторяет 429 с паузой из ответа сервера', () => {
    const error = rateLimited(1_200);
    expect(shouldRetryQuery(0, error)).toBe(true);
    expect(shouldRetryQuery(1, error)).toBe(true);
    expect(queryRetryDelay(0, error)).toBe(1_450);
  });

  it('после двух повторов сдаётся: ошибку видит пользователь', () => {
    expect(shouldRetryQuery(2, rateLimited(500))).toBe(false);
  });

  it('не ждёт паузу длиннее предела — сразу показывает ошибку', () => {
    // Минутный лимит загрузок: ждать внутри запроса бессмысленно.
    expect(shouldRetryQuery(0, rateLimited(60_000))).toBe(false);
    expect(queryRetryDelay(0, rateLimited(60_000))).toBe(5_000);
  });

  it('429 без retry_after_ms повторяет с обычной выдержкой', () => {
    const error = new ApiError(429, 'rate_limited', 'too many requests');
    expect(shouldRetryQuery(0, error)).toBe(true);
    expect(queryRetryDelay(0, error)).toBe(1_000);
    expect(queryRetryDelay(1, error)).toBe(2_000);
  });

  it('не повторяет отказы, которые не изменятся: 403, 404, 422', () => {
    for (const status of [400, 403, 404, 422]) {
      const error = new ApiError(status, 'validation.failed', 'no');
      expect(shouldRetryQuery(0, error)).toBe(false);
    }
  });

  it('повторяет сетевые сбои и 5xx по одному разу', () => {
    expect(shouldRetryQuery(0, new TypeError('Failed to fetch'))).toBe(true);
    expect(shouldRetryQuery(1, new TypeError('Failed to fetch'))).toBe(false);
    expect(shouldRetryQuery(0, new ApiError(503, 'internal.error', 'later'))).toBe(true);
    expect(shouldRetryQuery(1, new ApiError(503, 'internal.error', 'later'))).toBe(false);
  });

  it('не повторяет отменённый запрос', () => {
    expect(shouldRetryQuery(0, new DOMException('aborted', 'AbortError'))).toBe(false);
  });

  it('общий клиент приложения использует эту политику', () => {
    const options = getQueryClient().getDefaultOptions().queries;
    expect(options?.retry).toBe(shouldRetryQuery);
    expect(options?.retryDelay).toBe(queryRetryDelay);
  });

  it('запрос действительно переживает 429 и отдаёт данные', async () => {
    const queryClient = new QueryClient({
      defaultOptions: {
        queries: { retry: shouldRetryQuery, retryDelay: () => 1 },
      },
    });
    setQueryClient(queryClient);
    const loader = vi
      .fn<() => Promise<string>>()
      .mockRejectedValueOnce(rateLimited(1))
      .mockRejectedValueOnce(rateLimited(1))
      .mockResolvedValue('участники');

    await expect(
      queryClient.fetchQuery({ queryKey: ['members-ok'], queryFn: loader }),
    ).resolves.toBe('участники');
    expect(loader).toHaveBeenCalledTimes(3);

    // Третьего повтора нет: следующий 429 доходит до вызывающего кода.
    const failing = vi.fn<() => Promise<string>>().mockRejectedValue(rateLimited(1));
    await expect(
      queryClient.fetchQuery({ queryKey: ['members-fail'], queryFn: failing }),
    ).rejects.toThrow('too many requests');
    expect(failing).toHaveBeenCalledTimes(3);
  });
});