123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341 |
- // Licensed to the .NET Foundation under one or more agreements.
- // The .NET Foundation licenses this file to you under the MIT License.
- // See the LICENSE file in the project root for more information.
- #if !NO_THREAD
- using System;
- using System.Collections.Generic;
- using System.Linq;
- using System.Reactive.Concurrency;
- using System.Threading;
- using Microsoft.Reactive.Testing;
- using Microsoft.VisualStudio.TestTools.UnitTesting;
- using Assert = Xunit.Assert;
- namespace ReactiveTests.Tests
- {
- [TestClass]
- public class ThreadPoolSchedulerTest
- {
- [TestMethod]
- public void Schedule_ArgumentChecking()
- {
- ReactiveAssert.Throws<ArgumentNullException>(() => ThreadPoolScheduler.Instance.Schedule(42, default));
- ReactiveAssert.Throws<ArgumentNullException>(() => ThreadPoolScheduler.Instance.Schedule(42, DateTimeOffset.Now, default));
- ReactiveAssert.Throws<ArgumentNullException>(() => ThreadPoolScheduler.Instance.Schedule(42, TimeSpan.Zero, default));
- }
- [TestMethod]
- public void Get_Now()
- {
- var res = ThreadPoolScheduler.Instance.Now - DateTime.Now;
- Assert.True(res.Seconds < 1);
- }
- [TestMethod]
- public void ScheduleAction()
- {
- var id = Thread.CurrentThread.ManagedThreadId;
- var nt = ThreadPoolScheduler.Instance;
- var evt = new ManualResetEvent(false);
- nt.Schedule(() => { Assert.NotEqual(id, Thread.CurrentThread.ManagedThreadId); evt.Set(); });
- evt.WaitOne();
- }
- [TestMethod]
- public void ProperRooting_NoGC_SingleShot()
- {
- var cts = new CancellationTokenSource();
- new Thread(() =>
- {
- while (!cts.IsCancellationRequested)
- {
- Thread.Sleep(50);
- GC.Collect();
- GC.WaitForPendingFinalizers();
- }
- }).Start();
- var tp = ThreadPoolScheduler.Instance;
- var N = 100;
- var cd = new CountdownEvent(N);
- for (var i = 0; i < N; i++)
- {
- tp.Schedule(TimeSpan.FromMilliseconds(100 + i), () => { cd.Signal(); });
- }
- Assert.True(cd.Wait(TimeSpan.FromMinutes(1)));
- cts.Cancel();
- }
- [TestMethod]
- public void ProperRooting_NoGC_Periodic()
- {
- var cts = new CancellationTokenSource();
- new Thread(() =>
- {
- while (!cts.IsCancellationRequested)
- {
- Thread.Sleep(50);
- GC.Collect();
- GC.WaitForPendingFinalizers();
- }
- }).Start();
- var tp = ThreadPoolScheduler.Instance;
- var N = 5;
- var e = new ManualResetEvent(false);
- var n = 0;
- var d = tp.SchedulePeriodic(TimeSpan.FromMilliseconds(80), () => { if (Interlocked.Increment(ref n) == N) { e.Set(); } });
- Assert.True(e.WaitOne(TimeSpan.FromMinutes(1)));
- d.Dispose();
- cts.Cancel();
- }
- [TestMethod]
- public void ScheduleActionDueRelative()
- {
- var id = Thread.CurrentThread.ManagedThreadId;
- var nt = ThreadPoolScheduler.Instance;
- var evt = new ManualResetEvent(false);
- nt.Schedule(TimeSpan.FromSeconds(0.2), () => { Assert.NotEqual(id, Thread.CurrentThread.ManagedThreadId); evt.Set(); });
- evt.WaitOne();
- }
- [TestMethod]
- public void ScheduleActionDue0()
- {
- var id = Thread.CurrentThread.ManagedThreadId;
- var nt = ThreadPoolScheduler.Instance;
- var evt = new ManualResetEvent(false);
- nt.Schedule(TimeSpan.FromTicks(0), () => { Assert.NotEqual(id, Thread.CurrentThread.ManagedThreadId); evt.Set(); });
- evt.WaitOne();
- }
- [TestMethod]
- public void ScheduleActionDueAbsolute()
- {
- var id = Thread.CurrentThread.ManagedThreadId;
- var nt = ThreadPoolScheduler.Instance;
- var evt = new ManualResetEvent(false);
- nt.Schedule(DateTimeOffset.UtcNow + TimeSpan.FromSeconds(0.2), () => { Assert.NotEqual(id, Thread.CurrentThread.ManagedThreadId); evt.Set(); });
- evt.WaitOne();
- }
- [TestMethod]
- public void ScheduleActionCancel()
- {
- var id = Thread.CurrentThread.ManagedThreadId;
- var nt = ThreadPoolScheduler.Instance;
- var set = false;
- var d = nt.Schedule(TimeSpan.FromSeconds(0.2), () => { Assert.True(false); set = true; });
- d.Dispose();
- Thread.Sleep(400);
- Assert.False(set);
- }
- #if !NO_PERF
- #if !WINDOWS_UWP
- [TestMethod]
- public void ScheduleLongRunning_ArgumentChecking()
- {
- ReactiveAssert.Throws<ArgumentNullException>(() => ThreadPoolScheduler.Instance.ScheduleLongRunning(42, default));
- }
- [TestMethod]
- public void ScheduleLongRunning()
- {
- var id = Thread.CurrentThread.ManagedThreadId;
- var nt = ThreadPoolScheduler.Instance;
- var evt = new ManualResetEvent(false);
- nt.ScheduleLongRunning(42, (x, cancel) => { Assert.NotEqual(id, Thread.CurrentThread.ManagedThreadId); evt.Set(); });
- evt.WaitOne();
- }
- [TestMethod]
- public void ScheduleLongRunningCancel()
- {
- var nt = ThreadPoolScheduler.Instance;
- var started = new ManualResetEvent(false);
- var stopped = new ManualResetEvent(false);
- var n = 0;
- var d = nt.ScheduleLongRunning(42, (x, cancel) =>
- {
- for (n = 0; !cancel.IsDisposed; n++)
- {
- if (n == 10)
- {
- started.Set();
- }
- }
- stopped.Set();
- });
- started.WaitOne();
- d.Dispose();
- stopped.WaitOne();
- Assert.True(n >= 10);
- }
- #endif
- [TestMethod]
- public void Stopwatch()
- {
- var nt = ThreadPoolScheduler.Instance;
- var sw = nt.StartStopwatch();
- var s0 = sw.Elapsed.Ticks;
- Thread.Sleep(10);
- var s1 = sw.Elapsed.Ticks;
- Assert.True(s1 > s0);
- }
- [TestMethod]
- public void Periodic_ArgumentChecking()
- {
- ReactiveAssert.Throws<ArgumentNullException>(() => ThreadPoolScheduler.Instance.SchedulePeriodic(0, TimeSpan.FromSeconds(1), null));
- ReactiveAssert.Throws<ArgumentOutOfRangeException>(() => ThreadPoolScheduler.Instance.SchedulePeriodic(0, TimeSpan.FromSeconds(-1), _ => _));
- }
- [TestMethod]
- public void Periodic_Regular()
- {
- Periodic_Impl(TimeSpan.FromMilliseconds(25));
- }
- #if !WINDOWS_UWP
- [TestMethod]
- public void Periodic_Zero()
- {
- Periodic_Impl(TimeSpan.Zero);
- }
- #endif
- private void Periodic_Impl(TimeSpan period)
- {
- var gate = new object();
- var n = 0;
- var e = new ManualResetEvent(false);
- var lst = new List<int>();
- var d = ThreadPoolScheduler.Instance.SchedulePeriodic(0, period, x =>
- {
- lock (gate)
- {
- if (n++ == 10)
- {
- e.Set();
- }
- }
- lst.Add(x);
- return x + 1;
- });
- e.WaitOne();
- d.Dispose();
- var m = default(int);
- var k = default(int);
- var i = 0;
- do
- {
- lock (gate)
- {
- m = n;
- }
- Thread.Sleep(50);
- lock (gate)
- {
- k = n;
- }
- } while (m != k && i++ < 10); // Wait for Dispose to reach the timer; should be almost instantaneous due to nop'ing out of the action.
- Assert.NotEqual(10, i);
- var res = lst.ToArray();
- Assert.True(res.Length >= 10);
- Assert.True(res.Take(10).SequenceEqual(Enumerable.Range(0, 10)));
- }
- [TestMethod]
- public void Periodic_NonReentrant()
- {
- var n = 0;
- var fail = false;
- var d = ThreadPoolScheduler.Instance.SchedulePeriodic(0, TimeSpan.FromMilliseconds(50), x =>
- {
- try
- {
- if (Interlocked.Increment(ref n) > 1) // Without an AsyncLock this would fail.
- {
- fail = true;
- }
- Thread.Sleep(100);
- return x + 1;
- }
- finally
- {
- Interlocked.Decrement(ref n);
- }
- });
- Thread.Sleep(500);
- d.Dispose();
- Assert.False(fail);
- }
- #endif
- #if DESKTOPCLR
- [TestCategory("SkipCI")]
- [TestMethod]
- public void No_ThreadPool_Starvation_Dispose()
- {
- ThreadPool.GetAvailableThreads(out var bwt, out var bio);
- var N = Environment.ProcessorCount * 2;
- for (var i = 0; i < N; i++)
- {
- var e = new ManualResetEvent(false);
- var f = new ManualResetEvent(false);
- var d = ThreadPoolScheduler.Instance.Schedule(TimeSpan.FromMilliseconds(1), () => { e.Set(); f.WaitOne(); });
- e.WaitOne();
- d.Dispose();
- f.Set();
- }
- ThreadPool.GetAvailableThreads(out var ewt, out var eio);
- Assert.False(bwt - ewt >= N);
- }
- #endif
- }
- }
- #endif
|