import React from 'react'; import { mount } from 'enzyme'; import { themr } from 'react-css-themr'; import { CHIP } from '../../identifiers'; import { tooltipFactory } from '../../tooltip'; import { chipFactory } from '../Chip'; const Avatar = ({ title }) => {title}; // eslint-disable-line react/prop-types const Chip = themr(CHIP)(chipFactory(Avatar)); describe('Chip', () => { describe('with avatar', () => { it('adds the avatar class to the element', () => { const wrapper = mount( Test , ); const chipNode = wrapper.find('div').node; expect(chipNode.className).toMatch(/\bavatar-class\b/); }); it('works with non-flat children', () => { const TooltippedChip = tooltipFactory()(Chip); const wrapper = mount( Test , ); const chipNode = wrapper.find('div').node; expect(chipNode.className).toMatch(/\bavatar-class\b/); }); }); describe('without avatar', () => { it('does not add avatar class to the element', () => { const wrapper = mount( Test , ); const chipNode = wrapper.find('div').node; expect(chipNode.className).toNotMatch(/\bavatar-class\b/); }); }); });