Address PR feedback

Author: halter73

src/ModelContextProtocol.AspNetCore/StreamableHttpHandler.cs

@@ -110,8 +110,8 @@ await WriteJsonRpcErrorAsync(context,
             // We should flush headers to indicate a 200 success quickly, because the initialization response
             // will be sent in response to a different POST request. It might be a while before we send a message
             // over this response body.
-            await context.Response.Body.FlushAsync(sseCts.Token);
-            await session.Transport.HandleGetRequestAsync(context.Response.Body, sseCts.Token);
+            await context.Response.Body.FlushAsync(cancellationToken);
+            await session.Transport.HandleGetRequestAsync(context.Response.Body, cancellationToken);
         }
         catch (OperationCanceledException) when (cancellationToken.IsCancellationRequested)
         {

tests/ModelContextProtocol.AspNetCore.Tests/Utils/KestrelInMemoryTransport.cs

@@ -41,12 +41,9 @@ private sealed class KestrelInMemoryListener(EndPoint endpoint, Channel<Connecti
 
         public async ValueTask<ConnectionContext?> AcceptAsync(CancellationToken cancellationToken = default)
         {
-            while (await acceptQueue.Reader.WaitToReadAsync(cancellationToken))
+            await foreach (var item in acceptQueue.Reader.ReadAllAsync(cancellationToken))
             {
-                while (acceptQueue.Reader.TryRead(out var item))
-                {
-                    return item;
-                }
+                return item;
             }
 
             return null;