From 55e79c290d3ec8a8eb3d39539e2c05bf35dff3d9 Mon Sep 17 00:00:00 2001
From: Jeff Way <jeff.way@me.com>
Date: Thu, 1 Feb 2024 11:49:10 -0800
Subject: [PATCH] fix: Update types to support all possible react component
 return values (#1272)

* Update types to support all possible react component return values

* Update type test types
---
 types/index.d.ts | 8 ++++----
 types/test.tsx   | 8 ++++----
 2 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/types/index.d.ts b/types/index.d.ts
index 1f1135c5..5db1d201 100644
--- a/types/index.d.ts
+++ b/types/index.d.ts
@@ -39,7 +39,7 @@ export type RenderResult<
     maxLength?: number,
     options?: prettyFormat.OptionsReceived,
   ) => void
-  rerender: (ui: React.ReactElement) => void
+  rerender: (ui: React.ReactNode) => void
   unmount: () => void
   asFragment: () => DocumentFragment
 } & {[P in keyof Q]: BoundFunction<Q[P]>}
@@ -90,7 +90,7 @@ export interface RenderOptions<
    *
    *  @see https://testing-library.com/docs/react-testing-library/api/#wrapper
    */
-  wrapper?: React.JSXElementConstructor<{children: React.ReactElement}>
+  wrapper?: React.JSXElementConstructor<{children: React.ReactNode}>
 }
 
 type Omit<T, K extends keyof T> = Pick<T, Exclude<keyof T, K>>
@@ -103,11 +103,11 @@ export function render<
   Container extends Element | DocumentFragment = HTMLElement,
   BaseElement extends Element | DocumentFragment = Container,
 >(
-  ui: React.ReactElement,
+  ui: React.ReactNode,
   options: RenderOptions<Q, Container, BaseElement>,
 ): RenderResult<Q, Container, BaseElement>
 export function render(
-  ui: React.ReactElement,
+  ui: React.ReactNode,
   options?: Omit<RenderOptions, 'queries'>,
 ): RenderResult
 
diff --git a/types/test.tsx b/types/test.tsx
index 3486a9a8..6ff899de 100644
--- a/types/test.tsx
+++ b/types/test.tsx
@@ -123,10 +123,10 @@ export function testQueries() {
 }
 
 export function wrappedRender(
-  ui: React.ReactElement,
+  ui: React.ReactNode,
   options?: pure.RenderOptions,
 ) {
-  const Wrapper = ({children}: {children: React.ReactElement}): JSX.Element => {
+  const Wrapper = ({children}: {children: React.ReactNode}): JSX.Element => {
     return <div>{children}</div>
   }
 
@@ -134,7 +134,7 @@ export function wrappedRender(
 }
 
 export function wrappedRenderB(
-  ui: React.ReactElement,
+  ui: React.ReactNode,
   options?: pure.RenderOptions,
 ) {
   const Wrapper: React.FunctionComponent<{children?: React.ReactNode}> = ({
@@ -147,7 +147,7 @@ export function wrappedRenderB(
 }
 
 export function wrappedRenderC(
-  ui: React.ReactElement,
+  ui: React.ReactNode,
   options?: pure.RenderOptions,
 ) {
   interface AppWrapperProps {