3 * Copyright (C) 2009-2021 SonarSource SA
4 * mailto:info AT sonarsource DOT com
6 * This program is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 3 of the License, or (at your option) any later version.
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public License
17 * along with this program; if not, write to the Free Software Foundation,
18 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
20 import { shallow } from 'enzyme';
21 import * as React from 'react';
22 import { click } from '../../../../../../../helpers/testUtils';
23 import BadgeButton from '../BadgeButton';
24 import { BadgeType } from '../utils';
26 it('should display correctly', () => {
27 expect(getWrapper()).toMatchSnapshot();
28 expect(getWrapper({ selected: true })).toMatchSnapshot();
29 expect(getWrapper({ type: BadgeType.measure })).toMatchSnapshot();
32 it('should return the badge type on click', () => {
33 const onClick = jest.fn();
34 const wrapper = getWrapper({ onClick });
35 click(wrapper.find('Button'));
36 expect(onClick).toHaveBeenCalledWith(BadgeType.marketing);
39 function getWrapper(props = {}) {
44 type={BadgeType.marketing}