1
0

SchedulerTest.cs 51 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520
  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. #if NET45
  5. using System;
  6. using System.Collections.Generic;
  7. using System.Linq;
  8. using System.Reactive.Concurrency;
  9. using System.Reactive.Disposables;
  10. using System.Reactive.Linq;
  11. using System.Reactive.PlatformServices;
  12. using System.Runtime.CompilerServices;
  13. using System.Threading;
  14. using Xunit;
  15. using Microsoft.Reactive.Testing;
  16. #if HAS_WINFORMS
  17. using System.Windows.Forms;
  18. #endif
  19. #if HAS_AWAIT
  20. using System.Threading.Tasks;
  21. #endif
  22. namespace ReactiveTests.Tests
  23. {
  24. public class SchedulerTest : ReactiveTest
  25. {
  26. #region IScheduler
  27. [Fact]
  28. public void Scheduler_ArgumentChecks()
  29. {
  30. var ms = new MyScheduler();
  31. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.Schedule(default(IScheduler), a => { }));
  32. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.Schedule(default(IScheduler), () => { }));
  33. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.Schedule(default(IScheduler), 1, (a, s) => { }));
  34. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.Schedule(ms, default(Action<Action>)));
  35. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.Schedule(ms, 1, default(Action<int, Action<int>>)));
  36. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.Schedule(default(IScheduler), DateTimeOffset.Now, a => { }));
  37. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.Schedule(default(IScheduler), DateTimeOffset.Now, () => { }));
  38. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.Schedule(default(IScheduler), 1, DateTimeOffset.Now, (a, s) => { }));
  39. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.Schedule(ms, DateTimeOffset.Now, default(Action<Action<DateTimeOffset>>)));
  40. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.Schedule(ms, 1, DateTimeOffset.Now, default(Action<int, Action<int, DateTimeOffset>>)));
  41. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.Schedule(default(IScheduler), TimeSpan.Zero, a => { }));
  42. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.Schedule(default(IScheduler), TimeSpan.Zero, () => { }));
  43. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.Schedule(default(IScheduler), 1, TimeSpan.Zero, (a, s) => { }));
  44. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.Schedule(ms, TimeSpan.Zero, default(Action<Action<TimeSpan>>)));
  45. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.Schedule(ms, 1, TimeSpan.Zero, default(Action<int, Action<int, TimeSpan>>)));
  46. }
  47. [Fact]
  48. public void Schedulers_ArgumentChecks()
  49. {
  50. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.CurrentThread.Schedule(default(Action)));
  51. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.CurrentThread.Schedule(TimeSpan.Zero, default(Action)));
  52. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.CurrentThread.Schedule(DateTimeOffset.MaxValue, default(Action)));
  53. ReactiveAssert.Throws<ArgumentNullException>(() => DispatcherScheduler.Instance.Schedule(default(Action)));
  54. ReactiveAssert.Throws<ArgumentNullException>(() => DispatcherScheduler.Instance.Schedule(TimeSpan.Zero, default(Action)));
  55. ReactiveAssert.Throws<ArgumentNullException>(() => DispatcherScheduler.Instance.Schedule(DateTimeOffset.MaxValue, default(Action)));
  56. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.Immediate.Schedule(default(Action)));
  57. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.Immediate.Schedule(TimeSpan.Zero, default(Action)));
  58. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.Immediate.Schedule(DateTimeOffset.MaxValue, default(Action)));
  59. ReactiveAssert.Throws<ArgumentNullException>(() => NewThreadScheduler.Default.Schedule(default(Action)));
  60. ReactiveAssert.Throws<ArgumentNullException>(() => NewThreadScheduler.Default.Schedule(TimeSpan.Zero, default(Action)));
  61. ReactiveAssert.Throws<ArgumentNullException>(() => NewThreadScheduler.Default.Schedule(DateTimeOffset.MaxValue, default(Action)));
  62. #if !NO_TPL
  63. ReactiveAssert.Throws<ArgumentNullException>(() => TaskPoolScheduler.Default.Schedule(default(Action)));
  64. ReactiveAssert.Throws<ArgumentNullException>(() => TaskPoolScheduler.Default.Schedule(TimeSpan.Zero, default(Action)));
  65. ReactiveAssert.Throws<ArgumentNullException>(() => TaskPoolScheduler.Default.Schedule(DateTimeOffset.MaxValue, default(Action)));
  66. #endif
  67. ReactiveAssert.Throws<ArgumentNullException>(() => DefaultScheduler.Instance.Schedule(default(Action)));
  68. ReactiveAssert.Throws<ArgumentNullException>(() => DefaultScheduler.Instance.Schedule(TimeSpan.Zero, default(Action)));
  69. ReactiveAssert.Throws<ArgumentNullException>(() => DefaultScheduler.Instance.Schedule(DateTimeOffset.MaxValue, default(Action)));
  70. ReactiveAssert.Throws<ArgumentNullException>(() => DefaultScheduler.Instance.SchedulePeriodic(42, TimeSpan.FromSeconds(1), default(Func<int, int>)));
  71. ReactiveAssert.Throws<ArgumentOutOfRangeException>(() => DefaultScheduler.Instance.SchedulePeriodic(42, TimeSpan.FromSeconds(-1), _ => _));
  72. #if HAS_WINFORMS
  73. var lbl = new Label();
  74. ReactiveAssert.Throws<ArgumentNullException>(() => new ControlScheduler(lbl).Schedule(default(Action)));
  75. ReactiveAssert.Throws<ArgumentNullException>(() => new ControlScheduler(lbl).Schedule(TimeSpan.Zero, default(Action)));
  76. ReactiveAssert.Throws<ArgumentNullException>(() => new ControlScheduler(lbl).Schedule(DateTimeOffset.MaxValue, default(Action)));
  77. #endif
  78. var ctx = new SynchronizationContext();
  79. ReactiveAssert.Throws<ArgumentNullException>(() => new SynchronizationContextScheduler(ctx).Schedule(default(Action)));
  80. ReactiveAssert.Throws<ArgumentNullException>(() => new SynchronizationContextScheduler(ctx).Schedule(TimeSpan.Zero, default(Action)));
  81. ReactiveAssert.Throws<ArgumentNullException>(() => new SynchronizationContextScheduler(ctx).Schedule(DateTimeOffset.MaxValue, default(Action)));
  82. }
  83. [Fact]
  84. public void Scheduler_ScheduleNonRecursive()
  85. {
  86. var ms = new MyScheduler();
  87. var res = false;
  88. Scheduler.Schedule(ms, a => { res = true; });
  89. Assert.True(res);
  90. }
  91. [Fact]
  92. public void Scheduler_ScheduleRecursive()
  93. {
  94. var ms = new MyScheduler();
  95. var i = 0;
  96. Scheduler.Schedule(ms, a => { if (++i < 10) a(); });
  97. Assert.Equal(10, i);
  98. }
  99. [Fact]
  100. public void Scheduler_ScheduleWithTimeNonRecursive()
  101. {
  102. var now = DateTimeOffset.Now;
  103. var ms = new MyScheduler(now) { Check = (a, s, t) => { Assert.True(t == TimeSpan.Zero); } };
  104. var res = false;
  105. Scheduler.Schedule(ms, now, a => { res = true; });
  106. Assert.True(res);
  107. Assert.True(ms.WaitCycles == 0);
  108. }
  109. [Fact]
  110. public void Scheduler_ScheduleWithTimeRecursive()
  111. {
  112. var now = DateTimeOffset.Now;
  113. var i = 0;
  114. var ms = new MyScheduler(now) { Check = (a, s, t) => { Assert.True(t == TimeSpan.Zero); } };
  115. Scheduler.Schedule(ms, now, a => { if (++i < 10) a(now); });
  116. Assert.True(ms.WaitCycles == 0);
  117. Assert.Equal(10, i);
  118. }
  119. [Fact]
  120. public void Scheduler_ScheduleWithTimeSpanNonRecursive()
  121. {
  122. var now = DateTimeOffset.Now;
  123. var ms = new MyScheduler(now) { Check = (a, s, t) => { Assert.True(t == TimeSpan.Zero); } };
  124. var res = false;
  125. Scheduler.Schedule(ms, TimeSpan.Zero, a => { res = true; });
  126. Assert.True(res);
  127. Assert.True(ms.WaitCycles == 0);
  128. }
  129. [Fact]
  130. public void Scheduler_ScheduleWithTimeSpanRecursive()
  131. {
  132. var now = DateTimeOffset.Now;
  133. var ms = new MyScheduler(now) { Check = (a, s, t) => { Assert.True(t < TimeSpan.FromTicks(10)); } };
  134. var i = 0;
  135. Scheduler.Schedule(ms, TimeSpan.Zero, a => { if (++i < 10) a(TimeSpan.FromTicks(i)); });
  136. Assert.True(ms.WaitCycles == Enumerable.Range(1, 9).Sum());
  137. Assert.Equal(10, i);
  138. }
  139. [Fact]
  140. public void Scheduler_StateThreading()
  141. {
  142. var lst = new List<int>();
  143. Scheduler.Immediate.Schedule(0, (i, a) =>
  144. {
  145. lst.Add(i);
  146. if (i < 9)
  147. a(i + 1);
  148. });
  149. Assert.True(lst.SequenceEqual(Enumerable.Range(0, 10)));
  150. }
  151. [Fact]
  152. public void Scheduler_Builtins()
  153. {
  154. // Default
  155. {
  156. var e = new ManualResetEvent(false);
  157. Scheduler.Default.Schedule(() => e.Set());
  158. e.WaitOne();
  159. }
  160. if (!Utils.IsRunningWithPortableLibraryBinaries())
  161. {
  162. Scheduler_Builtins_NoPlib();
  163. }
  164. }
  165. [MethodImpl(MethodImplOptions.NoInlining)]
  166. private void Scheduler_Builtins_NoPlib()
  167. {
  168. #if !PLIB
  169. // ThreadPool
  170. {
  171. var e = new ManualResetEvent(false);
  172. Scheduler.ThreadPool.Schedule(() => e.Set());
  173. e.WaitOne();
  174. }
  175. #endif
  176. #if !NO_THREAD
  177. // NewThread
  178. {
  179. var e = new ManualResetEvent(false);
  180. Scheduler.NewThread.Schedule(() => e.Set());
  181. e.WaitOne();
  182. }
  183. #endif
  184. #if !NO_TPL
  185. // TaskPool
  186. {
  187. var e = new ManualResetEvent(false);
  188. Scheduler.TaskPool.Schedule(() => e.Set());
  189. e.WaitOne();
  190. }
  191. #endif
  192. }
  193. #endregion
  194. #region ISchedulerLongRunning
  195. #if !NO_PERF
  196. [Fact]
  197. public void Scheduler_LongRunning_ArgumentChecking()
  198. {
  199. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.ScheduleLongRunning(null, c => { }));
  200. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.ScheduleLongRunning(ThreadPoolScheduler.Instance, default(Action<ICancelable>)));
  201. }
  202. [Fact]
  203. public void Scheduler_Periodic_ArgumentChecking()
  204. {
  205. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.SchedulePeriodic(null, TimeSpan.FromSeconds(1), () => { }));
  206. ReactiveAssert.Throws<ArgumentOutOfRangeException>(() => Scheduler.SchedulePeriodic(ThreadPoolScheduler.Instance, TimeSpan.FromSeconds(-1), () => { }));
  207. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.SchedulePeriodic(ThreadPoolScheduler.Instance, TimeSpan.FromSeconds(1), default(Action)));
  208. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.SchedulePeriodic<int>(null, 42, TimeSpan.FromSeconds(1), _ => { }));
  209. ReactiveAssert.Throws<ArgumentOutOfRangeException>(() => Scheduler.SchedulePeriodic<int>(ThreadPoolScheduler.Instance, 42, TimeSpan.FromSeconds(-1), _ => { }));
  210. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.SchedulePeriodic<int>(ThreadPoolScheduler.Instance, 42, TimeSpan.FromSeconds(1), default(Action<int>)));
  211. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.SchedulePeriodic<int>(null, 42, TimeSpan.FromSeconds(1), _ => _));
  212. ReactiveAssert.Throws<ArgumentOutOfRangeException>(() => Scheduler.SchedulePeriodic<int>(ThreadPoolScheduler.Instance, 42, TimeSpan.FromSeconds(-1), _ => _));
  213. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.SchedulePeriodic<int>(ThreadPoolScheduler.Instance, 42, TimeSpan.FromSeconds(1), default(Func<int, int>)));
  214. }
  215. [Fact]
  216. public void Scheduler_Stopwatch_Emulation()
  217. {
  218. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.StartStopwatch(null));
  219. }
  220. #if !NO_TPL
  221. [Fact]
  222. public void Scheduler_LongRunning1()
  223. {
  224. var s = TaskPoolScheduler.Default;
  225. var x = new ManualResetEvent(false);
  226. var e = new ManualResetEvent(false);
  227. var d = s.ScheduleLongRunning(42, (state, cancel) =>
  228. {
  229. while (!cancel.IsDisposed)
  230. x.Set();
  231. e.Set();
  232. });
  233. x.WaitOne();
  234. d.Dispose();
  235. e.WaitOne();
  236. }
  237. [Fact]
  238. public void Scheduler_LongRunning2()
  239. {
  240. var s = TaskPoolScheduler.Default;
  241. var x = new ManualResetEvent(false);
  242. var e = new ManualResetEvent(false);
  243. var d = s.ScheduleLongRunning(cancel =>
  244. {
  245. while (!cancel.IsDisposed)
  246. x.Set();
  247. e.Set();
  248. });
  249. x.WaitOne();
  250. d.Dispose();
  251. e.WaitOne();
  252. }
  253. #endif
  254. #endif
  255. #endregion
  256. #region ISchedulerPeriodic
  257. #if !NO_PERF
  258. [Fact]
  259. public void Scheduler_Periodic1()
  260. {
  261. var n = 0;
  262. var e = new ManualResetEvent(false);
  263. var d = ThreadPoolScheduler.Instance.SchedulePeriodic(TimeSpan.FromMilliseconds(50), () =>
  264. {
  265. if (n++ == 10)
  266. e.Set();
  267. });
  268. e.WaitOne();
  269. d.Dispose();
  270. }
  271. [Fact]
  272. public void Scheduler_Periodic2()
  273. {
  274. var n = 0;
  275. var e = new ManualResetEvent(false);
  276. var d = ThreadPoolScheduler.Instance.SchedulePeriodic(42, TimeSpan.FromMilliseconds(50), x =>
  277. {
  278. Assert.Equal(42, x);
  279. if (n++ == 10)
  280. e.Set();
  281. });
  282. e.WaitOne();
  283. d.Dispose();
  284. }
  285. #if DESKTOPCLR
  286. [Fact]
  287. public void Scheduler_Periodic_HostLifecycleManagement()
  288. {
  289. var cur = AppDomain.CurrentDomain.BaseDirectory;
  290. var domain = AppDomain.CreateDomain("HLN", null, new AppDomainSetup { ApplicationBase = cur });
  291. domain.DoCallBack(Scheduler_Periodic_HostLifecycleManagement_Callback);
  292. }
  293. private static void Scheduler_Periodic_HostLifecycleManagement_Callback()
  294. {
  295. var pep = PlatformEnlightenmentProvider.Current;
  296. try
  297. {
  298. var hln = new HLN();
  299. PlatformEnlightenmentProvider.Current = new PEP(hln);
  300. var s = ThreadPoolScheduler.Instance.DisableOptimizations(typeof(ISchedulerPeriodic));
  301. var n = 0;
  302. var e = new ManualResetEvent(false);
  303. var d = Observable.Interval(TimeSpan.FromMilliseconds(100), s).Subscribe(_ =>
  304. {
  305. if (n++ == 10)
  306. e.Set();
  307. });
  308. hln.OnSuspending();
  309. hln.OnResuming();
  310. Thread.Sleep(250);
  311. hln.OnSuspending();
  312. Thread.Sleep(150);
  313. hln.OnResuming();
  314. Thread.Sleep(50);
  315. hln.OnSuspending();
  316. hln.OnResuming();
  317. e.WaitOne();
  318. d.Dispose();
  319. }
  320. finally
  321. {
  322. PlatformEnlightenmentProvider.Current = pep;
  323. }
  324. }
  325. class PEP : IPlatformEnlightenmentProvider
  326. {
  327. private readonly IPlatformEnlightenmentProvider _old;
  328. private readonly IHostLifecycleNotifications _hln;
  329. public PEP(HLN hln)
  330. {
  331. _old = PlatformEnlightenmentProvider.Current;
  332. _hln = hln;
  333. }
  334. public T GetService<T>(params object[] args) where T : class
  335. {
  336. if (typeof(T) == typeof(IHostLifecycleNotifications))
  337. {
  338. return (T)(object)_hln;
  339. }
  340. return _old.GetService<T>(args);
  341. }
  342. }
  343. class HLN : IHostLifecycleNotifications
  344. {
  345. public event EventHandler<HostSuspendingEventArgs> Suspending;
  346. public event EventHandler<HostResumingEventArgs> Resuming;
  347. public void OnSuspending()
  348. {
  349. var s = Suspending;
  350. if (s != null)
  351. s(this, null);
  352. }
  353. public void OnResuming()
  354. {
  355. var s = Resuming;
  356. if (s != null)
  357. s(this, null);
  358. }
  359. }
  360. #endif
  361. #endif
  362. #endregion
  363. #region DisableOptimizations
  364. #if !NO_PERF
  365. [Fact]
  366. public void DisableOptimizations_ArgumentChecking()
  367. {
  368. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.DisableOptimizations(default(IScheduler)));
  369. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.DisableOptimizations(default(IScheduler), new Type[0]));
  370. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.DisableOptimizations(ThreadPoolScheduler.Instance, default(Type[])));
  371. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.DisableOptimizations(Scheduler.Default).Schedule(42, default(Func<IScheduler, int, IDisposable>)));
  372. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.DisableOptimizations(Scheduler.Default).Schedule(42, TimeSpan.FromSeconds(1), default(Func<IScheduler, int, IDisposable>)));
  373. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.DisableOptimizations(Scheduler.Default).Schedule(42, DateTimeOffset.Now, default(Func<IScheduler, int, IDisposable>)));
  374. }
  375. #if !NO_TPL
  376. [Fact]
  377. public void DisableOptimizations1()
  378. {
  379. var s = TaskPoolScheduler.Default;
  380. Assert.True(s is IServiceProvider);
  381. var t = s.DisableOptimizations();
  382. var d = t.Now - s.Now;
  383. Assert.True(d.TotalSeconds < 1);
  384. var e1 = new ManualResetEvent(false);
  385. t.Schedule(42, (self, state) =>
  386. {
  387. e1.Set();
  388. return Disposable.Empty;
  389. });
  390. e1.WaitOne();
  391. var e2 = new ManualResetEvent(false);
  392. t.Schedule(42, TimeSpan.FromMilliseconds(1), (self, state) =>
  393. {
  394. e2.Set();
  395. return Disposable.Empty;
  396. });
  397. e2.WaitOne();
  398. var e3 = new ManualResetEvent(false);
  399. t.Schedule(42, DateTimeOffset.Now.AddMilliseconds(10), (self, state) =>
  400. {
  401. e3.Set();
  402. return Disposable.Empty;
  403. });
  404. e3.WaitOne();
  405. }
  406. [Fact]
  407. public void DisableOptimizations2()
  408. {
  409. var s = TaskPoolScheduler.Default;
  410. Assert.True(s is IServiceProvider);
  411. var lr1 = ((IServiceProvider)s).GetService(typeof(ISchedulerLongRunning));
  412. Assert.NotNull(lr1);
  413. var e1 = new ManualResetEvent(false);
  414. s.Schedule(42, (self, state) =>
  415. {
  416. Assert.True(self is IServiceProvider);
  417. var lrr1 = ((IServiceProvider)self).GetService(typeof(ISchedulerLongRunning));
  418. Assert.NotNull(lrr1);
  419. e1.Set();
  420. return Disposable.Empty;
  421. });
  422. e1.WaitOne();
  423. var t = s.DisableOptimizations();
  424. Assert.True(t is IServiceProvider);
  425. var lr2 = ((IServiceProvider)t).GetService(typeof(ISchedulerLongRunning));
  426. Assert.Null(lr2);
  427. var e2 = new ManualResetEvent(false);
  428. t.Schedule(42, (self, state) =>
  429. {
  430. Assert.True(self is IServiceProvider);
  431. var lrr2 = ((IServiceProvider)self).GetService(typeof(ISchedulerLongRunning));
  432. Assert.Null(lrr2);
  433. e2.Set();
  434. return Disposable.Empty;
  435. });
  436. e2.WaitOne();
  437. }
  438. [Fact]
  439. public void DisableOptimizations3()
  440. {
  441. var s = TaskPoolScheduler.Default;
  442. Assert.True(s is IServiceProvider);
  443. var lr1 = ((IServiceProvider)s).GetService(typeof(ISchedulerLongRunning));
  444. Assert.NotNull(lr1);
  445. var p1 = ((IServiceProvider)s).GetService(typeof(ISchedulerPeriodic));
  446. Assert.NotNull(p1);
  447. var e1 = new ManualResetEvent(false);
  448. s.Schedule(42, (self, state) =>
  449. {
  450. Assert.True(self is IServiceProvider);
  451. var lrr1 = ((IServiceProvider)self).GetService(typeof(ISchedulerLongRunning));
  452. Assert.NotNull(lrr1);
  453. var pr1 = ((IServiceProvider)self).GetService(typeof(ISchedulerPeriodic));
  454. Assert.NotNull(pr1);
  455. e1.Set();
  456. return Disposable.Empty;
  457. });
  458. e1.WaitOne();
  459. var t = s.DisableOptimizations(typeof(ISchedulerLongRunning));
  460. Assert.True(t is IServiceProvider);
  461. var lr2 = ((IServiceProvider)t).GetService(typeof(ISchedulerLongRunning));
  462. Assert.Null(lr2);
  463. var p2 = ((IServiceProvider)t).GetService(typeof(ISchedulerPeriodic));
  464. Assert.NotNull(p2);
  465. var e2 = new ManualResetEvent(false);
  466. t.Schedule(42, (self, state) =>
  467. {
  468. Assert.True(self is IServiceProvider);
  469. var lrr2 = ((IServiceProvider)self).GetService(typeof(ISchedulerLongRunning));
  470. Assert.Null(lrr2);
  471. var pr2 = ((IServiceProvider)self).GetService(typeof(ISchedulerPeriodic));
  472. Assert.NotNull(pr2);
  473. e2.Set();
  474. return Disposable.Empty;
  475. });
  476. e2.WaitOne();
  477. }
  478. #endif
  479. #endif
  480. [Fact]
  481. public void DisableOptimizations_UnknownService()
  482. {
  483. var s = new MyScheduler();
  484. Assert.False(s is IServiceProvider);
  485. var d = s.DisableOptimizations();
  486. Assert.True(d is IServiceProvider);
  487. Assert.Null(((IServiceProvider)d).GetService(typeof(bool)));
  488. }
  489. class MyScheduler : IScheduler
  490. {
  491. public MyScheduler()
  492. : this(DateTimeOffset.Now)
  493. {
  494. }
  495. public MyScheduler(DateTimeOffset now)
  496. {
  497. Now = now;
  498. }
  499. public DateTimeOffset Now
  500. {
  501. get;
  502. private set;
  503. }
  504. public IDisposable Schedule<T>(T state, Func<IScheduler, T, IDisposable> action)
  505. {
  506. return action(this, state);
  507. }
  508. public Action<Action<object>, object, TimeSpan> Check { get; set; }
  509. public long WaitCycles { get; set; }
  510. public IDisposable Schedule<T>(T state, TimeSpan dueTime, Func<IScheduler, T, IDisposable> action)
  511. {
  512. Check(o => action(this, (T)o), state, dueTime);
  513. WaitCycles += dueTime.Ticks;
  514. return action(this, state);
  515. }
  516. public IDisposable Schedule<T>(T state, DateTimeOffset dueTime, Func<IScheduler, T, IDisposable> action)
  517. {
  518. return Schedule(state, dueTime - Now, action);
  519. }
  520. }
  521. #endregion
  522. #region Catch
  523. [Fact]
  524. public void Catch_ArgumentChecking()
  525. {
  526. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.Catch<Exception>(default(IScheduler), _ => true));
  527. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.Catch<Exception>(Scheduler.Default, default(Func<Exception, bool>)));
  528. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.Catch<Exception>(Scheduler.Default, _ => true).Schedule(42, default(Func<IScheduler, int, IDisposable>)));
  529. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.Catch<Exception>(Scheduler.Default, _ => true).Schedule(42, TimeSpan.FromSeconds(1), default(Func<IScheduler, int, IDisposable>)));
  530. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.Catch<Exception>(Scheduler.Default, _ => true).Schedule(42, DateTimeOffset.Now, default(Func<IScheduler, int, IDisposable>)));
  531. }
  532. [Fact]
  533. public void Catch_Builtin_Swallow_Shallow()
  534. {
  535. var done = new ManualResetEvent(false);
  536. var swallow = Scheduler.Default.Catch<InvalidOperationException>(_ => { done.Set(); return true; });
  537. swallow.Schedule(() =>
  538. {
  539. throw new InvalidOperationException();
  540. });
  541. done.WaitOne();
  542. }
  543. [Fact]
  544. public void Catch_Builtin_Swallow_Recursive()
  545. {
  546. var done = new ManualResetEvent(false);
  547. var swallow = Scheduler.Default.Catch<InvalidOperationException>(_ => { done.Set(); return true; });
  548. swallow.Schedule(42, (self, state) =>
  549. {
  550. return self.Schedule(() =>
  551. {
  552. throw new InvalidOperationException();
  553. });
  554. });
  555. done.WaitOne();
  556. }
  557. [Fact]
  558. public void Catch_Custom_Unhandled()
  559. {
  560. var err = default(Exception);
  561. var scheduler = new MyExceptionScheduler(ex_ => err = ex_);
  562. scheduler.Catch<InvalidOperationException>(_ => true).Schedule(() =>
  563. {
  564. throw new InvalidOperationException();
  565. });
  566. Assert.Null(err);
  567. var ex = new ArgumentException();
  568. scheduler.Catch<InvalidOperationException>(_ => true).Schedule(() =>
  569. {
  570. throw ex;
  571. });
  572. Assert.Same(ex, err);
  573. }
  574. [Fact]
  575. public void Catch_Custom_Rethrow()
  576. {
  577. var err = default(Exception);
  578. var scheduler = new MyExceptionScheduler(ex_ => err = ex_);
  579. var ex = new InvalidOperationException();
  580. scheduler.Catch<InvalidOperationException>(_ => false).Schedule(() =>
  581. {
  582. throw ex;
  583. });
  584. Assert.Same(ex, err);
  585. }
  586. [Fact]
  587. public void Catch_Custom_LongRunning_Caught()
  588. {
  589. var err = default(Exception);
  590. var scheduler = new MyExceptionScheduler(ex_ => err = ex_);
  591. var caught = false;
  592. var @catch = scheduler.Catch<InvalidOperationException>(_ => caught = true);
  593. var slr = (ISchedulerLongRunning)((IServiceProvider)@catch).GetService(typeof(ISchedulerLongRunning));
  594. slr.ScheduleLongRunning(cancel =>
  595. {
  596. throw new InvalidOperationException();
  597. });
  598. Assert.True(caught);
  599. Assert.Null(err);
  600. var ex = new ArgumentException();
  601. slr.ScheduleLongRunning(cancel =>
  602. {
  603. throw ex;
  604. });
  605. Assert.Same(ex, err);
  606. }
  607. [Fact]
  608. public void Catch_Custom_LongRunning_Rethrow()
  609. {
  610. var err = default(Exception);
  611. var scheduler = new MyExceptionScheduler(ex_ => err = ex_);
  612. var @catch = scheduler.Catch<InvalidOperationException>(_ => false);
  613. var slr = (ISchedulerLongRunning)((IServiceProvider)@catch).GetService(typeof(ISchedulerLongRunning));
  614. var done = false;
  615. slr.ScheduleLongRunning(cancel =>
  616. {
  617. done = true;
  618. });
  619. Assert.True(done);
  620. var ex = new InvalidOperationException();
  621. slr.ScheduleLongRunning(cancel =>
  622. {
  623. throw ex;
  624. });
  625. Assert.Same(ex, err);
  626. }
  627. [Fact]
  628. public void Catch_Custom_Periodic_Regular()
  629. {
  630. var scheduler = new MyExceptionScheduler(_ => { });
  631. scheduler.PeriodicStopped = new ManualResetEvent(false);
  632. var @catch = scheduler.Catch<InvalidOperationException>(_ => true);
  633. var per = (ISchedulerPeriodic)((IServiceProvider)@catch).GetService(typeof(ISchedulerPeriodic));
  634. var madeProgress = new ManualResetEvent(false);
  635. var d = per.SchedulePeriodic(0, TimeSpan.Zero, x =>
  636. {
  637. if (x > 10)
  638. madeProgress.Set();
  639. return x + 1;
  640. });
  641. madeProgress.WaitOne();
  642. d.Dispose();
  643. scheduler.PeriodicStopped.WaitOne();
  644. }
  645. [Fact]
  646. public void Catch_Custom_Periodic_Uncaught1()
  647. {
  648. var err = default(Exception);
  649. var done = new ManualResetEvent(false);
  650. var scheduler = new MyExceptionScheduler(ex_ => { err = ex_; done.Set(); });
  651. scheduler.PeriodicStopped = new ManualResetEvent(false);
  652. var @catch = scheduler.Catch<InvalidOperationException>(_ => true);
  653. var per = (ISchedulerPeriodic)((IServiceProvider)@catch).GetService(typeof(ISchedulerPeriodic));
  654. var ex = new ArgumentException();
  655. per.SchedulePeriodic(42, TimeSpan.Zero, x =>
  656. {
  657. throw ex;
  658. });
  659. scheduler.PeriodicStopped.WaitOne();
  660. done.WaitOne();
  661. Assert.Same(ex, err);
  662. }
  663. [Fact]
  664. public void Catch_Custom_Periodic_Uncaught2()
  665. {
  666. var err = default(Exception);
  667. var done = new ManualResetEvent(false);
  668. var scheduler = new MyExceptionScheduler(ex_ => { err = ex_; done.Set(); });
  669. scheduler.PeriodicStopped = new ManualResetEvent(false);
  670. var @catch = scheduler.Catch<InvalidOperationException>(_ => false);
  671. var per = (ISchedulerPeriodic)((IServiceProvider)@catch).GetService(typeof(ISchedulerPeriodic));
  672. var ex = new InvalidOperationException();
  673. per.SchedulePeriodic(42, TimeSpan.Zero, x =>
  674. {
  675. throw ex;
  676. });
  677. scheduler.PeriodicStopped.WaitOne();
  678. done.WaitOne();
  679. Assert.Same(ex, err);
  680. }
  681. [Fact]
  682. public void Catch_Custom_Periodic_Caught()
  683. {
  684. var err = default(Exception);
  685. var scheduler = new MyExceptionScheduler(ex_ => err = ex_);
  686. scheduler.PeriodicStopped = new ManualResetEvent(false);
  687. var caught = new ManualResetEvent(false);
  688. var @catch = scheduler.Catch<InvalidOperationException>(_ => { caught.Set(); return true; });
  689. var per = (ISchedulerPeriodic)((IServiceProvider)@catch).GetService(typeof(ISchedulerPeriodic));
  690. per.SchedulePeriodic(42, TimeSpan.Zero, x =>
  691. {
  692. throw new InvalidOperationException();
  693. });
  694. scheduler.PeriodicStopped.WaitOne();
  695. caught.WaitOne();
  696. Assert.Null(err);
  697. }
  698. class MyExceptionScheduler : LocalScheduler, ISchedulerLongRunning, ISchedulerPeriodic
  699. {
  700. private readonly Action<Exception> _onError;
  701. public MyExceptionScheduler(Action<Exception> onError)
  702. {
  703. _onError = onError;
  704. }
  705. public override IDisposable Schedule<TState>(TState state, Func<IScheduler, TState, IDisposable> action)
  706. {
  707. try
  708. {
  709. return action(this, state);
  710. }
  711. catch (Exception exception)
  712. {
  713. _onError(exception);
  714. return Disposable.Empty;
  715. }
  716. }
  717. public override IDisposable Schedule<TState>(TState state, TimeSpan dueTime, Func<IScheduler, TState, IDisposable> action)
  718. {
  719. throw new NotImplementedException();
  720. }
  721. public IDisposable ScheduleLongRunning<TState>(TState state, Action<TState, ICancelable> action)
  722. {
  723. var b = new BooleanDisposable();
  724. try
  725. {
  726. action(state, b);
  727. }
  728. catch (Exception exception)
  729. {
  730. _onError(exception);
  731. return Disposable.Empty;
  732. }
  733. return b;
  734. }
  735. public ManualResetEvent PeriodicStopped { get; set; }
  736. public IDisposable SchedulePeriodic<TState>(TState state, TimeSpan period, Func<TState, TState> action)
  737. {
  738. var b = new BooleanDisposable();
  739. Scheduler.Default.Schedule(() =>
  740. {
  741. try
  742. {
  743. var s = state;
  744. for (int i = 0; true; i++)
  745. {
  746. if (i > 100 /* mimic delayed cancellation */ && b.IsDisposed)
  747. break;
  748. s = action(s);
  749. }
  750. }
  751. catch (Exception exception)
  752. {
  753. _onError(exception);
  754. }
  755. finally
  756. {
  757. PeriodicStopped.Set();
  758. }
  759. });
  760. return b;
  761. }
  762. }
  763. #endregion
  764. #region Services
  765. [Fact]
  766. public void InvalidService_Null()
  767. {
  768. var s = new MySchedulerWithoutServices();
  769. Assert.Null(((IServiceProvider)s).GetService(typeof(IAsyncResult)));
  770. }
  771. class MySchedulerWithoutServices : LocalScheduler
  772. {
  773. public override IDisposable Schedule<TState>(TState state, TimeSpan dueTime, Func<IScheduler, TState, IDisposable> action)
  774. {
  775. throw new NotImplementedException();
  776. }
  777. }
  778. [Fact]
  779. public void DetectServices_Null_1()
  780. {
  781. var s = new MyDumbScheduler1();
  782. Assert.Null(Scheduler.AsLongRunning(s));
  783. Assert.Null(Scheduler.AsPeriodic(s));
  784. Assert.Null(Scheduler.AsStopwatchProvider(s));
  785. }
  786. class MyDumbScheduler1 : IScheduler
  787. {
  788. public DateTimeOffset Now
  789. {
  790. get { throw new NotImplementedException(); }
  791. }
  792. public IDisposable Schedule<TState>(TState state, Func<IScheduler, TState, IDisposable> action)
  793. {
  794. throw new NotImplementedException();
  795. }
  796. public IDisposable Schedule<TState>(TState state, TimeSpan dueTime, Func<IScheduler, TState, IDisposable> action)
  797. {
  798. throw new NotImplementedException();
  799. }
  800. public IDisposable Schedule<TState>(TState state, DateTimeOffset dueTime, Func<IScheduler, TState, IDisposable> action)
  801. {
  802. throw new NotImplementedException();
  803. }
  804. }
  805. [Fact]
  806. public void DetectServices_Null_2()
  807. {
  808. var s = new MyDumbScheduler2(new Dictionary<Type, object>());
  809. Assert.Null(Scheduler.AsLongRunning(s));
  810. Assert.Null(Scheduler.AsPeriodic(s));
  811. Assert.Null(Scheduler.AsStopwatchProvider(s));
  812. }
  813. [Fact]
  814. public void DetectServices_Found()
  815. {
  816. {
  817. var x = new MyLongRunning();
  818. var s = new MyDumbScheduler2(new Dictionary<Type, object>
  819. {
  820. { typeof(ISchedulerLongRunning), x }
  821. });
  822. Assert.Equal(x, Scheduler.AsLongRunning(s));
  823. }
  824. {
  825. var x = new MyStopwatchProvider();
  826. var s = new MyDumbScheduler2(new Dictionary<Type, object>
  827. {
  828. { typeof(IStopwatchProvider), x }
  829. });
  830. Assert.Equal(x, Scheduler.AsStopwatchProvider(s));
  831. }
  832. {
  833. var x = new MyPeriodic();
  834. var s = new MyDumbScheduler2(new Dictionary<Type, object>
  835. {
  836. { typeof(ISchedulerPeriodic), x }
  837. });
  838. Assert.Equal(x, Scheduler.AsPeriodic(s));
  839. }
  840. }
  841. class MyDumbScheduler2 : IScheduler, IServiceProvider
  842. {
  843. private readonly Dictionary<Type, object> _services;
  844. public MyDumbScheduler2(Dictionary<Type, object> services)
  845. {
  846. _services = services;
  847. }
  848. public DateTimeOffset Now
  849. {
  850. get { throw new NotImplementedException(); }
  851. }
  852. public IDisposable Schedule<TState>(TState state, Func<IScheduler, TState, IDisposable> action)
  853. {
  854. throw new NotImplementedException();
  855. }
  856. public IDisposable Schedule<TState>(TState state, TimeSpan dueTime, Func<IScheduler, TState, IDisposable> action)
  857. {
  858. throw new NotImplementedException();
  859. }
  860. public IDisposable Schedule<TState>(TState state, DateTimeOffset dueTime, Func<IScheduler, TState, IDisposable> action)
  861. {
  862. throw new NotImplementedException();
  863. }
  864. public object GetService(Type serviceType)
  865. {
  866. var res = default(object);
  867. if (_services.TryGetValue(serviceType, out res))
  868. return res;
  869. return null;
  870. }
  871. }
  872. class MyLongRunning : ISchedulerLongRunning
  873. {
  874. public IDisposable ScheduleLongRunning<TState>(TState state, Action<TState, ICancelable> action)
  875. {
  876. throw new NotImplementedException();
  877. }
  878. }
  879. class MyStopwatchProvider : IStopwatchProvider
  880. {
  881. public IStopwatch StartStopwatch()
  882. {
  883. throw new NotImplementedException();
  884. }
  885. }
  886. class MyPeriodic : ISchedulerPeriodic
  887. {
  888. public IDisposable SchedulePeriodic<TState>(TState state, TimeSpan period, Func<TState, TState> action)
  889. {
  890. throw new NotImplementedException();
  891. }
  892. }
  893. #endregion
  894. #if HAS_AWAIT
  895. [Fact]
  896. public void SchedulerAsync_Yield_ArgumentChecking()
  897. {
  898. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.Yield(default(IScheduler)));
  899. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.Yield(default(IScheduler), CancellationToken.None));
  900. }
  901. [Fact]
  902. public void SchedulerAsync_Sleep_ArgumentChecking()
  903. {
  904. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.Sleep(default(IScheduler), TimeSpan.Zero));
  905. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.Sleep(default(IScheduler), TimeSpan.Zero, CancellationToken.None));
  906. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.Sleep(default(IScheduler), DateTimeOffset.MinValue));
  907. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.Sleep(default(IScheduler), DateTimeOffset.MinValue, CancellationToken.None));
  908. }
  909. [Fact]
  910. public void SchedulerAsync_ScheduleAsync_ArgumentChecking()
  911. {
  912. var tcs = new TaskCompletionSource<IDisposable>();
  913. var t = tcs.Task;
  914. var d = default(IScheduler);
  915. var s = Scheduler.Immediate;
  916. var rt = TimeSpan.Zero;
  917. var at = DateTimeOffset.MinValue;
  918. var a1 = new Func<IScheduler, int, CancellationToken, Task>((_, __, ___) => t);
  919. var d1 = default(Func<IScheduler, int, CancellationToken, Task>);
  920. var a2 = new Func<IScheduler, int, CancellationToken, Task<IDisposable>>((_, __, ___) => t);
  921. var d2 = default(Func<IScheduler, int, CancellationToken, Task<IDisposable>>);
  922. var a3 = new Func<IScheduler, CancellationToken, Task>((_, __) => t);
  923. var d3 = default(Func<IScheduler, CancellationToken, Task>);
  924. var a4 = new Func<IScheduler, CancellationToken, Task<IDisposable>>((_, __) => t);
  925. var d4 = default(Func<IScheduler, CancellationToken, Task<IDisposable>>);
  926. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.ScheduleAsync(d, 42, a1));
  927. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.ScheduleAsync(s, 42, d1));
  928. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.ScheduleAsync(d, 42, rt, a1));
  929. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.ScheduleAsync(s, 42, rt, d1));
  930. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.ScheduleAsync(d, 42, at, a1));
  931. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.ScheduleAsync(s, 42, at, d1));
  932. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.ScheduleAsync(d, 42, a2));
  933. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.ScheduleAsync(s, 42, d2));
  934. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.ScheduleAsync(d, 42, rt, a2));
  935. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.ScheduleAsync(s, 42, rt, d2));
  936. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.ScheduleAsync(d, 42, at, a2));
  937. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.ScheduleAsync(s, 42, at, d2));
  938. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.ScheduleAsync(d, a3));
  939. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.ScheduleAsync(s, d3));
  940. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.ScheduleAsync(d, rt, a3));
  941. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.ScheduleAsync(s, rt, d3));
  942. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.ScheduleAsync(d, at, a3));
  943. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.ScheduleAsync(s, at, d3));
  944. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.ScheduleAsync(d, a4));
  945. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.ScheduleAsync(s, d4));
  946. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.ScheduleAsync(d, rt, a4));
  947. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.ScheduleAsync(s, rt, d4));
  948. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.ScheduleAsync(d, at, a4));
  949. ReactiveAssert.Throws<ArgumentNullException>(() => Scheduler.ScheduleAsync(s, at, d4));
  950. }
  951. [Fact]
  952. public void SchedulerAsync_ScheduleAsync_Overloads1()
  953. {
  954. var tcsI = new TaskCompletionSource<int>();
  955. var t = tcsI.Task;
  956. tcsI.SetResult(0);
  957. var tcsD = new TaskCompletionSource<IDisposable>();
  958. var d = tcsD.Task;
  959. tcsD.SetResult(Disposable.Empty);
  960. var s = new TestScheduler();
  961. var o = s.CreateObserver<int>();
  962. s.ScheduleAsync((_, ct) =>
  963. {
  964. o.OnNext(42);
  965. return t;
  966. });
  967. s.ScheduleAsync((_, ct) =>
  968. {
  969. o.OnNext(43);
  970. return d;
  971. });
  972. s.ScheduleAsync(44, (_, x, ct) =>
  973. {
  974. o.OnNext(x);
  975. return t;
  976. });
  977. s.ScheduleAsync(45, (_, x, ct) =>
  978. {
  979. o.OnNext(45);
  980. return d;
  981. });
  982. s.Start();
  983. o.Messages.AssertEqual(
  984. OnNext(1, 42),
  985. OnNext(1, 43),
  986. OnNext(1, 44),
  987. OnNext(1, 45)
  988. );
  989. }
  990. [Fact]
  991. public void SchedulerAsync_ScheduleAsync_Overloads2()
  992. {
  993. var tcsI = new TaskCompletionSource<int>();
  994. var t = tcsI.Task;
  995. tcsI.SetResult(0);
  996. var tcsD = new TaskCompletionSource<IDisposable>();
  997. var d = tcsD.Task;
  998. tcsD.SetResult(Disposable.Empty);
  999. var s = new TestScheduler();
  1000. var o = s.CreateObserver<int>();
  1001. s.ScheduleAsync(TimeSpan.FromTicks(50), (_, ct) =>
  1002. {
  1003. o.OnNext(42);
  1004. return t;
  1005. });
  1006. s.ScheduleAsync(TimeSpan.FromTicks(60), (_, ct) =>
  1007. {
  1008. o.OnNext(43);
  1009. return d;
  1010. });
  1011. s.ScheduleAsync(44, TimeSpan.FromTicks(70), (_, x, ct) =>
  1012. {
  1013. o.OnNext(x);
  1014. return t;
  1015. });
  1016. s.ScheduleAsync(45, TimeSpan.FromTicks(80), (_, x, ct) =>
  1017. {
  1018. o.OnNext(45);
  1019. return d;
  1020. });
  1021. s.Start();
  1022. o.Messages.AssertEqual(
  1023. OnNext(50, 42),
  1024. OnNext(60, 43),
  1025. OnNext(70, 44),
  1026. OnNext(80, 45)
  1027. );
  1028. }
  1029. [Fact]
  1030. public void SchedulerAsync_ScheduleAsync_Overloads3()
  1031. {
  1032. var tcsI = new TaskCompletionSource<int>();
  1033. var t = tcsI.Task;
  1034. tcsI.SetResult(0);
  1035. var tcsD = new TaskCompletionSource<IDisposable>();
  1036. var d = tcsD.Task;
  1037. tcsD.SetResult(Disposable.Empty);
  1038. var s = new TestScheduler();
  1039. var o = s.CreateObserver<int>();
  1040. s.ScheduleAsync(new DateTimeOffset(50, TimeSpan.Zero), (_, ct) =>
  1041. {
  1042. o.OnNext(42);
  1043. return t;
  1044. });
  1045. s.ScheduleAsync(new DateTimeOffset(60, TimeSpan.Zero), (_, ct) =>
  1046. {
  1047. o.OnNext(43);
  1048. return d;
  1049. });
  1050. s.ScheduleAsync(44, new DateTimeOffset(70, TimeSpan.Zero), (_, x, ct) =>
  1051. {
  1052. o.OnNext(x);
  1053. return t;
  1054. });
  1055. s.ScheduleAsync(45, new DateTimeOffset(80, TimeSpan.Zero), (_, x, ct) =>
  1056. {
  1057. o.OnNext(45);
  1058. return d;
  1059. });
  1060. s.Start();
  1061. o.Messages.AssertEqual(
  1062. OnNext(50, 42),
  1063. OnNext(60, 43),
  1064. OnNext(70, 44),
  1065. OnNext(80, 45)
  1066. );
  1067. }
  1068. [Fact]
  1069. public void SchedulerAsync_ScheduleAsync_NoCancellation1()
  1070. {
  1071. var s = new TestScheduler();
  1072. var o = s.CreateObserver<int>();
  1073. s.ScheduleAsync(async (_, ct) =>
  1074. {
  1075. o.OnNext(42);
  1076. await _.Yield();
  1077. o.OnNext(43);
  1078. await _.Sleep(TimeSpan.FromTicks(10));
  1079. o.OnNext(44);
  1080. await _.Sleep(new DateTimeOffset(250, TimeSpan.Zero));
  1081. o.OnNext(45);
  1082. });
  1083. s.Start();
  1084. o.Messages.AssertEqual(
  1085. OnNext(1, 42),
  1086. OnNext(2, 43),
  1087. OnNext(12, 44),
  1088. OnNext(250, 45)
  1089. );
  1090. }
  1091. [Fact]
  1092. public void SchedulerAsync_ScheduleAsync_NoCancellation2()
  1093. {
  1094. var s = new TestScheduler();
  1095. var o = s.CreateObserver<int>();
  1096. s.ScheduleAsync(async (_, ct) =>
  1097. {
  1098. o.OnNext(42);
  1099. await _.Yield(ct);
  1100. o.OnNext(43);
  1101. await _.Sleep(TimeSpan.FromTicks(10), ct);
  1102. o.OnNext(44);
  1103. await _.Sleep(new DateTimeOffset(250, TimeSpan.Zero), ct);
  1104. o.OnNext(45);
  1105. });
  1106. s.Start();
  1107. o.Messages.AssertEqual(
  1108. OnNext(1, 42),
  1109. OnNext(2, 43),
  1110. OnNext(12, 44),
  1111. OnNext(250, 45)
  1112. );
  1113. }
  1114. [Fact]
  1115. public void SchedulerAsync_Awaiters()
  1116. {
  1117. var op = Scheduler.Immediate.Yield();
  1118. var aw = op.GetAwaiter();
  1119. ReactiveAssert.Throws<ArgumentNullException>(() => aw.OnCompleted(null));
  1120. aw.OnCompleted(() => { });
  1121. ReactiveAssert.Throws<InvalidOperationException>(() => aw.OnCompleted(() => { }));
  1122. }
  1123. [Fact]
  1124. public void SchedulerAsync_Yield_Cancel1()
  1125. {
  1126. var cts = new CancellationTokenSource();
  1127. var op = Scheduler.Immediate.Yield(cts.Token);
  1128. var aw = op.GetAwaiter();
  1129. cts.Cancel();
  1130. Assert.True(aw.IsCompleted);
  1131. ReactiveAssert.Throws<OperationCanceledException>(() => aw.GetResult());
  1132. }
  1133. [Fact]
  1134. public void SchedulerAsync_Yield_Cancel2()
  1135. {
  1136. var cts = new CancellationTokenSource();
  1137. var op = Scheduler.Immediate.Yield(cts.Token);
  1138. var aw = op.GetAwaiter();
  1139. Assert.False(aw.IsCompleted);
  1140. aw.OnCompleted(() =>
  1141. {
  1142. //
  1143. // TODO: Not needed for await pattern, but maybe should be wired up?
  1144. //
  1145. // Assert.True(aw.IsCompleted);
  1146. cts.Cancel();
  1147. ReactiveAssert.Throws<OperationCanceledException>(() => aw.GetResult());
  1148. });
  1149. }
  1150. [Fact]
  1151. public void SchedulerAsync_Sleep_Cancel()
  1152. {
  1153. var cts = new CancellationTokenSource();
  1154. var op = Scheduler.Default.Sleep(TimeSpan.FromHours(1), cts.Token);
  1155. var aw = op.GetAwaiter();
  1156. Assert.False(aw.IsCompleted);
  1157. var e = new ManualResetEvent(false);
  1158. aw.OnCompleted(() =>
  1159. {
  1160. ReactiveAssert.Throws<OperationCanceledException>(() => aw.GetResult());
  1161. e.Set();
  1162. });
  1163. cts.Cancel();
  1164. e.WaitOne();
  1165. }
  1166. #if !NO_SYNCCTX
  1167. [Fact]
  1168. public void SchedulerAsync_ScheduleAsync_SyncCtx()
  1169. {
  1170. var old = SynchronizationContext.Current;
  1171. try
  1172. {
  1173. var ctx = new MySyncCtx();
  1174. SynchronizationContext.SetSynchronizationContext(ctx);
  1175. var s = new TestScheduler();
  1176. var o = s.CreateObserver<int>();
  1177. s.ScheduleAsync(async (_, ct) =>
  1178. {
  1179. Assert.Same(ctx, SynchronizationContext.Current);
  1180. o.OnNext(42);
  1181. await _.Yield(ct).ConfigureAwait(true);
  1182. Assert.Same(ctx, SynchronizationContext.Current);
  1183. o.OnNext(43);
  1184. await _.Sleep(TimeSpan.FromTicks(10), ct).ConfigureAwait(true);
  1185. Assert.Same(ctx, SynchronizationContext.Current);
  1186. o.OnNext(44);
  1187. await _.Sleep(new DateTimeOffset(250, TimeSpan.Zero), ct).ConfigureAwait(true);
  1188. Assert.Same(ctx, SynchronizationContext.Current);
  1189. o.OnNext(45);
  1190. });
  1191. s.Start();
  1192. o.Messages.AssertEqual(
  1193. OnNext(1, 42),
  1194. OnNext(2, 43),
  1195. OnNext(12, 44),
  1196. OnNext(250, 45)
  1197. );
  1198. }
  1199. finally
  1200. {
  1201. SynchronizationContext.SetSynchronizationContext(old);
  1202. }
  1203. }
  1204. class MySyncCtx : SynchronizationContext
  1205. {
  1206. public override void Post(SendOrPostCallback d, object state)
  1207. {
  1208. d(state);
  1209. }
  1210. }
  1211. #endif
  1212. #endif
  1213. }
  1214. }
  1215. #endif