From 2bff8cc733c541deaaa13a454239bbf921ea5f4b Mon Sep 17 00:00:00 2001
From: Sebastian Malton <sebastian@malton.name>
Date: Fri, 24 Feb 2023 15:42:40 -0500
Subject: [PATCH] Cleanup renderFor impl to be simpler

Signed-off-by: Sebastian Malton <sebastian@malton.name>
---
 .../components/test-utils/renderFor.tsx       | 21 ++++++++-----------
 1 file changed, 9 insertions(+), 12 deletions(-)

diff --git a/packages/core/src/renderer/components/test-utils/renderFor.tsx b/packages/core/src/renderer/components/test-utils/renderFor.tsx
index fc42ae0e09..594cd66a64 100644
--- a/packages/core/src/renderer/components/test-utils/renderFor.tsx
+++ b/packages/core/src/renderer/components/test-utils/renderFor.tsx
@@ -4,7 +4,7 @@
  */
 import React from "react";
 import type { RenderResult } from "@testing-library/react";
-import { render as testingLibraryRender } from "@testing-library/react";
+import { render } from "@testing-library/react";
 import type { DiContainer } from "@ogre-tools/injectable";
 import { DiContextProvider } from "@ogre-tools/injectable-react";
 
@@ -12,20 +12,17 @@ export type DiRender = (ui: React.ReactElement) => RenderResult;
 
 type DiRenderFor = (di: DiContainer) => DiRender;
 
+const getDOM = (di: DiContainer, ui: React.ReactElement) => (
+  <DiContextProvider value={{ di }}>
+    {ui}
+  </DiContextProvider>
+);
+
 export const renderFor: DiRenderFor = (di) => (ui) => {
-  const result = testingLibraryRender(
-    (
-      <DiContextProvider value={{ di }}>
-        {ui}
-      </DiContextProvider>
-    ),
-  );
+  const result = render(getDOM(di, ui));
 
   return {
     ...result,
-
-    rerender: (ui: React.ReactElement) => result.rerender(
-      <DiContextProvider value={{ di }}>{ui}</DiContextProvider>,
-    ),
+    rerender: (ui) => result.rerender(getDOM(di, ui)),
   };
 };