lens/packages/utility-features/test-utils/src/render-for.tsx

import React from "react";
import type { RenderResult } from "@testing-library/react";
import { render as testingLibraryRender } from "@testing-library/react";
import type { DiContainer } from "@ogre-tools/injectable";
import { DiContextProvider } from "@ogre-tools/injectable-react";

export type DiRender = (ui: React.ReactElement) => RenderResult;

type DiRenderFor = (di: DiContainer) => DiRender;

export const renderFor: DiRenderFor = (di) => (ui) => {
  const result = testingLibraryRender(
    <DiContextProvider value={{ di }}>{ui}</DiContextProvider>
  );

  return {
    ...result,

    rerender: (ui: React.ReactElement) =>
      result.rerender(
        <DiContextProvider value={{ di }}>{ui}</DiContextProvider>
      ),
  };
};