+ code coverage for ClientOptions/Capabilities code

Author: Esteban Solano

tests/ModelContextProtocol.Tests/Client/McpClientFactoryTests.cs

@@ -1,9 +1,11 @@
 using System.Threading.Channels;
+using Microsoft.Extensions.Logging;
 using ModelContextProtocol.Client;
 using ModelContextProtocol.Configuration;
 using ModelContextProtocol.Protocol.Messages;
 using ModelContextProtocol.Protocol.Transport;
 using ModelContextProtocol.Protocol.Types;
+using Moq;
 
 namespace ModelContextProtocol.Tests.Client;
 
@@ -187,7 +189,68 @@ public async Task McpFactory_WithInvalidTransportOptions_ThrowsFormatException(s
         await Assert.ThrowsAsync<ArgumentException>(() => McpClientFactory.CreateAsync(config, _defaultOptions, cancellationToken: TestContext.Current.CancellationToken));
     }
 
-    private sealed class NopTransport : IClientTransport
+    [Theory]
+    [InlineData(typeof(NopTransport))]
+    [InlineData(typeof(FailureTransport))]
+    public async Task CreateAsync_WithCapabilities(Type transportType)
+    {
+        // Arrange
+        var serverConfig = new McpServerConfig
+        {
+            Id = "TestServer",
+            Name = "TestServer",
+            TransportType = "stdio",
+            Location = "test-location"
+        };
+
+        var clientOptions = new McpClientOptions
+        {
+            ClientInfo = new Implementation 
+            {
+                Name = "TestClient", 
+                Version = "1.0.0.0"
+            },
+            Capabilities = new ClientCapabilities
+            {
+                Sampling = new SamplingCapability
+                {
+                    SamplingHandler = (c, t) => Task.FromResult(
+                        new CreateMessageResult { 
+                            Content = new Content { Text = "result" }, 
+                            Model = "test-model", 
+                            Role = "test-role", 
+                            StopReason = "endTurn" 
+                    }),
+                },
+                Roots = new RootsCapability
+                {
+                    ListChanged = true,
+                    RootsHandler = (t, r) => Task.FromResult(new ListRootsResult { Roots = [] }),
+                }
+            }
+        };
+
+        var clientTransport = (IClientTransport?)Activator.CreateInstance(transportType);
+        IMcpClient? client = null;
+
+        var actionTask = McpClientFactory.CreateAsync(serverConfig, clientOptions, (config, logger) => clientTransport ?? new NopTransport(), new Mock<ILoggerFactory>().Object, CancellationToken.None);
+
+        // Act
+        if (clientTransport is FailureTransport)
+        {
+            var exception = await Assert.ThrowsAsync<InvalidOperationException>(async() => await actionTask);
+            Assert.Equal(FailureTransport.ExpectedMessage, exception.Message);
+        }
+        else
+        {
+            client = await actionTask;
+
+            // Assert
+            Assert.NotNull(client);
+        }        
+    }
+
+    private class NopTransport : IClientTransport
     {
         private readonly Channel<IJsonRpcMessage> _channel = Channel.CreateUnbounded<IJsonRpcMessage>();
 
@@ -200,7 +263,7 @@ public Task ConnectAsync(CancellationToken cancellationToken = default) =>
 
         public ValueTask DisposeAsync() => default;
 
-        public Task SendMessageAsync(IJsonRpcMessage message, CancellationToken cancellationToken = default)
+        public virtual Task SendMessageAsync(IJsonRpcMessage message, CancellationToken cancellationToken = default)
         {
             switch (message)
             {
@@ -225,4 +288,14 @@ public Task SendMessageAsync(IJsonRpcMessage message, CancellationToken cancella
             return Task.CompletedTask;
         }
     }
+
+    private sealed class FailureTransport : NopTransport 
+    {
+        public const string ExpectedMessage = "Something failed";
+
+        public override Task SendMessageAsync(IJsonRpcMessage message, CancellationToken cancellationToken = default)
+        {
+            throw new InvalidOperationException(ExpectedMessage);
+        }
+    }
 }