feast/ui/src/test-utils.tsx at get-historical-python-mode · feast-dev/feast · GitHub

37 lines (31 loc) · 1.05 KB

import React from "react";

import { render, RenderOptions } from "@testing-library/react";

import { QueryClient, QueryClientProvider } from "react-query";

import { QueryParamProvider } from "use-query-params";

import { MemoryRouter as Router } from "react-router-dom";

import RouteAdapter from "./hacks/RouteAdapter";

interface ProvidersProps {

children: React.ReactNode;

}

const queryClient = new QueryClient();

const AllTheProviders = ({ children }: ProvidersProps) => {

return (

<QueryParamProvider

ReactRouterRoute={RouteAdapter as unknown as React.FunctionComponent}

{children}

</QueryParamProvider>

</Router>

</QueryClientProvider>

);

};

const customRender = (

ui: React.ReactElement,

options?: Omit<RenderOptions, 'wrapper'>

) => render(ui, { wrapper: AllTheProviders, ...options });

// re-export everything

export * from "@testing-library/react";

// override render method

export { customRender as render };

Expand file tree