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) + }) + }) }) })