import React from 'react';
import { shallow } from 'enzyme';

const Add = ({ a, b }) => <div>{a + b}</div>;

Add.defaultProps = {
  a: 2,
  b: 3,
};

const MemoAdd = React.memo(Add);

describe('Enzyme Shallow Default Props Test', () => {
  it('default props should be applied to components', () => {
    const wrapper = shallow(<Add />);
    expect(wrapper.html()).toEqual('<div>5</div>');
  });

  it('default props should be applied to memo components', () => {
    const wrapper = shallow(<MemoAdd />);
    expect(wrapper.html()).toEqual('<div>5</div>');
  });
});