diff --git a/src/__tests__/index.test.js b/src/__tests__/index.test.js
index 060d0b3..af202ef 100644
--- 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) => )
+
+ const tree = foo
+
+ 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)
+ })
+ })
})
})