Cleanup useMainToken

GemWallet · Feb 17, 2024 · 7680338 · 7680338
1 parent 915008a
commit 7680338
Showing 1 changed file with 20 additions and 8 deletions.
diff --git 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);