Update Thirdweb.Utils.Tests.cs

thirdweb-dev · Apr 8, 2024 · a61f0f6 · a61f0f6
1 parent 58e0723
commit a61f0f6
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/Thirdweb.Tests/Thirdweb.Utils.Tests.cs b/Thirdweb.Tests/Thirdweb.Utils.Tests.cs
@@ -87,6 +87,12 @@ public async Task GetTransactionReceipt_CancellationToken()
         var infiniteReceipt = await Assert.ThrowsAsync<TaskCanceledException>(async () => await Utils.GetTransactionReceipt(client, chainId, infiniteTxHash, cts.Token));
         Assert.Equal("A task was canceled.", infiniteReceipt.Message);
 
+        cts = new CancellationTokenSource();
+        var infiniteReceipt2 = Assert.ThrowsAsync<TaskCanceledException>(() => Utils.GetTransactionReceipt(client, chainId, infiniteTxHash, cts.Token));
+        await Task.Delay(2000);
+        cts.Cancel();
+        Assert.Equal("A task was canceled.", (await infiniteReceipt2).Message);
+
         var aaReceipt2 = await Utils.GetTransactionReceipt(client, chainId, aaTxHash, CancellationToken.None);
         Assert.NotNull(aaReceipt2);
     }