portainer/app/react/test-utils/withTestQuery.tsx

import { ComponentType } from 'react';
import { QueryClient } from '@tanstack/react-query';

import { withReactQuery } from '@/react-tools/withReactQuery';

export function withTestQueryProvider<T>(
  WrappedComponent: ComponentType<T & JSX.IntrinsicAttributes>
) {
  const testQueryClient = new QueryClient({
    defaultOptions: { queries: { retry: false } },
  });

  return withReactQuery(WrappedComponent, testQueryClient);
}