Dispose the cancellation tokens in WireMockServerLifecycleHook

Author: Zguy

src/WireMock.Net.Aspire/WireMockServerLifecycleHook.cs

@@ -10,11 +10,12 @@ internal class WireMockServerLifecycleHook(ILoggerFactory loggerFactory) : IDist
 {
     private readonly CancellationTokenSource _shutdownCts = new();
 
+    private CancellationTokenSource? _linkedCts;
     private Task? _mappingTask;
 
-    public async Task AfterEndpointsAllocatedAsync(DistributedApplicationModel appModel, CancellationToken cancellationToken = default)
+    public Task AfterEndpointsAllocatedAsync(DistributedApplicationModel appModel, CancellationToken cancellationToken = default)
     {
-        var cts = CancellationTokenSource.CreateLinkedTokenSource(_shutdownCts.Token, cancellationToken);
+        _linkedCts = CancellationTokenSource.CreateLinkedTokenSource(_shutdownCts.Token, cancellationToken);
 
         _mappingTask = Task.Run(async () =>
         {
@@ -29,19 +30,27 @@ public async Task AfterEndpointsAllocatedAsync(DistributedApplicationModel appMo
                 var endpoint = wireMockServerResource.GetEndpoint();
                 System.Diagnostics.Debug.Assert(endpoint.IsAllocated);
 
-                await wireMockServerResource.WaitForHealthAsync(cts.Token);
+                await wireMockServerResource.WaitForHealthAsync(_linkedCts.Token);
 
-                await wireMockServerResource.CallApiMappingBuilderActionAsync(cts.Token);
+                await wireMockServerResource.CallApiMappingBuilderActionAsync(_linkedCts.Token);
 
-                wireMockServerResource.StartWatchingStaticMappings(cts.Token);
+                wireMockServerResource.StartWatchingStaticMappings(_linkedCts.Token);
             }
-        }, cts.Token);
+        }, _linkedCts.Token);
+
+        return Task.CompletedTask;
     }
 
     public async ValueTask DisposeAsync()
     {
         await _shutdownCts.CancelAsync();
+
+        _linkedCts?.Dispose();
+        _shutdownCts.Dispose();
+
         if (_mappingTask is not null)
+        {
             await _mappingTask;
+        }
     }
 }