From: Matthew Endsley <mendsley@gmail.com>
Date: Wed, 3 Feb 2021 11:10:41 +0000 (-0800)
Subject: Leverage existing tests for DTLS listener
X-Git-Tag: 1.0.0~20^2~7^2~2
X-Git-Url: https://git.deb.at/?a=commitdiff_plain;h=919fa9997199bbe5b4f9563bb49e290112baebbb;p=rhonda%2Fimpostor.hazel.git

Leverage existing tests for DTLS listener
---

diff --git a/Hazel.UnitTests/Dtls/ConnectionTests.cs b/Hazel.UnitTests/Dtls/ConnectionTests.cs
index c6132fa..0a8077e 100644
--- a/Hazel.UnitTests/Dtls/ConnectionTests.cs
+++ b/Hazel.UnitTests/Dtls/ConnectionTests.cs
@@ -1,4 +1,6 @@
 using Hazel.Dtls;
+using Hazel.Udp;
+using Hazel.Udp.FewerThreads;
 using Microsoft.VisualStudio.TestTools.UnitTesting;
 using System.Net;
 using System.Security.Cryptography;
@@ -8,7 +10,7 @@ using System.Threading;
 namespace Hazel.UnitTests.Dtls
 {
     [TestClass]
-    public class ConnectionTests
+    public class ConnectionTests : BaseThreadLimitedUdpConnectionTests
     {
         // Created with command line
         // openssl req -newkey rsa:2048 -nodes -keyout key.pem -x509 -days 100000 -out certificate.pem
@@ -85,6 +87,21 @@ IsdbLCwHYD3GVgk/D7NVxyU=
             return clientCertificates;
         }
 
+        protected override ThreadLimitedUdpConnectionListener CreateListener(int numWorkers, IPEndPoint endPoint, ILogger logger, IPMode ipMode = IPMode.IPv4)
+        {
+            DtlsConnectionListener listener = new DtlsConnectionListener(2, endPoint, logger, ipMode);
+            listener.SetCertificate(GetCertificateForServer());
+            return listener;
+
+        }
+
+        protected override UdpConnection CreateConnection(IPEndPoint endPoint, ILogger logger, IPMode ipMode = IPMode.IPv4)
+        {
+            DtlsUnityConnection connection = new DtlsUnityConnection(logger, endPoint, ipMode);
+            connection.SetValidServerCertificates(GetCertificateForClient());
+            return connection;
+        }
+
         [TestMethod]
         public void TestClientConnects()
         {