diff --git a/test/common/img/ShlinkLogo.test.tsx b/test/common/img/ShlinkLogo.test.tsx index 09793371..d1c6cc61 100644 --- a/test/common/img/ShlinkLogo.test.tsx +++ b/test/common/img/ShlinkLogo.test.tsx @@ -1,25 +1,17 @@ -import { shallow, ShallowWrapper } from 'enzyme'; +import { render } from '@testing-library/react'; import { ShlinkLogo, ShlinkLogoProps } from '../../../src/common/img/ShlinkLogo'; import { MAIN_COLOR } from '../../../src/utils/theme'; describe('', () => { - let wrapper: ShallowWrapper; - const createWrapper = (props: ShlinkLogoProps) => { - wrapper = shallow(); - - return wrapper; - }; - - afterEach(() => wrapper?.unmount()); + const setUp = (props: ShlinkLogoProps) => render(); it.each([ [undefined, MAIN_COLOR], ['red', 'red'], ['white', 'white'], ])('renders expected color', (color, expectedColor) => { - const wrapper = createWrapper({ color }); - - expect(wrapper.find('g').prop('fill')).toEqual(expectedColor); + const { container } = setUp({ color }); + expect(container.querySelector('g')).toHaveAttribute('fill', expectedColor); }); it.each([ @@ -27,8 +19,12 @@ describe('', () => { ['foo', 'foo'], ['bar', 'bar'], ])('renders expected class', (className, expectedClassName) => { - const wrapper = createWrapper({ className }); + const { container } = setUp({ className }); - expect(wrapper.prop('className')).toEqual(expectedClassName); + if (expectedClassName) { + expect(container.firstChild).toHaveAttribute('class', expectedClassName); + } else { + expect(container.firstChild).not.toHaveAttribute('class'); + } }); });