// @flow
import React from 'react'

import { mount } from 'enzyme'
import styled from '../index'
import ThemeProvider from '../../models/ThemeProvider'
import withTheme from '../../hoc/withTheme'

describe('withTheme', () => {
  it('should set innerRef instead of ref for stateless function components when rendered by a functional component that forwards props', () => {
    const Component = props => <div/>;

    const ThemedComponent = withTheme(Component);

    const WrapperComponent = props => <ThemedComponent {...props} />;

    const StyledComponent = styled(WrapperComponent)``;

    const wrapper = mount(
      <ThemeProvider theme={{}}>
        <StyledComponent />
      </ThemeProvider>
    )
    expect(wrapper.find(Component).first().prop('innerRef')).not.toBeUndefined()
  })
});