fix: Component also need support ref check

Author: zombieJ

src/ref.ts

@@ -20,9 +20,18 @@ export function composeRef<T>(...refs: React.Ref<T>[]): React.Ref<T> {
   };
 }
 
-export function supportRef(node: any): boolean {
+export function supportRef(nodeOrComponent: any): boolean {
+  // Function component node
+  if (
+    nodeOrComponent.type &&
+    nodeOrComponent.type.prototype &&
+    !nodeOrComponent.type.prototype.render
+  ) {
+    return false;
+  }
+
   // Function component
-  if (node.type && node.type.prototype && !node.type.prototype.render) {
+  if (typeof nodeOrComponent === 'function' && !nodeOrComponent.prototype.render) {
     return false;
   }
 

tests/ref.test.js

@@ -22,6 +22,7 @@ describe('ref', () => {
           <FC />
         </div>,
       );
+      expect(supportRef(FC)).toBeFalsy();
       expect(supportRef(wrapper.props().children)).toBeFalsy();
     });
 
@@ -32,6 +33,7 @@ describe('ref', () => {
           <FRC />
         </div>,
       );
+      expect(supportRef(FRC)).toBeTruthy();
       expect(supportRef(wrapper.props().children)).toBeTruthy();
     });
 
@@ -48,6 +50,7 @@ describe('ref', () => {
           <CC />
         </div>,
       );
+      expect(supportRef(CC)).toBeTruthy();
       expect(supportRef(wrapper.props().children)).toBeTruthy();
     });
   });