diff --git a/src/WireMock.Net.Minimal/ResponseProviders/WebSocketResponseProvider.cs b/src/WireMock.Net.Minimal/ResponseProviders/WebSocketResponseProvider.cs index 64460954..23a03bc6 100644 --- a/src/WireMock.Net.Minimal/ResponseProviders/WebSocketResponseProvider.cs +++ b/src/WireMock.Net.Minimal/ResponseProviders/WebSocketResponseProvider.cs @@ -67,7 +67,8 @@ internal class WebSocketResponseProvider(WebSocketBuilder builder, IGuidUtils gu builder, options, logger, - guidUtils + guidUtils, + dateTimeUtils ); // Add to registry diff --git a/src/WireMock.Net.Minimal/WebSockets/WireMockWebSocketContext.cs b/src/WireMock.Net.Minimal/WebSockets/WireMockWebSocketContext.cs index 5486c292..e5a9d380 100644 --- a/src/WireMock.Net.Minimal/WebSockets/WireMockWebSocketContext.cs +++ b/src/WireMock.Net.Minimal/WebSockets/WireMockWebSocketContext.cs @@ -18,6 +18,8 @@ namespace WireMock.WebSockets; /// public class WireMockWebSocketContext : IWebSocketContext { + private readonly IDateTimeUtils _dateTimeUtils; + /// public Guid ConnectionId { get; } @@ -50,7 +52,8 @@ public class WireMockWebSocketContext : IWebSocketContext WebSocketBuilder builder, IWireMockMiddlewareOptions options, IWireMockMiddlewareLogger logger, - IGuidUtils guidUtils + IGuidUtils guidUtils, + IDateTimeUtils dateTimeUtils ) { HttpContext = httpContext; @@ -61,6 +64,7 @@ public class WireMockWebSocketContext : IWebSocketContext Builder = builder; Options = options; Logger = logger; + _dateTimeUtils = dateTimeUtils; ConnectionId = guidUtils.NewGuid(); } @@ -169,7 +173,7 @@ public class WireMockWebSocketContext : IWebSocketContext null ) { - DateTime = DateTime.UtcNow + DateTime = _dateTimeUtils.UtcNow }; } else @@ -179,7 +183,7 @@ public class WireMockWebSocketContext : IWebSocketContext { Method = method, BodyData = bodyData, - DateTime = DateTime.UtcNow + DateTime = _dateTimeUtils.UtcNow }; }