Fix forwardRef component nested with a memo HOC (#15)

* Fix forwardRef component nested in memo HOC In the case of: memo(forwardRef(SomeComponent)) The memo component would be correctly detected but would assume that it may only contain a class or function component and render it. Instead of rendering the nested component, this fix returns a proper element from the memo component visitor. * Update visitor test for memo components

Author: kitten

src/__tests__/visitor.test.js

@@ -215,7 +215,7 @@ describe('visitElement', () => {
  })
 
  it('walks over memo components', () => {
- const Test = React.memo(() => <Noop />)
+ const Test = React.memo(Noop)
  const children = visitElement(<Test />, [], () => {})
  expect(children.length).toBe(1)
  expect(children[0].type).toBe(Noop)

src/visitor.js

@@ -134,8 +134,7 @@ export const visitElement = (
  case REACT_MEMO_TYPE: {
  const memoElement = ((element: any): MemoElement)
  const type = memoElement.type.type
- const fauxElement = (createElement((type: any), memoElement.props): any)
- const child = render(type, memoElement.props, queue, visitor, fauxElement)
+ const child = createElement((type: any), memoElement.props)
  return getChildrenArray(child)
  }