import React from 'react'; import { mount } from 'enzyme'; import Avatar from '..'; import mountTest from '../../../tests/shared/mountTest'; import rtlTest from '../../../tests/shared/rtlTest'; describe('Avatar Render', () => { mountTest(Avatar); rtlTest(Avatar); let originOffsetWidth; beforeAll(() => { // Mock offsetHeight originOffsetWidth = Object.getOwnPropertyDescriptor(HTMLElement.prototype, 'offsetWidth').get; Object.defineProperty(HTMLElement.prototype, 'offsetWidth', { get() { if (this.className === 'ant-avatar-string') { return 100; } return 80; }, }); }); afterAll(() => { // Restore Mock offsetHeight Object.defineProperty(HTMLElement.prototype, 'offsetWidth', { get: originOffsetWidth, }); }); it('Render long string correctly', () => { const wrapper = mount(TestString); const children = wrapper.find('.ant-avatar-string'); expect(children.length).toBe(1); }); it('should render fallback string correctly', () => { const div = global.document.createElement('div'); global.document.body.appendChild(div); const wrapper = mount(Fallback, { attachTo: div }); wrapper.find('img').simulate('error'); const children = wrapper.find('.ant-avatar-string'); expect(children.length).toBe(1); expect(children.text()).toBe('Fallback'); wrapper.detach(); global.document.body.removeChild(div); }); it('should handle onError correctly', () => { const LOAD_FAILURE_SRC = 'http://error.url'; const LOAD_SUCCESS_SRC = 'https://zos.alipayobjects.com/rmsportal/ODTLcjxAfvqbxHnVXCYX.png'; const div = global.document.createElement('div'); global.document.body.appendChild(div); class Foo extends React.Component { state = { src: LOAD_FAILURE_SRC, }; handleImgError = () => { this.setState({ src: LOAD_SUCCESS_SRC, }); return false; }; render() { const { src } = this.state; return ; } } const wrapper = mount(, { attachTo: div }); // mock img load Error, since jsdom do not load resource by default // https://github.com/jsdom/jsdom/issues/1816 wrapper.find('img').simulate('error'); expect(wrapper.find(Avatar).instance().state.isImgExist).toBe(true); expect(div.querySelector('img').getAttribute('src')).toBe(LOAD_SUCCESS_SRC); wrapper.detach(); global.document.body.removeChild(div); }); it('should show image on success after a failure state', () => { const LOAD_FAILURE_SRC = 'http://error.url'; const LOAD_SUCCESS_SRC = 'https://zos.alipayobjects.com/rmsportal/ODTLcjxAfvqbxHnVXCYX.png'; const div = global.document.createElement('div'); global.document.body.appendChild(div); // simulate error src url const wrapper = mount(Fallback, { attachTo: div }); wrapper.find('img').simulate('error'); expect(wrapper.find(Avatar).instance().state.isImgExist).toBe(false); expect(wrapper.find('.ant-avatar-string').length).toBe(1); // simulate successful src url wrapper.setProps({ src: LOAD_SUCCESS_SRC }); wrapper.update(); expect(wrapper.find(Avatar).instance().state.isImgExist).toBe(true); expect(wrapper.find('.ant-avatar-image').length).toBe(1); // cleanup wrapper.detach(); global.document.body.removeChild(div); }); it('should calculate scale of avatar children correctly', () => { const wrapper = mount(Avatar); expect(wrapper.state().scale).toBe(0.72); Object.defineProperty(HTMLElement.prototype, 'offsetWidth', { get() { if (this.className === 'ant-avatar-string') { return 100; } return 40; }, }); wrapper.setProps({ children: 'xx' }); expect(wrapper.state().scale).toBe(0.32); }); it('should calculate scale of avatar children correctly with gap', () => { const wrapper = mount(Avatar); expect(wrapper.state().scale).toBe(0.36); }); it('should warning when pass a string as icon props', () => { const warnSpy = jest.spyOn(console, 'error').mockImplementation(() => {}); mount(); expect(warnSpy).not.toHaveBeenCalled(); mount(); expect(warnSpy).toHaveBeenCalledWith( `Warning: [antd: Avatar] \`icon\` is using ReactNode instead of string naming in v4. Please check \`user\` at https://ant.design/components/icon`, ); warnSpy.mockRestore(); }); it('support size is number', () => { const wrapper = mount(TestString); expect(wrapper).toMatchRenderedSnapshot(); }); });