Add shorter timeout on Client_CanResumeUnsolicitedMessageStream_AfterDisconnection

Author: MackinnonBuck

tests/ModelContextProtocol.AspNetCore.Tests/ResumabilityIntegrationTests.cs

@@ -279,6 +279,7 @@ public async Task Client_CanResumePostResponseStream_AfterDisconnection()
     [Fact]
     public async Task Client_CanResumeUnsolicitedMessageStream_AfterDisconnection()
     {
+        var timeout = TimeSpan.FromSeconds(10);
         using var faultingStreamHandler = new FaultingStreamHandler()
         {
             InnerHandler = SocketsHttpHandler,
@@ -304,12 +305,12 @@ public async Task Client_CanResumeUnsolicitedMessageStream_AfterDisconnection()
         await using var client = await ConnectClientAsync();
 
         // Get the server instance
-        var server = await serverTcs.Task.WaitAsync(TestContext.Current.CancellationToken);
+        var server = await serverTcs.Task.WaitAsync(timeout, TestContext.Current.CancellationToken);
 
         // Set up notification tracking with unique messages
-        var clientReceivedInitialNotificationTcs = new TaskCompletionSource();
-        var clientReceivedReplayedNotificationTcs = new TaskCompletionSource();
-        var clientReceivedReconnectNotificationTcs = new TaskCompletionSource();
+        var clientReceivedInitialNotificationTcs = new TaskCompletionSource(TaskCreationOptions.RunContinuationsAsynchronously);
+        var clientReceivedReplayedNotificationTcs = new TaskCompletionSource(TaskCreationOptions.RunContinuationsAsynchronously);
+        var clientReceivedReconnectNotificationTcs = new TaskCompletionSource(TaskCreationOptions.RunContinuationsAsynchronously);
 
         const string CustomNotificationMethod = "test/custom_notification";
         const string InitialMessage = "Initial notification";
@@ -347,7 +348,7 @@ public async Task Client_CanResumeUnsolicitedMessageStream_AfterDisconnection()
         await server.SendNotificationAsync(CustomNotificationMethod, new JsonObject { ["message"] = InitialMessage }, cancellationToken: TestContext.Current.CancellationToken);
 
         // Wait for client to receive the first notification
-        await clientReceivedInitialNotificationTcs.Task.WaitAsync(TestContext.Current.CancellationToken);
+        await clientReceivedInitialNotificationTcs.Task.WaitAsync(timeout, TestContext.Current.CancellationToken);
 
         // Fault the unsolicited message stream (GET SSE)
         var reconnectAttempt = await faultingStreamHandler.TriggerFaultAsync(TestContext.Current.CancellationToken);
@@ -359,13 +360,13 @@ public async Task Client_CanResumeUnsolicitedMessageStream_AfterDisconnection()
         reconnectAttempt.Continue();
 
         // Wait for client to receive the notification via replay
-        await clientReceivedReplayedNotificationTcs.Task.WaitAsync(TestContext.Current.CancellationToken);
+        await clientReceivedReplayedNotificationTcs.Task.WaitAsync(timeout, TestContext.Current.CancellationToken);
 
         // Send a final notification while the client has reconnected - this should be handled by the transport
         await server.SendNotificationAsync(CustomNotificationMethod, new JsonObject { ["message"] = ReconnectMessage }, cancellationToken: TestContext.Current.CancellationToken);
 
         // Wait for the client to receive the final notification
-        await clientReceivedReconnectNotificationTcs.Task.WaitAsync(TestContext.Current.CancellationToken);
+        await clientReceivedReconnectNotificationTcs.Task.WaitAsync(timeout, TestContext.Current.CancellationToken);
 
         // Assert each notification was received exactly once
         Assert.Equal(1, initialNotificationReceivedCount);