Update notification handlers to include cancellation token in McpClient and McpServer tests

Author: Tyler-R-Kendrick

tests/ModelContextProtocol.Tests/Client/McpClientExtensionsTests.cs

@@ -386,7 +386,7 @@ public async Task Can_Handle_Notify_Cancel()
         var token = TestContext.Current.CancellationToken;
         TaskCompletionSource<JsonRpcNotification> clientReceived = new();
         await using var client = await CreateMcpClientForServer(
-            options: CreateClientOptions([new(NotificationMethods.CancelledNotification, notification =>
+            options: CreateClientOptions([new(NotificationMethods.CancelledNotification, (notification, cancellationToken) =>
             {
                 clientReceived.TrySetResult(notification);
                 return clientReceived.Task;
@@ -422,7 +422,7 @@ public async Task Should_Not_Intercept_Sent_Notifications()
         var token = TestContext.Current.CancellationToken;
         TaskCompletionSource<JsonRpcNotification> clientReceived = new();
         await using var client = await CreateMcpClientForServer(
-            options: CreateClientOptions([new(NotificationMethods.CancelledNotification, notification =>
+            options: CreateClientOptions([new(NotificationMethods.CancelledNotification, (notification, cancellationToken) =>
             {
                 var exception = new InvalidOperationException("Should not intercept sent notifications");
                 clientReceived.TrySetException(exception);
@@ -452,9 +452,9 @@ public async Task Can_Notify_Cancel()
         var token = TestContext.Current.CancellationToken;
         TaskCompletionSource clientReceived = new();
         await using var client = await CreateMcpClientForServer(
-            options: CreateClientOptions(new Dictionary<string, Func<JsonRpcNotification, Task>>()
+            options: CreateClientOptions(new Dictionary<string, Func<JsonRpcNotification, CancellationToken, Task>>()
             {
-                [NotificationMethods.CancelledNotification] = notification =>
+                [NotificationMethods.CancelledNotification] = (notification, cancellationToken) =>
                 {
                     InvalidOperationException exception = new("Should not intercept sent notifications");
                     clientReceived.TrySetException(exception);
@@ -480,7 +480,7 @@ await Assert.ThrowsAsync<TimeoutException>(
     }
 
     private static McpClientOptions CreateClientOptions(
-        IEnumerable<KeyValuePair<string, Func<JsonRpcNotification, Task>>>? notificationHandlers = null)
+        IEnumerable<KeyValuePair<string, Func<JsonRpcNotification, CancellationToken, Task>>>? notificationHandlers = null)
         => new()
         {
             Capabilities = new()

tests/ModelContextProtocol.Tests/Server/McpServerTests.cs

@@ -704,7 +704,7 @@ public async Task NotifyCancel_Should_Be_Handled()
         };
         var options = CreateOptions(new()
         {
-            NotificationHandlers = [new(NotificationMethods.CancelledNotification, notification =>
+            NotificationHandlers = [new(NotificationMethods.CancelledNotification, (notification, cancellationToken) =>
             {
                 InvalidOperationException exception = new("The sender of a notification shouldn't handle the notification.");
                 notificationIntercepted.TrySetException(exception);