From 3eca5d44ee8f6f0ff63661816b9f6896c13c7999 Mon Sep 17 00:00:00 2001
From: Stef Heyenrath <stef.heyenrath@gmail.com>
Date: Fri, 3 Apr 2026 10:59:03 +0200
Subject: [PATCH] Fix WebSocket tests

---
 .../WebSockets/WebSocketIntegrationTests.cs                   | 4 ++--
 test/WireMock.Net.Tests/WireMock.Net.Tests.csproj             | 1 +
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/test/WireMock.Net.Tests/WebSockets/WebSocketIntegrationTests.cs b/test/WireMock.Net.Tests/WebSockets/WebSocketIntegrationTests.cs
index 201e652a..abf3ffc2 100644
--- a/test/WireMock.Net.Tests/WebSockets/WebSocketIntegrationTests.cs
+++ b/test/WireMock.Net.Tests/WebSockets/WebSocketIntegrationTests.cs
@@ -811,10 +811,10 @@ public class WebSocketIntegrationTests(ITestOutputHelper output, ITestContextAcc
         // Act
         await client.SendAsync(new ArraySegment<byte>(testData), WebSocketMessageType.Binary, true, _ct);
 
-        var receivedData = await client.ReceiveAsBytesAsync(cancellationToken: _ct);
-
         await Task.Delay(500, _ct);
 
+        var receivedData = await client.ReceiveAsBytesAsync(cancellationToken: _ct);
+
         // Assert
         receivedData.Should().BeEquivalentTo(testData, "binary data should be proxied and echoed back");
 
diff --git a/test/WireMock.Net.Tests/WireMock.Net.Tests.csproj b/test/WireMock.Net.Tests/WireMock.Net.Tests.csproj
index e8bc4c08..d1e9a99c 100644
--- a/test/WireMock.Net.Tests/WireMock.Net.Tests.csproj
+++ b/test/WireMock.Net.Tests/WireMock.Net.Tests.csproj
@@ -64,6 +64,7 @@
         </PackageReference>
         <PackageReference Include="Meziantou.Extensions.Logging.Xunit.v3" Version="1.1.24" />
         <PackageReference Include="Verify.XunitV3" Version="31.13.2" />
+        <PackageReference Include="xRetry.v3" Version="1.0.0-rc3" />
         <PackageReference Include="xunit.runner.visualstudio" Version="3.1.5">
             <PrivateAssets>all</PrivateAssets>
             <IncludeAssets>runtime; build; native; contentfiles; analyzers; buildtransitive</IncludeAssets>