From 919fa9997199bbe5b4f9563bb49e290112baebbb Mon Sep 17 00:00:00 2001 From: Matthew Endsley Date: Wed, 3 Feb 2021 03:10:41 -0800 Subject: [PATCH] Leverage existing tests for DTLS listener --- Hazel.UnitTests/Dtls/ConnectionTests.cs | 19 ++++++++++++++++++- 1 file changed, 18 insertions(+), 1 deletion(-) 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() { -- 2.39.5