From a8f96006e199447c0949da0eab0ce5beb56cd15c Mon Sep 17 00:00:00 2001
From: Steven VAIDIE <steven.vaidie@gmail.com>
Date: Tue, 8 Sep 2020 09:45:58 +0200
Subject: [PATCH] RAC-225: fix unit tests

---
 .../src/components/Checkbox/Checkbox.unit.tsx  |  4 ++--
 akeneo-design-system/src/shared/test-util.tsx  | 18 ++++++++++++++++++
 2 files changed, 20 insertions(+), 2 deletions(-)
 create mode 100644 akeneo-design-system/src/shared/test-util.tsx

diff --git a/akeneo-design-system/src/components/Checkbox/Checkbox.unit.tsx b/akeneo-design-system/src/components/Checkbox/Checkbox.unit.tsx
index 8bbb5e2655b7..d1547ef70b1c 100644
--- a/akeneo-design-system/src/components/Checkbox/Checkbox.unit.tsx
+++ b/akeneo-design-system/src/components/Checkbox/Checkbox.unit.tsx
@@ -1,5 +1,5 @@
 import React from 'react';
-import {fireEvent, render} from '@testing-library/react';
+import {fireEvent, render} from '../../shared/test-util';
 import {Checkbox} from './Checkbox';
 
 it('it calls onChange handler when user clicks on checkbox', () => {
@@ -29,5 +29,5 @@ it('it does not call onChange handler when read-only', () => {
 it('it cannot be instantiated without handler when not readonly', () => {
     expect(() => {
         render(<Checkbox checked={true} label="Checkbox"/>);
-    }).toThrow('A Checkbox element expect a onChange attribute if not readOnly');
+    }).toThrow('A Checkbox element expect an onChange attribute if not readOnly');
 });
diff --git a/akeneo-design-system/src/shared/test-util.tsx b/akeneo-design-system/src/shared/test-util.tsx
new file mode 100644
index 000000000000..1b46961787ec
--- /dev/null
+++ b/akeneo-design-system/src/shared/test-util.tsx
@@ -0,0 +1,18 @@
+import React, {ComponentType, ReactElement, ReactNode} from 'react';
+import {render, RenderOptions} from '@testing-library/react';
+import AkeneoPim from '../themes/akeneo-pim';
+import { ThemeProvider } from 'styled-components';
+
+const AllTheProviders = ({children}: {children: ReactNode}) => {
+    return (
+        <ThemeProvider theme={AkeneoPim}>
+            {children}
+        </ThemeProvider>
+    );
+};
+
+const customRender = (ui: ReactElement, options?: Omit<RenderOptions, 'queries'>) =>
+    render(ui, { wrapper: AllTheProviders as ComponentType, ...options });
+
+export * from '@testing-library/react';
+export { customRender as render };