From 407ad322b5fff366b3e96fb7c8208533be4b9418 Mon Sep 17 00:00:00 2001
From: katspaugh <katspaugh@gmail.com>
Date: Wed, 29 May 2024 11:39:25 +0200
Subject: [PATCH] Fix tests

---
 .../settings/SettingsHeader/index.test.tsx    | 34 +++++++++++++------
 .../settings/SettingsHeader/index.tsx         | 17 +++++++---
 2 files changed, 36 insertions(+), 15 deletions(-)

diff --git a/src/components/settings/SettingsHeader/index.test.tsx b/src/components/settings/SettingsHeader/index.test.tsx
index b8ce692c58..decdb4e385 100644
--- a/src/components/settings/SettingsHeader/index.test.tsx
+++ b/src/components/settings/SettingsHeader/index.test.tsx
@@ -1,9 +1,11 @@
-import SettingsHeader from '@/components/settings/SettingsHeader/index'
+import { SettingsHeader } from '@/components/settings/SettingsHeader/index'
+import { CONFIG_SERVICE_CHAINS } from '@/tests/mocks/chains'
 import * as safeAddress from '@/hooks/useSafeAddress'
-import * as feature from '@/hooks/useChains'
 
 import { render } from '@/tests/test-utils'
 import { faker } from '@faker-js/faker'
+import { FEATURES } from '@/utils/chains'
+import { ChainInfo } from '@safe-global/safe-gateway-typescript-sdk'
 
 describe('SettingsHeader', () => {
   beforeEach(() => {
@@ -17,15 +19,21 @@ describe('SettingsHeader', () => {
     })
 
     it('displays safe specific preferences if on a safe', () => {
-      const result = render(<SettingsHeader />)
+      const result = render(<SettingsHeader safeAddress="0x1234" chain={CONFIG_SERVICE_CHAINS[0]} />)
 
       expect(result.getByText('Setup')).toBeInTheDocument()
     })
 
     it('displays Notifications if feature is enabled', () => {
-      jest.spyOn(feature, 'useHasFeature').mockReturnValue(true)
-
-      const result = render(<SettingsHeader />)
+      const result = render(
+        <SettingsHeader
+          safeAddress="0x1234"
+          chain={{
+            ...CONFIG_SERVICE_CHAINS[0],
+            features: [FEATURES.PUSH_NOTIFICATIONS] as unknown as ChainInfo['features'],
+          }}
+        />,
+      )
 
       expect(result.getByText('Notifications')).toBeInTheDocument()
     })
@@ -38,7 +46,7 @@ describe('SettingsHeader', () => {
     })
 
     it('displays general preferences if no safe is open', () => {
-      const result = render(<SettingsHeader />)
+      const result = render(<SettingsHeader safeAddress="" chain={CONFIG_SERVICE_CHAINS[0]} />)
 
       expect(result.getByText('Cookies')).toBeInTheDocument()
       expect(result.getByText('Appearance')).toBeInTheDocument()
@@ -47,9 +55,15 @@ describe('SettingsHeader', () => {
     })
 
     it('displays Notifications if feature is enabled', () => {
-      jest.spyOn(feature, 'useHasFeature').mockReturnValue(true)
-
-      const result = render(<SettingsHeader />)
+      const result = render(
+        <SettingsHeader
+          safeAddress=""
+          chain={{
+            ...CONFIG_SERVICE_CHAINS[0],
+            features: [FEATURES.PUSH_NOTIFICATIONS] as unknown as ChainInfo['features'],
+          }}
+        />,
+      )
 
       expect(result.getByText('Notifications')).toBeInTheDocument()
     })
diff --git a/src/components/settings/SettingsHeader/index.tsx b/src/components/settings/SettingsHeader/index.tsx
index 856515a387..f20def13f1 100644
--- a/src/components/settings/SettingsHeader/index.tsx
+++ b/src/components/settings/SettingsHeader/index.tsx
@@ -7,11 +7,15 @@ import css from '@/components/common/PageHeader/styles.module.css'
 import useSafeAddress from '@/hooks/useSafeAddress'
 import { useCurrentChain } from '@/hooks/useChains'
 import { isRouteEnabled } from '@/utils/chains'
+import madProps from '@/utils/mad-props'
 
-const SettingsHeader = (): ReactElement => {
-  const safeAddress = useSafeAddress()
-  const chain = useCurrentChain()
-
+export const SettingsHeader = ({
+  safeAddress,
+  chain,
+}: {
+  safeAddress: ReturnType<typeof useSafeAddress>
+  chain: ReturnType<typeof useCurrentChain>
+}): ReactElement => {
   const navItems = safeAddress
     ? settingsNavItems.filter((route) => isRouteEnabled(route.href, chain))
     : generalSettingsNavItems
@@ -28,4 +32,7 @@ const SettingsHeader = (): ReactElement => {
   )
 }
 
-export default SettingsHeader
+export default madProps(SettingsHeader, {
+  safeAddress: useSafeAddress,
+  chain: useCurrentChain,
+})