Add form wrappedComponentRef test

This commit is contained in:
Wei Zhu 2017-10-31 13:56:43 +08:00 committed by Benjy Cui
parent 7ff7519207
commit 8c904de7d4

View File

@ -1,5 +1,5 @@
import React from 'react';
import { shallow } from 'enzyme';
import { shallow, mount } from 'enzyme';
import Form from '..';
describe('Form', () => {
@ -9,4 +9,28 @@ describe('Form', () => {
);
expect(wrapper.hasClass('ant-form-hide-required-mark')).toBe(true);
});
describe('wrappedComponentRef', () => {
it('warns on functional component', () => {
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);
});
});
});