1 files changed, 27 insertions, 0 deletions
diff --git a/Tools/Hazel-Networking/Hazel.UnitTests/Dtls/TestDtlsHandshakeDropUnityConnection.cs b/Tools/Hazel-Networking/Hazel.UnitTests/Dtls/TestDtlsHandshakeDropUnityConnection.cs
new file mode 100644
index 0000000..f33cd14
--- /dev/null
+++ b/Tools/Hazel-Networking/Hazel.UnitTests/Dtls/TestDtlsHandshakeDropUnityConnection.cs
@@ -0,0 +1,27 @@
+using Hazel.Dtls;
+using System.Net;
+
+namespace Hazel.UnitTests.Dtls
+{
+    internal class TestDtlsHandshakeDropUnityConnection : DtlsUnityConnection
+    {
+        public int DropSendClientKeyExchangeFlightCount = 0;
+
+        public TestDtlsHandshakeDropUnityConnection(ILogger logger, IPEndPoint remoteEndPoint, IPMode ipMode = IPMode.IPv4) : base(logger, remoteEndPoint, ipMode)
+        {
+
+        }
+
+        protected override bool DropClientKeyExchangeFlight()
+        {
+            if (DropSendClientKeyExchangeFlightCount > 0)
+            {
+                this.logger.WriteInfo($"Dropping SendClientKeyExchangeFlight");
+                --DropSendClientKeyExchangeFlightCount;
+                return true;
+            }
+
+            return false;
+        }
+    }
+}