Feature/mqttv5 subscriptions server (#1246)

* Add support for RetainHandling. * Add support for No Local. * Add support for _SubscriptionIdentifier_. * Add more Unit Tests. * Update docs.
2026-10-16 17:23:04 +08:00 · 2021-09-23 20:30:43 +02:00
parent feb53e78c6
commit c756a4092f
83 changed files with 2032 additions and 1237 deletions
@@ -0,0 +1,113 @@
+using System.Collections.Concurrent;
+using System.Threading.Tasks;
+using Microsoft.VisualStudio.TestTools.UnitTesting;
+using MQTTnet.Packets;
+using MQTTnet.Protocol;
+using MQTTnet.Server;
+using MQTTnet.Server.Internal;
+using MQTTnet.Tests.Mockups;
+
+namespace MQTTnet.Tests.Server
+{
+    [TestClass]
+    public class MqttSubscriptionsManager_Tests
+    {
+        [TestMethod]
+        public async Task MqttSubscriptionsManager_SubscribeSingleSuccess()
+        {
+            var s = CreateSession();
+
+            var sm = new MqttClientSubscriptionsManager(s, new MqttServerOptions(), new MqttServerEventDispatcher(new TestLogger()), new MqttRetainedMessagesManager());
+
+            var sp = new MqttSubscribePacket();
+            sp.TopicFilters.Add(new MqttTopicFilterBuilder().WithTopic("A/B/C").Build());
+
+            await sm.Subscribe(sp);
+
+            var result = sm.CheckSubscriptions("A/B/C", MqttQualityOfServiceLevel.AtMostOnce, "");
+            Assert.IsTrue(result.IsSubscribed);
+            Assert.AreEqual(result.QualityOfServiceLevel, MqttQualityOfServiceLevel.AtMostOnce);
+        }
+
+        [TestMethod]
+        public async Task MqttSubscriptionsManager_SubscribeDifferentQoSSuccess()
+        {
+            var s = CreateSession();
+
+            var sm = new MqttClientSubscriptionsManager(s, new MqttServerOptions(), new MqttServerEventDispatcher(new TestLogger()), new MqttRetainedMessagesManager());
+
+            var sp = new MqttSubscribePacket();
+            sp.TopicFilters.Add(new MqttTopicFilter { Topic = "A/B/C", QualityOfServiceLevel = MqttQualityOfServiceLevel.AtMostOnce });
+
+            await sm.Subscribe(sp);
+
+            var result = sm.CheckSubscriptions("A/B/C", MqttQualityOfServiceLevel.ExactlyOnce, "");
+            Assert.IsTrue(result.IsSubscribed);
+            Assert.AreEqual(result.QualityOfServiceLevel, MqttQualityOfServiceLevel.AtMostOnce);
+        }
+
+        [TestMethod]
+        public async Task MqttSubscriptionsManager_SubscribeTwoTimesSuccess()
+        {
+            var s = CreateSession();
+
+            var sm = new MqttClientSubscriptionsManager(s, new MqttServerOptions(), new MqttServerEventDispatcher(new TestLogger()), new MqttRetainedMessagesManager());
+
+            var sp = new MqttSubscribePacket();
+            sp.TopicFilters.Add(new MqttTopicFilter { Topic = "#", QualityOfServiceLevel = MqttQualityOfServiceLevel.AtMostOnce });
+            sp.TopicFilters.Add(new MqttTopicFilter { Topic = "A/B/C", QualityOfServiceLevel = MqttQualityOfServiceLevel.AtLeastOnce });
+
+            await sm.Subscribe(sp);
+
+            var result = sm.CheckSubscriptions("A/B/C", MqttQualityOfServiceLevel.ExactlyOnce, "");
+            Assert.IsTrue(result.IsSubscribed);
+            Assert.AreEqual(result.QualityOfServiceLevel, MqttQualityOfServiceLevel.AtLeastOnce);
+        }
+
+        [TestMethod]
+        public async Task MqttSubscriptionsManager_SubscribeSingleNoSuccess()
+        {
+            var s = CreateSession();
+
+            var sm = new MqttClientSubscriptionsManager(s, new MqttServerOptions(), new MqttServerEventDispatcher(new TestLogger()), new MqttRetainedMessagesManager());
+
+            var sp = new MqttSubscribePacket();
+            sp.TopicFilters.Add(new MqttTopicFilterBuilder().WithTopic("A/B/C").Build());
+
+            await sm.Subscribe(sp);
+
+            Assert.IsFalse(sm.CheckSubscriptions("A/B/X", MqttQualityOfServiceLevel.AtMostOnce, "").IsSubscribed);
+        }
+
+        [TestMethod]
+        public async Task MqttSubscriptionsManager_SubscribeAndUnsubscribeSingle()
+        {
+            var s = CreateSession();
+
+            var sm = new MqttClientSubscriptionsManager(s, new MqttServerOptions(), new MqttServerEventDispatcher(new TestLogger()), new MqttRetainedMessagesManager());
+
+            var sp = new MqttSubscribePacket();
+            sp.TopicFilters.Add(new MqttTopicFilterBuilder().WithTopic("A/B/C").Build());
+
+            await sm.Subscribe(sp);
+
+            Assert.IsTrue(sm.CheckSubscriptions("A/B/C", MqttQualityOfServiceLevel.AtMostOnce, "").IsSubscribed);
+
+            var up = new MqttUnsubscribePacket();
+            up.TopicFilters.Add("A/B/C");
+            await sm.Unsubscribe(up);
+
+            Assert.IsFalse(sm.CheckSubscriptions("A/B/C", MqttQualityOfServiceLevel.AtMostOnce, "").IsSubscribed);
+        }
+
+        MqttClientSession CreateSession()
+        {
+            return new MqttClientSession(
+                "",
+                new ConcurrentDictionary<object, object>(),
+                new MqttServerEventDispatcher(new TestLogger()),
+                new MqttServerOptions(),
+                new MqttRetainedMessagesManager());
+        }
+    }
+}