From 768033866edb7f2f211419c983b98b1ca33b8190 Mon Sep 17 00:00:00 2001
From: Florian Bouron <bouron.florian@yahoo.fr>
Date: Sat, 17 Feb 2024 20:46:39 +0100
Subject: [PATCH] Cleanup useMainToken

---
 .../hooks/useMainToken/useMainToken.test.ts   | 28 +++++++++++++------
 1 file changed, 20 insertions(+), 8 deletions(-)

diff --git a/packages/extension/src/hooks/useMainToken/useMainToken.test.ts b/packages/extension/src/hooks/useMainToken/useMainToken.test.ts
index bff5b97ba..e39bf5567 100644
--- a/packages/extension/src/hooks/useMainToken/useMainToken.test.ts
+++ b/packages/extension/src/hooks/useMainToken/useMainToken.test.ts
@@ -12,21 +12,33 @@ vi.mock('../../contexts', () => ({
 }));
 
 describe('useMainToken', () => {
-  beforeEach(() => {
-    vi.resetAllMocks();
-  });
-
   it('returns XRP_TOKEN for default network', () => {
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    vi.mocked(contexts.useNetwork).mockReturnValue({ chainName: 'SomeChain' } as any);
+    vi.mocked(contexts.useNetwork).mockReturnValue({
+      chainName: Chain.XRPL,
+      reconnectToNetwork: vi.fn(),
+      switchNetwork: vi.fn(),
+      resetNetwork: vi.fn(),
+      switchChain: vi.fn(),
+      networkName: 'XRPL',
+      isConnectionFailed: false,
+      hasOfflineBanner: false
+    });
 
     const { result } = renderHook(() => useMainToken());
     expect(result.current).toBe(XRP_TOKEN);
   });
 
   it('returns XAH_TOKEN for XAHAU_TESTNET', () => {
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    vi.mocked(contexts.useNetwork).mockReturnValue({ chainName: Chain.XAHAU } as any);
+    vi.mocked(contexts.useNetwork).mockReturnValue({
+      chainName: Chain.XAHAU,
+      reconnectToNetwork: vi.fn(),
+      switchNetwork: vi.fn(),
+      resetNetwork: vi.fn(),
+      switchChain: vi.fn(),
+      networkName: 'Xahau',
+      isConnectionFailed: false,
+      hasOfflineBanner: false
+    });
 
     const { result } = renderHook(() => useMainToken());
     expect(result.current).toBe(XAH_TOKEN);