testing function expression components

intelie · Aug 26, 2021 · 896f8f1 · 896f8f1
1 parent 3540cc9
commit 896f8f1
Show file tree

Hide file tree

Showing 2 changed files with 37 additions and 0 deletions.
diff --git a/__tests__/__snapshots__/react-display-name.js.snap b/__tests__/__snapshots__/react-display-name.js.snap
@@ -58,3 +58,15 @@ function factory(React) {
   };
 }"
 `;
+
+exports[`react-display-name test suite Transforms function expression components 1`] = `
+"\\"use strict\\";
+
+function factory(React) {
+  var MyFunctionComponent = function MyFunctionComponent() {
+    return /*#__PURE__*/React.createElement(\\"div\\", null, \\"1\\");
+  };
+
+  return MyFunctionComponent;
+}"
+`;
diff --git a/__tests__/react-display-name.js b/__tests__/react-display-name.js
@@ -55,4 +55,29 @@ return factory;
 
         expect(element.type.name).toBe('MyFunctionComponent')
     })
+
+    it('Transforms function expression components', async () => {
+        const classComponentInput = `
+            function factory (React) {
+                const MyFunctionComponent = () => {
+                    return <div>1</div>
+                }
+
+                return MyFunctionComponent
+            }
+    `
+
+        const { code: classComponentOutput } = await transformAsync(classComponentInput, options)
+
+        expect(classComponentOutput).toMatchSnapshot()
+
+        const MyComponentFactory = new Function(`
+        ${classComponentOutput}
+return factory;
+`)()
+        const MyComponent = MyComponentFactory(React)
+        const element = <MyComponent />
+
+        expect(element.type.name).toBe('MyFunctionComponent')
+    })
 })