test(sdk)

Author: gregfromstl

packages/thirdweb/src/wallets/manager/connection-manager.test.ts

@@ -120,4 +120,61 @@ describe.runIf(process.env.TW_SECRET_KEY)("Connection Manager", () => {
       "Cannot set a wallet without an account as active",
     );
   });
+
+  it("should set active wallet and update storage", async () => {
+    const manager = createConnectionManager(storage);
+
+    await manager.setActiveWallet(wallet);
+
+    expect(manager.activeWalletStore.getValue()).toBe(wallet);
+    expect(storage.setItem).toHaveBeenCalledWith(
+      "thirdweb:active-wallet-id",
+      wallet.id,
+    );
+  });
+
+  it("should switch active wallet chain", async () => {
+    const manager = createConnectionManager(storage);
+
+    await manager.handleConnection(wallet, { client });
+
+    const newChain = {
+      id: 2,
+      name: "New Chain",
+      rpc: "https://rpc.example.com",
+    };
+
+    // Mock the switchChain method to update the activeWalletChainStore
+    wallet.switchChain = vi.fn().mockImplementation((chain) => {
+      manager.activeWalletChainStore.setValue(chain);
+    });
+
+    await manager.switchActiveWalletChain(newChain);
+
+    expect(wallet.switchChain).toHaveBeenCalledWith(newChain);
+  });
+
+  it("should define chains", async () => {
+    const manager = createConnectionManager(storage);
+    await manager.handleConnection(wallet, { client });
+
+    const chains = [
+      { id: 1, name: "Chain 1", rpc: "https://rpc1.example.com" },
+      { id: 2, name: "Chain 2", rpc: "https://rpc2.example.com" },
+    ];
+    manager.defineChains(chains);
+
+    const activeWalletChain = manager.activeWalletChainStore.getValue();
+    expect(activeWalletChain).toBeDefined();
+  });
+
+  it("should remove connected wallet", () => {
+    const manager = createConnectionManager(storage);
+
+    manager.addConnectedWallet(wallet);
+    expect(manager.connectedWallets.getValue()).toContain(wallet);
+
+    manager.removeConnectedWallet(wallet);
+    expect(manager.connectedWallets.getValue()).not.toContain(wallet);
+  });
 });