fix(emotion): support ReactTestUtils in decorators that use forwardRef

instructure · Oct 18, 2023 · c7389cb · c7389cb
1 parent 4aed81e
commit c7389cb
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/packages/emotion/src/withStyle.tsx b/packages/emotion/src/withStyle.tsx
@@ -173,6 +173,7 @@ const withStyle = decorator(
     > & {
       generateComponentTheme?: GenerateComponentTheme
       allowedProps?: string[]
+      originalType?: WithStyleComponent
     } = forwardRef((props, ref) => {
       const theme = useTheme()
 
@@ -243,6 +244,10 @@ const withStyle = decorator(
 
     hoistNonReactStatics(WithStyle, ComposedComponent)
 
+    // workaround so this can be tested with ReactTestUtils
+    // more info: https://github.com/facebook/react/issues/13455
+    WithStyle.originalType = ComposedComponent
+
     // we have to pass these on, because sometimes users
     // access propTypes of the component in other components
     // eslint-disable-next-line react/forbid-foreign-prop-types