setup-react-query-tests.tsx

import { QueryClient, QueryClientProvider } from "@tanstack/react-query";
import { render } from "@testing-library/react-native";
import { ReactElement, ReactNode } from "react";

import { AppContexts } from "../../App";

const createTestQueryClient = () =>
  new QueryClient({
    defaultOptions: {
      queries: {
        retry: false,
        gcTime: 0,
      },
      mutations: {
        retry: false,
        gcTime: 0,
      },
    },
  });

export function createWrapper() {
  const testQueryClient = createTestQueryClient();
  return function QueryClientWrapper(props: { children: ReactNode }) {
    return (
      <QueryClientProvider client={testQueryClient}>
        {props.children}
      </QueryClientProvider>
    );
  };
}

export const renderWithContexts: typeof render = (
  ui: ReactElement,
  options,
) => {
  const testQueryClient = createTestQueryClient();
  const { rerender, ...result } = render(
    <AppContexts reactQueryClient={testQueryClient}>{ui}</AppContexts>,
    options,
  );

  return {
    ...result,
    rerender: (rerenderUi: ReactElement) =>
      rerender(
        <QueryClientProvider client={testQueryClient}>
          {rerenderUi}
        </QueryClientProvider>,
      ),
  };
};