?

src/WireMock.Net.Minimal/Owin/AspNetCoreSelfHost.cs

@@ -64,6 +64,7 @@ internal partial class AspNetCoreSelfHost
                 services.AddSingleton<IOwinRequestMapper, OwinRequestMapper>();
                 services.AddSingleton<IOwinResponseMapper, OwinResponseMapper>();
                 services.AddSingleton<IGuidUtils, GuidUtils>();
+                services.AddSingleton<IDateTimeUtils, DateTimeUtils>();
                 services.AddSingleton<LogEntryMapper>();
                 services.AddSingleton<IWireMockMiddlewareLogger, WireMockMiddlewareLogger>();
 

src/WireMock.Net.Minimal/Owin/WireMockMiddleware.cs

@@ -23,7 +23,8 @@ internal class WireMockMiddleware(
     IOwinResponseMapper responseMapper,
     IMappingMatcher mappingMatcher,
     IWireMockMiddlewareLogger logger,
-    IGuidUtils guidUtils
+    IGuidUtils guidUtils,
+    IDateTimeUtils dateTimeUtils
 )
 {
     private readonly object _lock = new();
@@ -47,6 +48,7 @@ internal class WireMockMiddleware(
         ctx.Items[nameof(IWireMockMiddlewareOptions)] = options;
         ctx.Items[nameof(IWireMockMiddlewareLogger)] = logger;
         ctx.Items[nameof(IGuidUtils)] = guidUtils;
+        ctx.Items[nameof(IDateTimeUtils)] = dateTimeUtils;
 
         var request = await requestMapper.MapAsync(ctx, options).ConfigureAwait(false);
 

src/WireMock.Net.Minimal/ResponseProviders/WebSocketHandledResponse.cs

@@ -9,9 +9,10 @@ namespace WireMock.ResponseProviders;
 /// </summary>
 internal class WebSocketHandledResponse : ResponseMessage
 {
-    public WebSocketHandledResponse()
+    public WebSocketHandledResponse(DateTime dateTime)
     {
         // 101 Switching Protocols
         StatusCode = (int)HttpStatusCode.SwitchingProtocols;
+        DateTime = dateTime;
     }
 }

src/WireMock.Net.Minimal/ResponseProviders/WebSocketResponseProvider.cs

@@ -16,7 +16,7 @@ using WireMock.WebSockets;
 
 namespace WireMock.ResponseProviders;
 
-internal class WebSocketResponseProvider(WebSocketBuilder builder) : IResponseProvider
+internal class WebSocketResponseProvider(WebSocketBuilder builder, IGuidUtils guidUtils, IDateTimeUtils dateTimeUtils) : IResponseProvider
 {
     public async Task<(IResponseMessage Message, IMapping? Mapping)> ProvideResponseAsync(
         IMapping mapping,
@@ -30,6 +30,16 @@ internal class WebSocketResponseProvider(WebSocketBuilder builder) : IResponsePr
             return (ResponseMessageBuilder.Create(HttpStatusCode.BadRequest, "Bad Request: Not a WebSocket upgrade request"), null);
         }
 
+        if (!context.Items.TryGetValue<IWireMockMiddlewareOptions>(nameof(IWireMockMiddlewareOptions), out var options))
+        {
+            throw new InvalidOperationException("IWireMockMiddlewareOptions not found in HttpContext.Items");
+        }
+
+        if (!context.Items.TryGetValue<IWireMockMiddlewareLogger>(nameof(IWireMockMiddlewareLogger), out var logger))
+        {
+            throw new InvalidOperationException("IWireMockMiddlewareLogger not found in HttpContext.Items");
+        }
+
         try
         {
             // Accept the WebSocket connection
@@ -45,21 +55,6 @@ internal class WebSocketResponseProvider(WebSocketBuilder builder) : IResponsePr
             var webSocket = await context.WebSockets.AcceptWebSocketAsync(builder.AcceptProtocol).ConfigureAwait(false);
 #endif
 
-            if (!context.Items.TryGetValue<IWireMockMiddlewareOptions>(nameof(IWireMockMiddlewareOptions), out var options))
-            {
-                throw new InvalidOperationException("IWireMockMiddlewareOptions not found in HttpContext.Items");
-            }
-
-            if (!context.Items.TryGetValue<IWireMockMiddlewareLogger>(nameof(IWireMockMiddlewareLogger), out var logger))
-            {
-                throw new InvalidOperationException("IWireMockMiddlewareLogger not found in HttpContext.Items");
-            }
-
-            if (!context.Items.TryGetValue<IGuidUtils>(nameof(IGuidUtils), out var guidUtils))
-            {
-                throw new InvalidOperationException("IGuidUtils not found in HttpContext.Items");
-            }
-
             // Get or create registry from options
             var registry = builder.IsBroadcast
                 ? options.WebSocketRegistries.GetOrAdd(mapping.Guid, _ => new WebSocketConnectionRegistry())
@@ -109,7 +104,7 @@ internal class WebSocketResponseProvider(WebSocketBuilder builder) : IResponsePr
             }
 
             // Return special marker to indicate WebSocket was handled
-            return (new WebSocketHandledResponse(), null);
+            return (new WebSocketHandledResponse(dateTimeUtils.UtcNow), null);
         }
         catch (Exception ex)
         {
@@ -122,7 +117,7 @@ internal class WebSocketResponseProvider(WebSocketBuilder builder) : IResponsePr
             }
 
             // Already upgraded - return marker
-            return (new WebSocketHandledResponse(), null);
+            return (new WebSocketHandledResponse(dateTimeUtils.UtcNow), null);
         }
     }
 

src/WireMock.Net.Minimal/Server/RespondWithAProvider.cs

@@ -2,8 +2,6 @@
 
 // This source file is based on mock4net by Alexandre Victoor which is licensed under the Apache 2.0 License.
 // For more details see 'mock4net/LICENSE.txt' and 'mock4net/readme.md' in this project root.
-using System;
-using System.Collections.Generic;
 using System.Net;
 using Stef.Validation;
 using WireMock.Matchers.Request;
@@ -25,6 +23,8 @@ internal class RespondWithAProvider : IRespondWithAProvider
     private readonly IRequestMatcher _requestMatcher;
     private readonly WireMockServerSettings _settings;
     private readonly IDateTimeUtils _dateTimeUtils;
+    private readonly IGuidUtils _guidUtils;
+
     private readonly bool _saveToFile;
 
     private int _priority;
@@ -49,16 +49,7 @@ internal class RespondWithAProvider : IRespondWithAProvider
 
     public IdOrTexts? ProtoDefinition { get; private set; }
 
-    /// <summary>
-    /// Initializes a new instance of the <see cref="RespondWithAProvider"/> class.
-    /// </summary>
-    /// <param name="registrationCallback">The registration callback.</param>
-    /// <param name="requestMatcher">The request matcher.</param>
-    /// <param name="settings">The WireMockServerSettings.</param>
-    /// <param name="guidUtils">GuidUtils to make unit testing possible.</param>
-    /// <param name="dateTimeUtils">DateTimeUtils to make unit testing possible.</param>
-    /// <param name="saveToFile">Optional boolean to indicate if this mapping should be saved as static mapping file.</param>
-    public RespondWithAProvider(
+    internal RespondWithAProvider(
         RegistrationCallback registrationCallback,
         IRequestMatcher requestMatcher,
         WireMockServerSettings settings,
@@ -71,6 +62,8 @@ internal class RespondWithAProvider : IRespondWithAProvider
         _requestMatcher = Guard.NotNull(requestMatcher);
         _settings = Guard.NotNull(settings);
         _dateTimeUtils = Guard.NotNull(dateTimeUtils);
+        _guidUtils = Guard.NotNull(guidUtils);
+
         _saveToFile = saveToFile;
 
         Guid = guidUtils.NewGuid();
@@ -82,7 +75,11 @@ internal class RespondWithAProvider : IRespondWithAProvider
         if (provider is Response response && response.WebSocketBuilder != null)
         {
             // If the provider is a Response with a WebSocketBuilder, we need to use a WebSocketResponseProvider instead.
-            provider = new WebSocketResponseProvider(response.WebSocketBuilder);
+            provider = new WebSocketResponseProvider(
+                response.WebSocketBuilder,
+                _guidUtils,
+                _dateTimeUtils
+            );
         }
 
         var mapping = new Mapping

src/WireMock.Net.Minimal/Util/DateTimeUtils.cs

@@ -1,7 +1,5 @@
 // Copyright © WireMock.Net
 
-using System;
-
 namespace WireMock.Util;
 
 internal interface IDateTimeUtils