Verify connection shutdown cancellation propagation

Author: danielmarbach

projects/Test/Integration/TestConnectionShutdown.cs

@@ -162,6 +162,29 @@ public async Task TestShutdownSignalPropagationToChannels()
             await WaitAsync(tcs, TimeSpan.FromSeconds(3), "channel shutdown");
         }
 
+        [Fact]
+        public async Task TestShutdownCancellation()
+        {
+            using var cts = new CancellationTokenSource(TimeSpan.FromSeconds(1));
+            var tcs = new TaskCompletionSource<bool>(TaskCreationOptions.RunContinuationsAsynchronously);
+
+            _conn.ConnectionShutdownAsync += async (channel, args) =>
+            {
+                try
+                {
+                    await Task.Delay(TimeSpan.FromMinutes(1), args.CancellationToken);
+                }
+                catch (OperationCanceledException)
+                {
+                    tcs.SetResult(true);
+                }
+            };
+
+            await _conn.CloseAsync(cancellationToken: cts.Token);
+
+            await WaitAsync(tcs, TimeSpan.FromSeconds(3), "connection shutdown");
+        }
+
         [Fact]
         public async Task TestShutdownSignalPropagationWithCancellationToChannels()
         {