fix find forwardRef itself

packages/enzyme-test-suite/test/ReactWrapper-spec.jsx

@@ -210,6 +210,21 @@ describeWithDOM('mount', () => {
 
         expect(wrapper.find('.child2')).to.have.length(1);
       });
+
+      it('should find the forwardRef element itself', () => {
+        const testRef = () => {};
+        const OtherComponent = () => (
+          <div />
+        );
+        const SomeComponent = forwardRef((props, ref) => (
+          <OtherComponent className="dumb" ref={ref} />
+        ));
+
+        const wrapper = mount(<div><SomeComponent ref={testRef} /></div>);
+
+        expect(wrapper.find(SomeComponent)).to.have.length(1);
+        expect(wrapper.find(OtherComponent)).to.have.length(1);
+      });
     });
 
     describeIf(!REACT013, 'stateless components', () => {

packages/enzyme/src/selectors.js

@@ -13,7 +13,7 @@ import {
   childrenOfNode,
   hasClassName,
 } from './RSTTraversal';
-import { nodeHasType, propsOfNode } from './Utils';
+import { nodeHasType, propsOfNode, typeOf, ForwardRef } from './Utils';
 // our CSS selector parser instance
 const parser = createParser();
 
@@ -227,6 +227,12 @@ export function buildPredicate(selector) {
       if (hasUndefinedValues) {
         throw new TypeError('Enzyme::Props can’t have `undefined` values. Try using ‘findWhere()’ instead.');
       }
+      // the selector could also be a forwardRef
+      if (typeOf(selector) === ForwardRef) {
+        // re-build the predicate based on what is wrapped by forwardRef
+        // rather than the forwardRef itself
+        return buildPredicate(selector.render().props);
+      }
       return node => nodeMatchesObjectProps(node, selector);
     }
     throw new TypeError('Enzyme::Selector does not support an array, null, or empty object as a selector');