test that there is no data change on error

dfinity · Sep 3, 2024 · 731f58c · 731f58c
1 parent 7d0f053
commit 731f58c
Showing 1 changed file with 30 additions and 5 deletions.
diff --git a/frontend/src/tests/lib/pages/IcrcWallet.spec.ts b/frontend/src/tests/lib/pages/IcrcWallet.spec.ts
@@ -607,12 +607,24 @@ describe("IcrcWallet", () => {
 
     it("should stay on the same page when removal is unsuccessful", async () => {
       vi.spyOn(console, "error").mockReturnValue();
-      vi.spyOn(importedTokensApi, "setImportedTokens").mockRejectedValue(
-        new Error()
+      const spyOnSetImportedTokens = vi
+        .spyOn(importedTokensApi, "setImportedTokens")
+        .mockRejectedValue(new Error());
+      const spyOnGetImportedTokens = vi.spyOn(
+        importedTokensApi,
+        "getImportedTokens"
       );
-      vi.spyOn(importedTokensApi, "getImportedTokens").mockResolvedValue({
-        imported_tokens: [],
-      });
+
+      expect(get(importedTokensStore).importedTokens).toEqual([
+        {
+          ledgerCanisterId,
+          indexCanisterId: undefined,
+        },
+        {
+          ledgerCanisterId: ledgerCanisterId2,
+          indexCanisterId: undefined,
+        },
+      ]);
 
       const po = await renderWallet({});
       const morePopoverPo = po.getWalletMorePopoverPo();
@@ -624,8 +636,21 @@ describe("IcrcWallet", () => {
       await runResolvedPromises();
       await confirmationPo.clickYes();
       await runResolvedPromises();
+      expect(spyOnSetImportedTokens).toBeCalledTimes(1);
       // should stay on wallet page
       expect(get(pageStore).path).toEqual(AppPath.Wallet);
+      // without data change
+      expect(spyOnGetImportedTokens).toBeCalledTimes(0);
+      expect(get(importedTokensStore).importedTokens).toEqual([
+        {
+          ledgerCanisterId,
+          indexCanisterId: undefined,
+        },
+        {
+          ledgerCanisterId: ledgerCanisterId2,
+          indexCanisterId: undefined,
+        },
+      ]);
     });
   });
 });