* Unblock so that CloseAsync can succeed.

rabbitmq · Sep 25, 2024 · 748ec8c · 748ec8c
1 parent 8b54bf4
commit 748ec8c
Showing 1 changed file with 17 additions and 9 deletions.
diff --git a/projects/Test/SequentialIntegration/TestConnectionRecovery.cs b/projects/Test/SequentialIntegration/TestConnectionRecovery.cs
@@ -145,16 +145,24 @@ Task _channel_ChannelShutdownAsync(object sender, ShutdownEventArgs e)
         [Fact]
         public async Task TestBlockedListenersRecovery()
         {
-            var tcs = new TaskCompletionSource<bool>(TaskCreationOptions.RunContinuationsAsynchronously);
-            _conn.ConnectionBlockedAsync += (c, reason) =>
+            try
             {
-                tcs.SetResult(true);
-                return Task.CompletedTask;
-            };
-            await CloseAndWaitForRecoveryAsync();
-            await CloseAndWaitForRecoveryAsync();
-            await BlockAndPublishAsync();
-            await WaitAsync(tcs, "connection blocked");
+                var tcs = new TaskCompletionSource<bool>(TaskCreationOptions.RunContinuationsAsynchronously);
+                _conn.ConnectionBlockedAsync += (c, reason) =>
+                {
+                    tcs.SetResult(true);
+                    return Task.CompletedTask;
+                };
+                await CloseAndWaitForRecoveryAsync();
+                await CloseAndWaitForRecoveryAsync();
+                await BlockAndPublishAsync();
+                await WaitAsync(tcs, "connection blocked");
+            }
+            finally
+            {
+                // NOTE: must unblock so that close succeeeds on test tear-down
+                await UnblockAsync();
+            }
         }
 
         [Fact]