use the RemoteEndpoint property from the ConnectionContext and added test

2026-10-20 11:45:43 +08:00 · 2020-04-14 19:16:32 +02:00
parent c10145f743
commit d49a4384c0
4 changed files with 108 additions and 8 deletions
@@ -0,0 +1,50 @@
+using Microsoft.AspNetCore.Connections;
+using MQTTnet.Adapter;
+using MQTTnet.Formatter;
+using MQTTnet.Server;
+using System;
+using System.Threading.Tasks;
+
+namespace MQTTnet.AspNetCore.Tests.Mockups
+{
+    public class ConnectionHandlerMockup: IMqttServerAdapter
+    {
+        public TaskCompletionSource<MqttConnectionContext> Context { get; } = new TaskCompletionSource<MqttConnectionContext>();
+        public Func<IMqttChannelAdapter, Task> ClientHandler { get; set; }
+
+        public ConnectionHandlerMockup()
+        {
+        }
+
+        public async Task OnConnectedAsync(ConnectionContext connection)
+        {
+            try
+            {
+                var writer = new SpanBasedMqttPacketWriter();
+                var formatter = new MqttPacketFormatterAdapter(writer);
+                var context = new MqttConnectionContext(formatter, connection);
+                Context.TrySetResult(context);
+
+                await ClientHandler(context);
+            }
+            catch (Exception ex)
+            {
+                Context.TrySetException(ex);
+            }
+        }
+
+        public Task StartAsync(IMqttServerOptions options)
+        {
+            return Task.CompletedTask;
+        }
+
+        public Task StopAsync()
+        {
+            return Task.CompletedTask;
+        }
+
+        public void Dispose()
+        {
+        }
+    }
+}