Fix undisposed resources

Author: ericstj

tests/ModelContextProtocol.Tests/Configuration/McpServerBuilderExtensionsTransportsTests.cs

@@ -9,15 +9,15 @@ namespace ModelContextProtocol.Tests.Configuration;
 public class McpServerBuilderExtensionsTransportsTests
 {
     [Fact]
-    public void WithStdioServerTransport_Sets_Transport()
+    public async Task WithStdioServerTransport_Sets_Transport()
     {
         var services = new ServiceCollection();
         services.AddMcpServer().WithStdioServerTransport();
 
         var transportServiceType = services.FirstOrDefault(s => s.ServiceType == typeof(ITransport));
         Assert.NotNull(transportServiceType);
 
-        var serviceProvider = services.BuildServiceProvider();
+        await using var serviceProvider = services.BuildServiceProvider();
         Assert.IsType<StdioServerTransport>(serviceProvider.GetRequiredService<ITransport>());
     }
 

tests/ModelContextProtocol.Tests/Server/McpServerLoggingLevelTests.cs

@@ -15,29 +15,29 @@ public McpServerLoggingLevelTests()
     }
 
     [Fact]
-    public void CanCreateServerWithLoggingLevelHandler()
+    public async Task CanCreateServerWithLoggingLevelHandler()
     {
         var services = new ServiceCollection();
 
         services.AddMcpServer()
             .WithStdioServerTransport()
             .WithSetLoggingLevelHandler(async (ctx, ct) => new EmptyResult());
 
-        var provider = services.BuildServiceProvider();
+        await using var provider = services.BuildServiceProvider();
 
         provider.GetRequiredService<McpServer>();
     }
 
     [Fact]
-    public void AddingLoggingLevelHandlerSetsLoggingCapability()
+    public async Task AddingLoggingLevelHandlerSetsLoggingCapability()
     {
         var services = new ServiceCollection();
 
         services.AddMcpServer()
             .WithStdioServerTransport()
             .WithSetLoggingLevelHandler(async (ctx, ct) => new EmptyResult());
 
-        var provider = services.BuildServiceProvider();
+        await using var provider = services.BuildServiceProvider();
 
         var server = provider.GetRequiredService<McpServer>();
 
@@ -46,12 +46,12 @@ public void AddingLoggingLevelHandlerSetsLoggingCapability()
     }
 
     [Fact]
-    public void ServerWithoutCallingLoggingLevelHandlerDoesNotSetLoggingCapability()
+    public async Task ServerWithoutCallingLoggingLevelHandlerDoesNotSetLoggingCapability()
     {
         var services = new ServiceCollection();
         services.AddMcpServer()
             .WithStdioServerTransport();
-        var provider = services.BuildServiceProvider();
+        await using var provider = services.BuildServiceProvider();
         var server = provider.GetRequiredService<McpServer>();
         Assert.Null(server.ServerOptions.Capabilities?.Logging);
     }

tests/ModelContextProtocol.Tests/Server/McpServerResourceTests.cs

@@ -29,7 +29,7 @@ public McpServerResourceTests()
     }
 
     [Fact]
-    public void CanCreateServerWithResource()
+    public async Task CanCreateServerWithResource()
     {
         var services = new ServiceCollection();
 
@@ -58,14 +58,14 @@ public void CanCreateServerWithResource()
                 };
             });
 
-        var provider = services.BuildServiceProvider();
+        await using var provider = services.BuildServiceProvider();
 
         provider.GetRequiredService<McpServer>();
     }
 
 
     [Fact]
-    public void CanCreateServerWithResourceTemplates()
+    public async Task CanCreateServerWithResourceTemplates()
     {
         var services = new ServiceCollection();
 
@@ -94,13 +94,13 @@ public void CanCreateServerWithResourceTemplates()
                 };
             });
 
-        var provider = services.BuildServiceProvider();
+        await using var provider = services.BuildServiceProvider();
 
         provider.GetRequiredService<McpServer>();
     }
 
     [Fact]
-    public void CreatingReadHandlerWithNoListHandlerSucceeds()
+    public async Task CreatingReadHandlerWithNoListHandlerSucceeds()
     {
         var services = new ServiceCollection();
         services.AddMcpServer()
@@ -117,7 +117,7 @@ public void CreatingReadHandlerWithNoListHandlerSucceeds()
                     }]
                 };
             });
-        var sp = services.BuildServiceProvider();
+        await using var sp = services.BuildServiceProvider();
 
         sp.GetRequiredService<McpServer>();
     }