Add tests for React.forwardRef

ctrlplusb · Aug 10, 2018 · 0d9f69e · 0d9f69e
1 parent 7cd93e0
commit 0d9f69e
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/src/__tests__/index.test.js b/src/__tests__/index.test.js
@@ -467,5 +467,26 @@ describe('reactTreeWalker', () => {
         expect(actual).toEqual(expected)
       })
     })
+
+    it('supports forwardRef', () => {
+      class Foo extends ReactComponent {
+        render() {
+          return this.props.children
+        }
+      }
+
+      const Bar = React.forwardRef((props, ref) => <Foo ref={ref} {...props} />)
+
+      const tree = <Bar>foo</Bar>
+
+      const elements = []
+      return reactTreeWalker(tree, element => {
+        elements.push(element)
+      }).then(() => {
+        expect(elements.pop()).toBe('foo')
+        expect(elements.pop().type).toBe(Foo)
+        expect(elements.pop().type).toBe(Bar)
+      })
+    })
   })
 })