test: adapt for agent-js last version

dfinity · Jul 24, 2023 · b451cda · b451cda
1 parent 616912b
commit b451cda
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/packages/nns/src/governance.canister.spec.ts b/packages/nns/src/governance.canister.spec.ts
@@ -2,6 +2,7 @@ import {
   ActorSubclass,
   AnonymousIdentity,
   polling,
+  SubmitResponse,
   type Agent,
   type RequestId,
 } from "@dfinity/agent";
@@ -54,7 +55,7 @@ describe("GovernanceCanister", () => {
       status: 13,
       statusText: "good",
     },
-  };
+  } as SubmitResponse;
   const newSpawnNeuronId = new PbNeuronId();
   newSpawnNeuronId.setId("1234");
   const spawnResponse = new PbManageNeuronResponse.SpawnResponse();
@@ -1335,7 +1336,7 @@ describe("GovernanceCanister", () => {
           status: 13,
           statusText: "good",
         },
-      };
+      } as SubmitResponse;
       agent.call.mockResolvedValue(response);
 
       const governance = GovernanceCanister.create({

diff --git a/packages/sns/src/sns.wrapper.spec.ts b/packages/sns/src/sns.wrapper.spec.ts
@@ -52,6 +52,7 @@ describe("SnsWrapper", () => {
 
   const mockIndexCanister = mock<IcrcIndexCanister>();
   mockIndexCanister.getTransactions.mockResolvedValue({
+    balance: 2n,
     transactions: [],
     oldest_tx_id: [BigInt(2)],
   });