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');
+ }
});
});