Added Work Item passive timeout

2026-06-28 23:01:25 +03:00 · 2009-12-19 20:49:16 +02:00
parent 5d037fdec8
commit ccfd49f237
6 changed files with 166 additions and 10 deletions
@@ -0,0 +1,126 @@
+using System.Threading;
+
+using NUnit.Framework;
+
+using Amib.Threading;
+
+namespace SmartThreadPoolTests
+{
+    [TestFixture]
+    [Category("TestWorkItemTimeout")]
+    public class TestWorkItemTimeout
+    {
+        /// <summary>
+        /// 1. Create STP in suspended mode
+        /// 2. Queue work item into the STP
+        /// 3. Wait for the work item to expire
+        /// 4. Work item's GetResult should throw WorkItemCancelException
+        /// </summary>        
+        [Test]
+        [ExpectedException(typeof(WorkItemCancelException))]
+        public void CancelInQueueWorkItem()
+        {
+            STPStartInfo stpStartInfo = new STPStartInfo();
+            stpStartInfo.StartSuspended = true;
+
+            bool hasRun = false;
+
+            SmartThreadPool stp = new SmartThreadPool(stpStartInfo);
+            IWorkItemResult wir = stp.QueueWorkItem(
+                new WorkItemInfo() 
+                { 
+                    Timeout = 1000 }, 
+                    arg =>
+                        {
+                            hasRun = true;
+                            return null;
+                        }
+                    );
+
+            Assert.IsFalse(wir.IsCanceled);
+
+            Thread.Sleep(2000);
+
+            Assert.IsTrue(wir.IsCanceled);
+
+            stp.Start();
+            stp.WaitForIdle();
+
+            Assert.IsFalse(hasRun);
+
+            try
+            {
+                wir.GetResult();
+            }
+            finally
+            {
+                stp.Shutdown();
+            }
+        }
+
+        /// <summary>
+        /// 1. Create STP
+        /// 2. Queue work item that takes some time
+        /// 3. Wait for it to start
+        /// 4. The work item timeout expires
+        /// 5. Make sure, in the work item, that SmartThreadPool.IsWorkItemCanceled is true
+        /// 5. Wait for the STP to get idle
+        /// 6. Work item's GetResult should throw WorkItemCancelException
+        /// </summary>        
+        [Test]
+        public void TimeoutInProgressWorkItemWithSample()
+        {
+            bool timedout = false;
+            ManualResetEvent waitToStart = new ManualResetEvent(false);
+            ManualResetEvent waitToComplete = new ManualResetEvent(false);
+
+            SmartThreadPool stp = new SmartThreadPool();
+            IWorkItemResult wir = stp.QueueWorkItem(
+                new WorkItemInfo() { Timeout = 1000 },
+                state =>
+                {
+                    waitToStart.Set();
+                    Thread.Sleep(1000);
+                    timedout = SmartThreadPool.IsWorkItemCanceled;
+                    waitToComplete.Set();
+                    return null;
+                });
+
+            waitToStart.WaitOne();
+
+            waitToComplete.WaitOne();
+
+            Assert.IsTrue(timedout);
+
+            stp.Shutdown();
+        }
+
+        /// <summary>
+        /// 1. Create STP
+        /// 2. Queue work item into the STP
+        /// 3. Wait for the STP to get idle
+        /// 4. Work item's GetResult should return value
+        /// 4. The work item expires
+        /// 5. Work item's GetResult should return value
+        /// </summary>        
+        [Test]
+        public void TimeoutCompletedWorkItem()
+        {
+            SmartThreadPool stp = new SmartThreadPool();
+            IWorkItemResult wir = 
+                stp.QueueWorkItem(
+                new WorkItemInfo() { Timeout = 1000 },
+                state => 1);
+
+            stp.WaitForIdle();
+
+            Assert.AreEqual(wir.GetResult(), 1);
+
+            Thread.Sleep(1000);
+
+            Assert.AreEqual(wir.GetResult(), 1);
+
+            stp.Shutdown();
+        }
+    }
+}