fix: add retry for getLastConnectedWallet test

starknet-io · Oct 15, 2024 · 037073a · 037073a
1 parent 9c59bdb
commit 037073a
Showing 1 changed file with 18 additions and 14 deletions.
diff --git a/packages/core/src/__test__/storage.test.ts b/packages/core/src/__test__/storage.test.ts
@@ -88,21 +88,25 @@ describe("getLastConnectedWallet()", () => {
     const lastConnectedWallet = await sn.getLastConnectedWallet()
     expect(lastConnectedWallet?.id).toBe(BraavosMock.id)
   })
-  it("should set the last connected wallet when enabled", async () => {
-    const sn = getWallet({
-      "starknet-argentX": makeConnected(true)(ArgentXMock),
-      "starknet-braavos": makeConnected(true)(BraavosMock),
-      starknet_okxwallet: makeConnected(true)(OKXMock),
-      starknet_keplr: makePreAuthorized(true)(KeplrMock),
-    })
-    const lastConnectedWallet = await sn.getLastConnectedWallet()
-    expect(lastConnectedWallet).toBe(null)
+  it(
+    "should set the last connected wallet when enabled",
+    { retry: 5 },
+    async () => {
+      const sn = getWallet({
+        "starknet-argentX": makeConnected(true)(ArgentXMock),
+        "starknet-braavos": makeConnected(true)(BraavosMock),
+        starknet_okxwallet: makeConnected(true)(OKXMock),
+        starknet_keplr: makePreAuthorized(true)(KeplrMock),
+      })
+      const lastConnectedWallet = await sn.getLastConnectedWallet()
+      expect(lastConnectedWallet).toBe(null)
 
-    const [newLastConnectedWallet] = await sn.getAvailableWallets()
-    await sn.enable(newLastConnectedWallet)
-    const lastConnectedWalletAfterEnable = await sn.getLastConnectedWallet()
-    expect(lastConnectedWalletAfterEnable).toBe(newLastConnectedWallet)
-  })
+      const [newLastConnectedWallet] = await sn.getAvailableWallets()
+      await sn.enable(newLastConnectedWallet)
+      const lastConnectedWalletAfterEnable = await sn.getLastConnectedWallet()
+      expect(lastConnectedWalletAfterEnable).toBe(newLastConnectedWallet)
+    },
+  )
   it("enabling fails", async () => {
     const sn = getWallet({
       "starknet-argentX": makeConnected(false)(ArgentXMock),