fix test

Author: theothersideofgods

packages/react/__tests__/hooks/useChain.test.ts

@@ -2,15 +2,15 @@
  * @jest-environment jsdom
  */
 
-import { renderHook, act, waitFor } from '@testing-library/react';
-import { useChain } from '../../src/hooks/useChain';
-import { useWalletManager } from '../../src/hooks/useWalletManager';
 import { ChainNameNotExist, WalletState } from '@interchain-kit/core';
+import { ChainWalletState, InterchainStore, WalletManagerStore } from '@interchain-kit/store';
+import { WalletStore } from '@interchain-kit/store';
+import { act, renderHook, waitFor } from '@testing-library/react';
 
-import { MockWallet } from '../helpers/mock-wallet';
 import { useWalletModal } from '../../src/hooks';
-import { WalletManagerStore, ChainWalletState, InterchainStore } from '@interchain-kit/store';
-import { WalletStore } from '@interchain-kit/store';
+import { useChain } from '../../src/hooks/useChain';
+import { useWalletManager } from '../../src/hooks/useWalletManager';
+import { MockWallet } from '../helpers/mock-wallet';
 
 jest.mock('../../src/hooks/useWalletManager');
 jest.mock('../../src/hooks/useWalletModal.ts');
@@ -48,6 +48,7 @@ describe('useChain', () => {
     addAssetList: jest.fn(),
     getChainById: jest.fn(),
     getAssetListByChainId: jest.fn(),
+    getWalletOfType: jest.fn(),
   } as jest.Mocked<WalletStore>;
 
 
@@ -95,7 +96,7 @@ describe('useChain', () => {
     getSigningClient: jest.fn(),
     getEnv: jest.fn(),
     getDownloadLink: jest.fn(),
-  } as jest.Mocked<WalletManagerStore>
+  } as jest.Mocked<WalletManagerStore>;
 
   const mockWalletModal = {
     open: jest.fn(),
@@ -139,7 +140,7 @@ describe('useChain', () => {
       expect(result.current.username).toBe('test-user');
       expect(result.current.address).toBe('test-address');
       expect(result.current.rpcEndpoint).toBe('http://localhost:26657');
-    })
+    });
 
 
   });
@@ -157,7 +158,7 @@ describe('useChain', () => {
     await waitFor(() => {
       expect(mockWalletManager.setCurrentChainName).toHaveBeenCalledWith('test-chain');
       expect(mockWalletModal.open).toHaveBeenCalled();
-    })
+    });
   });
 
   it('should call disconnect when disconnect is invoked', async () => {
@@ -173,7 +174,7 @@ describe('useChain', () => {
     await waitFor(() => {
 
       expect(mockWalletManager.disconnect).toHaveBeenCalledWith('test-wallet', 'test-chain');
-    })
+    });
 
   });
 
@@ -190,7 +191,7 @@ describe('useChain', () => {
     await waitFor(() => {
       expect(mockWalletManager.setCurrentChainName).toHaveBeenCalledWith('test-chain');
       expect(mockWalletModal.open).toHaveBeenCalled();
-    })
+    });
 
   });
 
@@ -206,6 +207,6 @@ describe('useChain', () => {
 
     await waitFor(() => {
       expect(mockWalletModal.close).toHaveBeenCalled();
-    })
+    });
   });
 });

packages/react/__tests__/modal/modal.test.tsx

@@ -2,16 +2,18 @@
  * @jest-environment jsdom
  */
 
+import "@testing-library/jest-dom";
+
+import { InterchainStore, WalletStore } from "@interchain-kit/store";
+import { fireEvent, render, screen } from "@testing-library/react";
 import React from "react";
-import { render, screen, fireEvent } from "@testing-library/react";
+
 import {
-  WalletModalElementProps,
   WalletModalElement,
+  WalletModalElementProps,
 } from "../../src/modal/modal";
-import "@testing-library/jest-dom";
 import { transferToWalletUISchema } from "../../src/utils";
 import { MockWallet } from "../helpers/mock-wallet";
-import { WalletStore, InterchainStore } from "@interchain-kit/store";
 
 describe("WalletModal", () => {
   const mockWallet = new MockWallet({
@@ -45,6 +47,7 @@ describe("WalletModal", () => {
     addAssetList: jest.fn(),
     getChainById: jest.fn(),
     getAssetListByChainId: jest.fn(),
+    getWalletOfType: jest.fn(),
   } as jest.Mocked<WalletStore>;
   const mockProps: WalletModalElementProps = {
     shouldShowList: false,

packages/store/__tests__/chain-wallet-store.test.ts

@@ -197,7 +197,7 @@ describe('ChainWalletStore', () => {
       expect(mockInterchainStore.updateChainWalletState).toHaveBeenCalledWith(
         'keplr',
         'cosmoshub',
-        { walletState: WalletState.Connecting }
+        { walletState: WalletState.Connecting, errorMessage: '' }
       );
       expect(mockWallet.connect).toHaveBeenCalledWith('cosmoshub-4');
       expect(mockWallet.getAccount).toHaveBeenCalledWith('cosmoshub-4');
@@ -217,7 +217,7 @@ describe('ChainWalletStore', () => {
       expect(mockInterchainStore.updateChainWalletState).toHaveBeenCalledWith(
         'keplr',
         'cosmoshub',
-        { walletState: WalletState.Connecting }
+        { walletState: WalletState.Connecting, errorMessage: '' }
       );
       expect(mockInterchainStore.updateChainWalletState).toHaveBeenCalledWith(
         'keplr',

packages/store/src/wallet-manager/chain-wallet-store.ts

@@ -52,7 +52,6 @@ export class ChainWalletStore extends BaseWallet {
     }
 
     try {
-      console.log(1);
       this.store.updateChainWalletState(this.wallet.info.name, this.chain.chainName, { walletState: WalletState.Connecting, errorMessage: '' });
       await this.wallet.connect(this.chain.chainId);
       const account = await this.getAccount();