Improve session handling.

2026-10-22 04:43:04 +08:00 · 2019-01-01 19:18:54 +01:00
parent 4cba814716
commit 96a67579e2
13 changed files with 478 additions and 219 deletions
@@ -0,0 +1,92 @@
+using System;
+using System.Collections.Generic;
+using System.Threading;
+using System.Threading.Tasks;
+using MQTTnet.Client;
+using MQTTnet.Client.Options;
+using MQTTnet.Diagnostics;
+using MQTTnet.Server;
+
+namespace MQTTnet.Tests
+{
+    public class TestSetup : IDisposable
+    {
+        private readonly MqttFactory _mqttFactory = new MqttFactory();
+        private readonly List<IMqttClient> _clients = new List<IMqttClient>();
+        private readonly IMqttNetLogger _serverLogger = new MqttNetLogger("server");
+        private readonly IMqttNetLogger _clientLogger = new MqttNetLogger("client");
+
+        private IMqttServer _server;
+
+        private long _serverErrorsCount;
+        private long _clientErrorsCount;
+
+        public TestSetup()
+        {
+            _serverLogger.LogMessagePublished += (s, e) =>
+            {
+                if (e.TraceMessage.Level == MqttNetLogLevel.Error)
+                {
+                    Interlocked.Increment(ref _serverErrorsCount);
+                }
+            };
+
+            _clientLogger.LogMessagePublished += (s, e) =>
+            {
+                if (e.TraceMessage.Level == MqttNetLogLevel.Error)
+                {
+                    Interlocked.Increment(ref _clientErrorsCount);
+                }
+            };
+        }
+
+        public async Task<IMqttServer> StartServerAsync(MqttServerOptionsBuilder options)
+        {
+            if (_server != null)
+            {
+                throw new InvalidOperationException("Server already started.");
+            }
+
+            _server = _mqttFactory.CreateMqttServer(_serverLogger);
+            await _server.StartAsync(options.WithDefaultEndpointPort(1888).Build());
+
+            return _server;
+        }
+
+        public async Task<IMqttClient> ConnectClient(MqttClientOptionsBuilder options)
+        {
+            var client = _mqttFactory.CreateMqttClient(_clientLogger);
+            _clients.Add(client);
+
+            await client.ConnectAsync(options.WithTcpServer("localhost", 1888).Build());
+
+            return client;
+        }
+
+        public void ThrowIfLogErrors()
+        {
+            if (_serverErrorsCount > 0)
+            {
+                throw new Exception($"Server had {_serverErrorsCount} errors.");
+            }
+
+            if (_clientErrorsCount > 0)
+            {
+                throw new Exception($"Client(s) had {_clientErrorsCount} errors.");
+            }
+        }
+
+        public void Dispose()
+        {
+            ThrowIfLogErrors();
+
+            foreach (var mqttClient in _clients)
+            {
+                mqttClient.DisconnectAsync().GetAwaiter().GetResult();
+                mqttClient.Dispose();
+            }
+
+            _server.StopAsync().GetAwaiter().GetResult();
+        }
+    }
+}