From 6e84865d8ba830b2b9a71075172177df8fa3842c Mon Sep 17 00:00:00 2001
From: "martin.stypinski" <martin.stypinski@caffeine.ai>
Date: Fri, 19 Dec 2025 11:46:17 +0100
Subject: [PATCH 1/3] Added function for customizable
 AWAIT_INGRESS_STATUS_ROUNDS.

---
 packages/pic/src/pocket-ic-client.ts | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/packages/pic/src/pocket-ic-client.ts b/packages/pic/src/pocket-ic-client.ts
index ab62328..40ae8e9 100644
--- a/packages/pic/src/pocket-ic-client.ts
+++ b/packages/pic/src/pocket-ic-client.ts
@@ -350,8 +350,9 @@ export class PocketIcClient {
     return decodeIngressStatusResponse(res);
   }
 
-  public async awaitCall(
+  public async awaitCallWithSpecifiedRounds(
     req: AwaitCanisterCallRequest,
+    rounds: number,
   ): Promise<AwaitCanisterCallResponse> {
     this.assertInstanceNotDeleted();
     const encodedReq = {
@@ -361,7 +362,7 @@ export class PocketIcClient {
       caller: undefined,
     };
 
-    for (let i = 0; i < AWAIT_INGRESS_STATUS_ROUNDS; i++) {
+    for (let i = 0; i < rounds; i++) {
       await this.tick();
       const result = await this.ingressStatus(encodedReq);
       if (isNotNil(result)) {
@@ -370,10 +371,16 @@ export class PocketIcClient {
     }
 
     throw new Error(
-      `PocketIC did not complete the update call within ${AWAIT_INGRESS_STATUS_ROUNDS} rounds`,
+      `PocketIC did not complete the update call within ${rounds} rounds`,
     );
   }
 
+  public async awaitCall(
+    req: AwaitCanisterCallRequest,
+  ): Promise<AwaitCanisterCallResponse> {
+    return this.awaitCallWithSpecifiedRounds(req, AWAIT_INGRESS_STATUS_ROUNDS);
+  }
+
   private async post<B, R extends {}>(
     endpoint: string,
     body?: B,

From c982812b664a74972cfc2d66bb33b75903584ce7 Mon Sep 17 00:00:00 2001
From: Martin Stypinski <martin.stypinski@dfinity.org>
Date: Fri, 19 Dec 2025 15:16:15 +0100
Subject: [PATCH 2/3] Update packages/pic/src/pocket-ic-client.ts

Co-authored-by: Luca Bertelli <luca.berte98@hotmail.it>
---
 packages/pic/src/pocket-ic-client.ts | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/packages/pic/src/pocket-ic-client.ts b/packages/pic/src/pocket-ic-client.ts
index 40ae8e9..5fc19a2 100644
--- a/packages/pic/src/pocket-ic-client.ts
+++ b/packages/pic/src/pocket-ic-client.ts
@@ -350,7 +350,7 @@ export class PocketIcClient {
     return decodeIngressStatusResponse(res);
   }
 
-  public async awaitCallWithSpecifiedRounds(
+  public async awaitCallWithRounds(
     req: AwaitCanisterCallRequest,
     rounds: number,
   ): Promise<AwaitCanisterCallResponse> {

From 51ab7a77972062f2f9ffedc3f62b0636834449d1 Mon Sep 17 00:00:00 2001
From: Martin Stypinski <martin.stypinski@dfinity.org>
Date: Fri, 19 Dec 2025 15:27:30 +0100
Subject: [PATCH 3/3] Refactor awaitCall method to use awaitCallWithRounds

---
 packages/pic/src/pocket-ic-client.ts | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/packages/pic/src/pocket-ic-client.ts b/packages/pic/src/pocket-ic-client.ts
index 5fc19a2..33fe283 100644
--- a/packages/pic/src/pocket-ic-client.ts
+++ b/packages/pic/src/pocket-ic-client.ts
@@ -378,7 +378,7 @@ export class PocketIcClient {
   public async awaitCall(
     req: AwaitCanisterCallRequest,
   ): Promise<AwaitCanisterCallResponse> {
-    return this.awaitCallWithSpecifiedRounds(req, AWAIT_INGRESS_STATUS_ROUNDS);
+    return this.awaitCallWithRounds(req, AWAIT_INGRESS_STATUS_ROUNDS);
   }
 
   private async post<B, R extends {}>(