DefaultSchedulerTest.cs 4.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121
  1. // Licensed to the .NET Foundation under one or more agreements.
  2. // The .NET Foundation licenses this file to you under the Apache 2.0 License.
  3. // See the LICENSE file in the project root for more information.
  4. using System;
  5. using System.Reactive.Concurrency;
  6. using System.Threading;
  7. using Microsoft.Reactive.Testing;
  8. using Xunit;
  9. namespace ReactiveTests.Tests
  10. {
  11. public class DefaultSchedulerTest
  12. {
  13. [Fact]
  14. public void Schedule_ArgumentChecking()
  15. {
  16. ReactiveAssert.Throws<ArgumentNullException>(() => DefaultScheduler.Instance.Schedule<int>(42, default));
  17. ReactiveAssert.Throws<ArgumentNullException>(() => DefaultScheduler.Instance.Schedule<int>(42, DateTimeOffset.Now, default));
  18. ReactiveAssert.Throws<ArgumentNullException>(() => DefaultScheduler.Instance.Schedule<int>(42, TimeSpan.Zero, default));
  19. ReactiveAssert.Throws<ArgumentNullException>(() => DefaultScheduler.Instance.SchedulePeriodic(42, TimeSpan.FromSeconds(1), default));
  20. ReactiveAssert.Throws<ArgumentOutOfRangeException>(() => DefaultScheduler.Instance.SchedulePeriodic(42, TimeSpan.FromSeconds(-1), _ => _));
  21. }
  22. [Fact]
  23. public void Get_Now()
  24. {
  25. var res = DefaultScheduler.Instance.Now - DateTime.Now;
  26. Assert.True(res.Seconds < 1);
  27. }
  28. #if !NO_THREAD
  29. [Fact]
  30. public void ScheduleAction()
  31. {
  32. var id = Thread.CurrentThread.ManagedThreadId;
  33. var nt = DefaultScheduler.Instance;
  34. var evt = new ManualResetEvent(false);
  35. nt.Schedule(() => { Assert.NotEqual(id, Thread.CurrentThread.ManagedThreadId); evt.Set(); });
  36. evt.WaitOne();
  37. }
  38. [Fact]
  39. public void ScheduleActionDue()
  40. {
  41. var id = Thread.CurrentThread.ManagedThreadId;
  42. var nt = DefaultScheduler.Instance;
  43. var evt = new ManualResetEvent(false);
  44. nt.Schedule(TimeSpan.FromSeconds(0.2), () => { Assert.NotEqual(id, Thread.CurrentThread.ManagedThreadId); evt.Set(); });
  45. evt.WaitOne();
  46. }
  47. [Fact]
  48. public void ScheduleActionCancel()
  49. {
  50. var id = Thread.CurrentThread.ManagedThreadId;
  51. var nt = DefaultScheduler.Instance;
  52. var set = false;
  53. var d = nt.Schedule(TimeSpan.FromSeconds(0.2), () => { Assert.True(false); set = true; });
  54. d.Dispose();
  55. Thread.Sleep(400);
  56. Assert.False(set);
  57. }
  58. [Fact]
  59. public void Periodic_NonReentrant()
  60. {
  61. var n = 0;
  62. var fail = false;
  63. var d = DefaultScheduler.Instance.SchedulePeriodic(0, TimeSpan.FromMilliseconds(50), x =>
  64. {
  65. try
  66. {
  67. if (Interlocked.Increment(ref n) > 1) // Without an AsyncLock this would fail.
  68. {
  69. fail = true;
  70. }
  71. Thread.Sleep(100);
  72. return x + 1;
  73. }
  74. finally
  75. {
  76. Interlocked.Decrement(ref n);
  77. }
  78. });
  79. Thread.Sleep(500);
  80. d.Dispose();
  81. Assert.False(fail);
  82. }
  83. #endif
  84. #if DESKTOPCLR
  85. [Trait("SkipCI", "true")]
  86. [Fact]
  87. public void No_ThreadPool_Starvation_Dispose()
  88. {
  89. ThreadPool.GetAvailableThreads(out var bwt, out var bio);
  90. var N = Environment.ProcessorCount * 2;
  91. for (var i = 0; i < N; i++)
  92. {
  93. var e = new ManualResetEvent(false);
  94. var f = new ManualResetEvent(false);
  95. var d = Scheduler.Default.Schedule(TimeSpan.FromMilliseconds(1), () => { e.Set(); f.WaitOne(); });
  96. e.WaitOne();
  97. d.Dispose();
  98. f.Set();
  99. }
  100. ThreadPool.GetAvailableThreads(out var ewt, out var eio);
  101. Assert.False(bwt - ewt >= N);
  102. }
  103. #endif
  104. }
  105. }