fix multipleWallet didn't sync up chainMap

Author: theothersideofgods

packages/core/__tests__/wallets/base-wallet.test.ts

@@ -63,4 +63,25 @@ describe("BaseWallet", () => {
     const account = await wallet.getAccount("test-chain-id");
     expect(account.address).toBe("test-address");
   });
+
+  it('should set chain map', () => {
+    const chains: Chain[] = [
+      { chainId: 'chain-1', chainName: 'Chain 1' } as Chain,
+      { chainId: 'chain-2', chainName: 'Chain 2' } as Chain,
+    ];
+    wallet.setChainMap(chains);
+    expect(wallet.chainMap.size).toBe(2);
+    expect(wallet.chainMap.get('chain-1')?.chainName).toBe('Chain 1');
+  })
+
+  it('should addChain', () => {
+    const chain: Chain = { chainId: 'chain-3', chainName: 'Chain 3' } as Chain;
+    wallet.addChain(chain);
+    expect(wallet.chainMap.size).toBe(1);
+    expect(wallet.chainMap.get('chain-3')?.chainName).toBe('Chain 3');
+  })
+
+
+
+
 });

packages/core/__tests__/wallets/cosmos-wallet.test.ts

@@ -139,4 +139,6 @@ describe('CosmosWallet', () => {
 
         await expect(wallet.addSuggestChain('cosmoshub-4')).rejects.toThrow('suggestion error');
     });
+
+
 });

packages/core/__tests__/wallets/multichain-wallet.test.ts

@@ -10,6 +10,7 @@ describe('MultiChainWallet', () => {
     beforeEach(() => {
         mockBaseWallet = {
             setChainMap: jest.fn(),
+            addChain: jest.fn(),
             setAssetLists: jest.fn(),
             init: jest.fn(),
             connect: jest.fn(),
@@ -43,6 +44,16 @@ describe('MultiChainWallet', () => {
         expect(mockBaseWallet.setChainMap).toHaveBeenCalledWith(chains);
     });
 
+    it('should add chain and propagate to network wallets', () => {
+        const chain: Chain = { chainId: 'cosmoshub-4', chainType: 'cosmos', chainName: 'Cosmos Hub' };
+
+        multiChainWallet.setNetworkWallet('cosmos', mockBaseWallet);
+        multiChainWallet.addChain(chain);
+
+        expect(multiChainWallet.chainMap.get(chain.chainId)).toBe(chain);
+        expect(mockBaseWallet.addChain).toHaveBeenCalledWith(chain);
+    })
+
     it('should initialize all network wallets', async () => {
         multiChainWallet.setNetworkWallet('cosmos', mockBaseWallet);
 
@@ -109,4 +120,5 @@ describe('MultiChainWallet', () => {
         expect(provider).toBe('mockProvider');
         expect(mockBaseWallet.getProvider).toHaveBeenCalledWith(chain.chainId);
     });
+
 });

packages/core/src/wallets/base-wallet.ts

@@ -9,7 +9,7 @@ export abstract class BaseWallet {
   errorMessage: string
   walletState: WalletState
   events: EventEmitter<WalletEvents> = new EventEmitter()
-  chainMap: Map<Chain['chainId'], Chain>
+  chainMap: Map<Chain['chainId'], Chain> = new Map()
   assetLists: AssetList[] = []
   client: any
   constructor(info: Wallet) {

packages/core/src/wallets/multichain-wallet.ts

@@ -8,7 +8,6 @@ import { isInstanceOf } from '../utils';
 export class MultiChainWallet extends BaseWallet {
 
   networkWalletMap: Map<Chain['chainType'], BaseWallet> = new Map()
-  chainMap: Map<Chain['chainId'], Chain> = new Map()
 
   constructor(info?: Wallet, networkWalletMap?: Map<Chain['chainType'], BaseWallet>) {
     super(info);
@@ -36,6 +35,13 @@ export class MultiChainWallet extends BaseWallet {
     })
   }
 
+  addChain(chain: Chain): void {
+    this.chainMap.set(chain.chainId, chain);
+    this.networkWalletMap.forEach(wallet => {
+      wallet.addChain(chain)
+    })
+  }
+
   setAssetLists(assetLists: AssetList[]): void {
     this.networkWalletMap.forEach(wallet => {
       wallet.setAssetLists(assetLists)

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

@@ -94,20 +94,6 @@ describe("useChainWallet", () => {
         expect(result.current.logoUrl).toBe("http://logo.url")
     })
 
-    it("should call connect and getAccount when connect is invoked", async () => {
-        const chainName = "test-chain"
-        const walletName = "test-wallet"
-
-        const { result } = renderHook(() => useChainWallet(chainName, walletName))
-
-        await result.current.connect()
-
-        expect(mockWalletManager.setCurrentWalletName).toHaveBeenCalledWith(walletName)
-        expect(mockWalletManager.setCurrentChainName).toHaveBeenCalledWith(chainName)
-        expect(mockWalletManager.connect).toHaveBeenCalledWith(walletName, chainName)
-        expect(mockWalletManager.getAccount).toHaveBeenCalledWith(walletName, chainName)
-    })
-
     it("should call disconnect when disconnect is invoked", () => {
         const chainName = "test-chain"
         const walletName = "test-wallet"