From 8d3d35f9e51afe30376eb4440d23ee439e4db9a9 Mon Sep 17 00:00:00 2001
From: Manank Patni <manank321@gmail.com>
Date: Mon, 7 Aug 2023 22:07:39 +0530
Subject: [PATCH] Fix network switch with inmemory signer

Signed-off-by: Manank Patni <manank321@gmail.com>
---
 src/services/beacon/hooks/useTezos.ts | 20 +++++++++++++++-----
 1 file changed, 15 insertions(+), 5 deletions(-)

diff --git a/src/services/beacon/hooks/useTezos.ts b/src/services/beacon/hooks/useTezos.ts
index b1ea2e7e..00161881 100644
--- a/src/services/beacon/hooks/useTezos.ts
+++ b/src/services/beacon/hooks/useTezos.ts
@@ -98,17 +98,27 @@ export const useTezos = (): WalletConnectReturn => {
           }
         })
       } else {
-        const { wallet } = await connectWithBeacon(newNetwork)
-        newTezos.setProvider({ wallet })
-        const newAccount = await newTezos.wallet.pkh()
+        let wallet, account
+
+        if (getEnv(EnvKey.REACT_APP_IS_NOT_TESTING) === "true") {
+          const { wallet: beaconWallet } = await connectWithBeacon(network)
+          wallet = beaconWallet
+          newTezos.setProvider({ wallet })
+          account = await newTezos.wallet.pkh()
+        } else {
+          const signer = await InMemorySigner.fromSecretKey(ALICE_PRIV_KEY)
+          wallet = signer
+          account = await signer.publicKeyHash()
+          newTezos.setProvider({ signer })
+        }
 
         dispatch({
           type: TezosActionType.UPDATE_TEZOS,
           payload: {
             network: newNetwork,
             tezos: newTezos,
-            account: newAccount,
-            wallet
+            account,
+            wallet: wallet as BeaconWallet
           }
         })
       }