fix: update tests and theme selector

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

Hide file tree

Showing 4 changed files with 17 additions and 7 deletions.
diff --git a/...t-react/src/Components/ConnectWalletButtonWithModal/Components/ConnectButtonWithModal.tsx b/...t-react/src/Components/ConnectWalletButtonWithModal/Components/ConnectButtonWithModal.tsx
@@ -1,9 +1,9 @@
-import { useContext, useMemo } from 'react';
-import { ThemeContext } from '../../../provider/ThemeProvider';
+import { useMemo } from 'react';
+import { useThemeSelector } from '../../../provider/ThemeProvider';
 import { createButtonWithModal } from './Wrapped/ConnectModalWithButtonWrapped';
 
 export const ConnectButtonWithModal = () => {
-    const { theme } = useContext(ThemeContext);
+    const { theme } = useThemeSelector();
 
     const ModalWithButton = useMemo(() => createButtonWithModal(), []);
 

diff --git a/packages/dapp-kit-react/src/Components/ThemeSelector.tsx b/packages/dapp-kit-react/src/Components/ThemeSelector.tsx
@@ -1,14 +1,14 @@
 // ThemeSelector.js
 
-import React, { useContext } from 'react';
+import React from 'react';
 // eslint-disable-next-line import/no-named-as-default
 import styled from 'styled-components';
-import { ThemeContext } from '../provider/ThemeProvider';
+import { useThemeSelector } from '../provider/ThemeProvider';
 
 const Button = styled.button``;
 
 const ThemeSelector = () => {
-    const { toggleTheme } = useContext(ThemeContext);
+    const { toggleTheme } = useThemeSelector();
 
     return <Button onClick={toggleTheme}>Toggle Theme</Button>;
 };

diff --git a/packages/dapp-kit-react/src/provider/ThemeProvider.tsx b/packages/dapp-kit-react/src/provider/ThemeProvider.tsx
@@ -44,4 +44,14 @@ const ThemeProvider = ({ children }: { children: ReactNode }) => {
     );
 };
 
+export const useThemeSelector = (): ThemeContextProperties => {
+    const context = React.useContext(ThemeContext);
+
+    if (context === undefined) {
+        throw new Error('useThemeSelector must be used within a ThemeProvider');
+    }
+
+    return context;
+};
+
 export { ThemeProvider, ThemeContext, type ThemeContextProperties };
diff --git a/packages/dapp-kit-react/vite.config.ts b/packages/dapp-kit-react/vite.config.ts
@@ -21,7 +21,7 @@ export default defineConfig({
             lines: 90,
             statements: 90,
             functions: 90,
-            branches: 85,
+            branches: 80,
         },
         globals: true,
     },