Add queue with async support to free up threads.

2026-10-18 10:26:11 +08:00 · 2019-03-22 23:12:37 +01:00
parent 7ed5580352
commit 11a5b5a080
5 changed files with 111 additions and 99 deletions
@@ -0,0 +1,51 @@
+using System.Threading;
+using System.Threading.Tasks;
+using Microsoft.VisualStudio.TestTools.UnitTesting;
+using MQTTnet.Internal;
+
+namespace MQTTnet.Tests
+{
+    [TestClass]
+    public class AsyncQueue_Tests
+    {
+        [TestMethod]
+        public async Task Preserve_Order()
+        {
+            var queue = new AsyncQueue<string>();
+            queue.Enqueue("1");
+            queue.Enqueue("2");
+            queue.Enqueue("3");
+
+            Assert.AreEqual("1", await queue.DequeueAsync(CancellationToken.None));
+            Assert.AreEqual("2", await queue.DequeueAsync(CancellationToken.None));
+            Assert.AreEqual("3", await queue.DequeueAsync(CancellationToken.None));
+        }
+
+        [TestMethod]
+        public async Task Preserve_ProcessAsync()
+        {
+            var queue = new AsyncQueue<int>();
+
+            var sum = 0;
+            var worker = Task.Run(async () => 
+            {
+                while (sum < 6)
+                {
+                    sum += await queue.DequeueAsync(CancellationToken.None);
+                }
+            });
+            
+            queue.Enqueue(1);
+            await Task.Delay(500);
+
+            queue.Enqueue(2);
+            await Task.Delay(500);
+
+            queue.Enqueue(3);
+            await Task.Delay(500);
+
+            Assert.AreEqual(6, sum);
+            Assert.AreEqual(TaskStatus.RanToCompletion, worker.Status);
+        }
+    }
+}