ant-design/components/form/__tests__/index.test.js

import React from 'react';
import { shallow, mount } from 'enzyme';
import Form from '..';

describe('Form', () => {
  it('hideRequiredMark', () => {
    const wrapper = shallow(
      <Form hideRequiredMark />
    );
    expect(wrapper.hasClass('ant-form-hide-required-mark')).toBe(true);
  });

  describe('wrappedComponentRef', () => {
    it('warns on functional component', () => {
      if (process.env.REACT === '15') {
        return;
      }
      const spy = jest.spyOn(console, 'error').mockImplementation(() => {});
      const TestForm = () => <Form />;
      const Wrapped = Form.create()(TestForm);
      mount(<Wrapped wrappedComponentRef={() => {}} />);
      expect(spy).toHaveBeenCalled();
      spy.mockReset();
      spy.mockRestore();
    });

    it('get component ref', () => {
      class TestForm extends React.Component { // eslint-disable-line
        render() {
          return <Form />;
        }
      }
      const Wrapped = Form.create()(TestForm);
      let form;
      mount(<Wrapped wrappedComponentRef={node => form = node} />);
      expect(form).toBeInstanceOf(TestForm);
    });
  });
});