use createRoot for snapshots when available

robinweser · Apr 6, 2022 · 34c3537 · 34c3537 · vercel · Apr 6, 2022
1 parent 9aeea63
commit 34c3537
Show file tree

Hide file tree

Showing 4 changed files with 165 additions and 31 deletions.
diff --git a/packages/fela-enforce-longhands/src/__tests__/__snapshots__/index-test.js.snap b/packages/fela-enforce-longhands/src/__tests__/__snapshots__/index-test.js.snap
diff --git a/packages/fela-integration/package.json b/packages/fela-integration/package.json
@@ -17,6 +17,7 @@
     "fela-monolithic": "^12.0.2",
     "fela-tools": "^12.0.2",
     "jest-react-fela": "^12.0.2",
+    "jsdom": "^19.0.0",
     "raf": "^3.4.1",
     "react": "^17.0.2",
     "react-dom": "^17.0.2",

diff --git a/packages/jest-react-fela/src/createSnapshot.js b/packages/jest-react-fela/src/createSnapshot.js
@@ -5,9 +5,29 @@ import { RendererProvider, ThemeProvider } from 'react-fela'
 
 import { createSnapshotFactory } from 'jest-fela-bindings'
 
+let createRoot
+try {
+  const ReactDOMClient = require('react-dom/client')
+  if (
+    typeof ReactDOMClient !== 'undefined' &&
+    typeof ReactDOMClient.createRoot !== 'undefined'
+  ) {
+    createRoot = ReactDOMClient.createRoot
+  }
+} catch (e) {}
+
+function renderComponent(component, node) {
+  if (createRoot) {
+    const root = ReactDOMClient.createRoot(component)
+    root.render(node)
+  } else {
+    render(component, node)
+  }
+}
+
 export default createSnapshotFactory(
   createElement,
-  render,
+  renderComponent,
   createRenderer(),
   RendererProvider,
   ThemeProvider