fix: unit tests

vechain · Dec 4, 2023 · a7d2639 · a7d2639
1 parent ff5cb21
commit a7d2639
Show file tree

Hide file tree

Showing 4 changed files with 21 additions and 17 deletions.
diff --git a/packages/dapp-kit-react/src/provider/ThemeProvider.tsx b/packages/dapp-kit-react/src/provider/ThemeProvider.tsx
@@ -9,7 +9,7 @@ interface Theme {
     mode: ThemeMode;
 }
 
-interface ContextProperties {
+interface ThemeContextProperties {
     theme: Theme;
     toggleTheme: () => void;
 }
@@ -18,11 +18,9 @@ const defaultTheme: Theme = {
     mode: 'LIGHT',
 };
 
-const ThemeContext = createContext<ContextProperties>({
-    theme: defaultTheme,
-    // eslint-disable-next-line @typescript-eslint/no-empty-function
-    toggleTheme: () => {},
-});
+const ThemeContext = createContext<ThemeContextProperties | undefined>(
+    undefined,
+);
 
 const ThemeProvider = ({ children }: { children: ReactNode }) => {
     const [currentTheme, setCurrentTheme] = useState(defaultTheme);
@@ -46,4 +44,4 @@ const ThemeProvider = ({ children }: { children: ReactNode }) => {
     );
 };
 
-export { ThemeProvider, ThemeContext };
+export { ThemeProvider, ThemeContext, type ThemeContextProperties };
diff --git a/packages/dapp-kit-react/test/helpers/react-test-helpers.tsx b/packages/dapp-kit-react/test/helpers/react-test-helpers.tsx
@@ -4,12 +4,15 @@ import {
     DAppKitProvider,
     SelectWalletModal,
 } from '../../src';
+import { ThemeProvider } from '../../src/provider/ThemeProvider';
 
 export const wrapper = ({ children }: { children?: React.ReactNode }) => (
     <DAppKitProvider nodeUrl="https://testnet.vechain.org">
-        {children}
-        <ConnectWalletButtonWithModal />
-        <SelectWalletModal isOpen={false} />
-        <ConnectWalletModal isOpen={false} />
+        <ThemeProvider>
+            {children}
+            <ConnectWalletButtonWithModal />
+            <SelectWalletModal isOpen={false} />
+            <ConnectWalletModal isOpen={false} />
+        </ThemeProvider>
     </DAppKitProvider>
 );
diff --git a/packages/dapp-kit-react/test/provider/ThemeProvider.test.tsx b/packages/dapp-kit-react/test/provider/ThemeProvider.test.tsx
@@ -12,12 +12,15 @@ describe('useThemeContext', () => {
 
         expect(result.current).toBeDefined();
 
-        expect(result.current.theme).toEqual({ mode: 'LIGHT' });
+        expect(result.current?.theme).toEqual({ mode: 'LIGHT' });
 
-        result.current.toggleTheme();
+        result.current?.toggleTheme();
 
-        await waitFor(() => {
-            expect(result.current.theme).toEqual({ mode: 'DARK' });
-        });
+        await waitFor(
+            () => {
+                expect(result.current?.theme).toEqual({ mode: 'DARK' });
+            },
+            { timeout: 3000 },
+        );
     });
 });
diff --git a/packages/dapp-kit-ui/src/components/connect-button-with-modal.ts b/packages/dapp-kit-ui/src/components/connect-button-with-modal.ts
@@ -83,7 +83,7 @@ export class ConnectButtonWithModal extends LitElement {
     }
 
     private handleOpen = (): void => {
-        this.open = false;
+        this.open = true;
         DAppKitUI.wallet.disconnect();
         this.requestUpdate();
     };