Fix test cancellation

nats-io · Jan 25, 2025 · 5ed2adc · 5ed2adc
1 parent 5e277b8
commit 5ed2adc
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/tests/NATS.Client.JetStream.Tests/ErrorHandlerTest.cs b/tests/NATS.Client.JetStream.Tests/ErrorHandlerTest.cs
@@ -310,9 +310,16 @@ public async Task Ordered_consumer_consume_handling()
 
         await Retry.Until("timed out", () => Volatile.Read(ref timeoutNotifications) > 0, timeout: TimeSpan.FromSeconds(30));
         consumeCts.Cancel();
-        await consume;
 
         Assert.True(Volatile.Read(ref timeoutNotifications) > 0);
+
+        try
+        {
+            await consume;
+        }
+        catch (OperationCanceledException)
+        {
+        }
     }
 
     [Fact]