Adding Wait on 202 of v3 calls

edjCase · Nov 5, 2024 · 6b197a2 · 6b197a2
1 parent 26fe26a
commit 6b197a2
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 7 deletions.
diff --git a/src/Agent/Agents/HttpAgent.cs b/src/Agent/Agents/HttpAgent.cs
@@ -91,12 +91,11 @@ public async Task<CandidArg> CallAsync(
 				this.v3CallSupported = false;
 				return await this.CallAsynchronousAndWaitAsync(canisterId, method, arg, effectiveCanisterId, cancellationToken);
 			}
-			// TODO
-			// if (httpResponse.StatusCode == System.Net.HttpStatusCode.Accepted){
-			// 	// If request takes too long, then it will return 202 Accepted
-			// 	// and polling is required
-			// 	return await this.WaitAsync();
-			// }
+			if (httpResponse.StatusCode == System.Net.HttpStatusCode.Accepted)
+			{
+				// If request takes too long, then it will return 202 Accepted and polling is required
+				return await this.WaitForRequestAsync(canisterId, requestId, cancellationToken);
+			}
 			await httpResponse.ThrowIfErrorAsync();
 
 			byte[] cborBytes = await httpResponse.GetContentAsync();

diff --git a/src/Agent/Agents/IAgent.cs b/src/Agent/Agents/IAgent.cs
@@ -135,7 +135,7 @@ public static async Task<CandidArg> CallAsynchronousAndWaitAsync(
 			Principal? effectiveCanisterId = null,
 			CancellationToken? cancellationToken = null)
 		{
-			RequestId id = await agent.CallAsync(canisterId, method, arg, effectiveCanisterId);
+			RequestId id = await agent.CallAsynchronousAsync(canisterId, method, arg, effectiveCanisterId);
 			return await agent.WaitForRequestAsync(canisterId, id, cancellationToken);
 		}