SkipWhileTest.cs 13 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514
  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.Linq;
  6. using System.Reactive.Linq;
  7. using Microsoft.Reactive.Testing;
  8. using ReactiveTests.Dummies;
  9. using Xunit;
  10. namespace ReactiveTests.Tests
  11. {
  12. public class SkipWhileTest : ReactiveTest
  13. {
  14. [Fact]
  15. public void SkipWhile_ArgumentChecking()
  16. {
  17. ReactiveAssert.Throws<ArgumentNullException>(() => ((IObservable<int>)null).SkipWhile(DummyFunc<int, bool>.Instance));
  18. ReactiveAssert.Throws<ArgumentNullException>(() => DummyObservable<int>.Instance.SkipWhile(default(Func<int, bool>)));
  19. ReactiveAssert.Throws<ArgumentNullException>(() => DummyObservable<int>.Instance.SkipWhile(DummyFunc<int, bool>.Instance).Subscribe(null));
  20. ReactiveAssert.Throws<ArgumentNullException>(() => ((IObservable<int>)null).SkipWhile((x, i) => true));
  21. ReactiveAssert.Throws<ArgumentNullException>(() => DummyObservable<int>.Instance.SkipWhile(default(Func<int, int, bool>)));
  22. }
  23. [Fact]
  24. public void SkipWhile_Complete_Before()
  25. {
  26. var scheduler = new TestScheduler();
  27. var xs = scheduler.CreateHotObservable(
  28. OnNext(90, -1),
  29. OnNext(110, -1),
  30. OnNext(210, 2),
  31. OnNext(260, 5),
  32. OnNext(290, 13),
  33. OnNext(320, 3),
  34. OnCompleted<int>(330),
  35. OnNext(350, 7),
  36. OnNext(390, 4),
  37. OnNext(410, 17),
  38. OnNext(450, 8),
  39. OnNext(500, 23),
  40. OnCompleted<int>(600)
  41. );
  42. var invoked = 0;
  43. var res = scheduler.Start(() =>
  44. xs.SkipWhile(x =>
  45. {
  46. invoked++;
  47. return IsPrime(x);
  48. })
  49. );
  50. res.Messages.AssertEqual(
  51. OnCompleted<int>(330)
  52. );
  53. xs.Subscriptions.AssertEqual(
  54. Subscribe(200, 330)
  55. );
  56. Assert.Equal(4, invoked);
  57. }
  58. [Fact]
  59. public void SkipWhile_Complete_After()
  60. {
  61. var scheduler = new TestScheduler();
  62. var xs = scheduler.CreateHotObservable(
  63. OnNext(90, -1),
  64. OnNext(110, -1),
  65. OnNext(210, 2),
  66. OnNext(260, 5),
  67. OnNext(290, 13),
  68. OnNext(320, 3),
  69. OnNext(350, 7),
  70. OnNext(390, 4),
  71. OnNext(410, 17),
  72. OnNext(450, 8),
  73. OnNext(500, 23),
  74. OnCompleted<int>(600)
  75. );
  76. var invoked = 0;
  77. var res = scheduler.Start(() =>
  78. xs.SkipWhile(x =>
  79. {
  80. invoked++;
  81. return IsPrime(x);
  82. })
  83. );
  84. res.Messages.AssertEqual(
  85. OnNext(390, 4),
  86. OnNext(410, 17),
  87. OnNext(450, 8),
  88. OnNext(500, 23),
  89. OnCompleted<int>(600)
  90. );
  91. xs.Subscriptions.AssertEqual(
  92. Subscribe(200, 600)
  93. );
  94. Assert.Equal(6, invoked);
  95. }
  96. [Fact]
  97. public void SkipWhile_Error_Before()
  98. {
  99. var scheduler = new TestScheduler();
  100. var ex = new Exception();
  101. var xs = scheduler.CreateHotObservable(
  102. OnNext(90, -1),
  103. OnNext(110, -1),
  104. OnNext(210, 2),
  105. OnNext(260, 5),
  106. OnError<int>(270, ex),
  107. OnNext(290, 13),
  108. OnNext(320, 3),
  109. OnNext(350, 7),
  110. OnNext(390, 4),
  111. OnNext(410, 17),
  112. OnNext(450, 8),
  113. OnNext(500, 23)
  114. );
  115. var invoked = 0;
  116. var res = scheduler.Start(() =>
  117. xs.SkipWhile(x =>
  118. {
  119. invoked++;
  120. return IsPrime(x);
  121. })
  122. );
  123. res.Messages.AssertEqual(
  124. OnError<int>(270, ex)
  125. );
  126. xs.Subscriptions.AssertEqual(
  127. Subscribe(200, 270)
  128. );
  129. Assert.Equal(2, invoked);
  130. }
  131. [Fact]
  132. public void SkipWhile_Error_After()
  133. {
  134. var scheduler = new TestScheduler();
  135. var ex = new Exception();
  136. var xs = scheduler.CreateHotObservable(
  137. OnNext(90, -1),
  138. OnNext(110, -1),
  139. OnNext(210, 2),
  140. OnNext(260, 5),
  141. OnNext(290, 13),
  142. OnNext(320, 3),
  143. OnNext(350, 7),
  144. OnNext(390, 4),
  145. OnNext(410, 17),
  146. OnNext(450, 8),
  147. OnNext(500, 23),
  148. OnError<int>(600, ex)
  149. );
  150. var invoked = 0;
  151. var res = scheduler.Start(() =>
  152. xs.SkipWhile(x =>
  153. {
  154. invoked++;
  155. return IsPrime(x);
  156. })
  157. );
  158. res.Messages.AssertEqual(
  159. OnNext(390, 4),
  160. OnNext(410, 17),
  161. OnNext(450, 8),
  162. OnNext(500, 23),
  163. OnError<int>(600, ex)
  164. );
  165. xs.Subscriptions.AssertEqual(
  166. Subscribe(200, 600)
  167. );
  168. Assert.Equal(6, invoked);
  169. }
  170. [Fact]
  171. public void SkipWhile_Dispose_Before()
  172. {
  173. var scheduler = new TestScheduler();
  174. var xs = scheduler.CreateHotObservable(
  175. OnNext(90, -1),
  176. OnNext(110, -1),
  177. OnNext(210, 2),
  178. OnNext(260, 5),
  179. OnNext(290, 13),
  180. OnNext(320, 3),
  181. OnNext(350, 7),
  182. OnNext(390, 4),
  183. OnNext(410, 17),
  184. OnNext(450, 8),
  185. OnNext(500, 23),
  186. OnCompleted<int>(600)
  187. );
  188. var invoked = 0;
  189. var res = scheduler.Start(() =>
  190. xs.SkipWhile(x =>
  191. {
  192. invoked++;
  193. return IsPrime(x);
  194. }),
  195. 300
  196. );
  197. res.Messages.AssertEqual(
  198. );
  199. xs.Subscriptions.AssertEqual(
  200. Subscribe(200, 300)
  201. );
  202. Assert.Equal(3, invoked);
  203. }
  204. [Fact]
  205. public void SkipWhile_Dispose_After()
  206. {
  207. var scheduler = new TestScheduler();
  208. var xs = scheduler.CreateHotObservable(
  209. OnNext(90, -1),
  210. OnNext(110, -1),
  211. OnNext(210, 2),
  212. OnNext(260, 5),
  213. OnNext(290, 13),
  214. OnNext(320, 3),
  215. OnNext(350, 7),
  216. OnNext(390, 4),
  217. OnNext(410, 17),
  218. OnNext(450, 8),
  219. OnNext(500, 23),
  220. OnCompleted<int>(600)
  221. );
  222. var invoked = 0;
  223. var res = scheduler.Start(() =>
  224. xs.SkipWhile(x =>
  225. {
  226. invoked++;
  227. return IsPrime(x);
  228. }),
  229. 470
  230. );
  231. res.Messages.AssertEqual(
  232. OnNext(390, 4),
  233. OnNext(410, 17),
  234. OnNext(450, 8)
  235. );
  236. xs.Subscriptions.AssertEqual(
  237. Subscribe(200, 470)
  238. );
  239. Assert.Equal(6, invoked);
  240. }
  241. [Fact]
  242. public void SkipWhile_Zero()
  243. {
  244. var scheduler = new TestScheduler();
  245. var xs = scheduler.CreateHotObservable(
  246. OnNext(90, -1),
  247. OnNext(110, -1),
  248. OnNext(205, 100),
  249. OnNext(210, 2),
  250. OnNext(260, 5),
  251. OnNext(290, 13),
  252. OnNext(320, 3),
  253. OnNext(350, 7),
  254. OnNext(390, 4),
  255. OnNext(410, 17),
  256. OnNext(450, 8),
  257. OnNext(500, 23),
  258. OnCompleted<int>(600)
  259. );
  260. var invoked = 0;
  261. var res = scheduler.Start(() =>
  262. xs.SkipWhile(x =>
  263. {
  264. invoked++;
  265. return IsPrime(x);
  266. })
  267. );
  268. res.Messages.AssertEqual(
  269. OnNext(205, 100),
  270. OnNext(210, 2),
  271. OnNext(260, 5),
  272. OnNext(290, 13),
  273. OnNext(320, 3),
  274. OnNext(350, 7),
  275. OnNext(390, 4),
  276. OnNext(410, 17),
  277. OnNext(450, 8),
  278. OnNext(500, 23),
  279. OnCompleted<int>(600)
  280. );
  281. xs.Subscriptions.AssertEqual(
  282. Subscribe(200, 600)
  283. );
  284. Assert.Equal(1, invoked);
  285. }
  286. [Fact]
  287. public void SkipWhile_Throw()
  288. {
  289. var scheduler = new TestScheduler();
  290. var xs = scheduler.CreateHotObservable(
  291. OnNext(90, -1),
  292. OnNext(110, -1),
  293. OnNext(210, 2),
  294. OnNext(260, 5),
  295. OnNext(290, 13),
  296. OnNext(320, 3),
  297. OnNext(350, 7),
  298. OnNext(390, 4),
  299. OnNext(410, 17),
  300. OnNext(450, 8),
  301. OnNext(500, 23),
  302. OnCompleted<int>(600)
  303. );
  304. var invoked = 0;
  305. var ex = new Exception();
  306. var res = scheduler.Start(() =>
  307. xs.SkipWhile(x =>
  308. {
  309. invoked++;
  310. if (invoked == 3)
  311. {
  312. throw ex;
  313. }
  314. return IsPrime(x);
  315. })
  316. );
  317. res.Messages.AssertEqual(
  318. OnError<int>(290, ex)
  319. );
  320. xs.Subscriptions.AssertEqual(
  321. Subscribe(200, 290)
  322. );
  323. Assert.Equal(3, invoked);
  324. }
  325. [Fact]
  326. public void SkipWhile_Index()
  327. {
  328. var scheduler = new TestScheduler();
  329. var xs = scheduler.CreateHotObservable(
  330. OnNext(90, -1),
  331. OnNext(110, -1),
  332. OnNext(205, 100),
  333. OnNext(210, 2),
  334. OnNext(260, 5),
  335. OnNext(290, 13),
  336. OnNext(320, 3),
  337. OnNext(350, 7),
  338. OnNext(390, 4),
  339. OnNext(410, 17),
  340. OnNext(450, 8),
  341. OnNext(500, 23),
  342. OnCompleted<int>(600)
  343. );
  344. var res = scheduler.Start(() =>
  345. xs.SkipWhile((x, i) => i < 5)
  346. );
  347. res.Messages.AssertEqual(
  348. OnNext(350, 7),
  349. OnNext(390, 4),
  350. OnNext(410, 17),
  351. OnNext(450, 8),
  352. OnNext(500, 23),
  353. OnCompleted<int>(600)
  354. );
  355. xs.Subscriptions.AssertEqual(
  356. Subscribe(200, 600)
  357. );
  358. }
  359. [Fact]
  360. public void SkipWhile_Index_Throw()
  361. {
  362. var scheduler = new TestScheduler();
  363. var ex = new Exception();
  364. var xs = scheduler.CreateHotObservable(
  365. OnNext(90, -1),
  366. OnNext(110, -1),
  367. OnNext(205, 100),
  368. OnNext(210, 2),
  369. OnNext(260, 5),
  370. OnNext(290, 13),
  371. OnNext(320, 3),
  372. OnNext(350, 7),
  373. OnNext(390, 4),
  374. OnError<int>(400, ex)
  375. );
  376. var res = scheduler.Start(() =>
  377. xs.SkipWhile((x, i) => i < 5)
  378. );
  379. res.Messages.AssertEqual(
  380. OnNext(350, 7),
  381. OnNext(390, 4),
  382. OnError<int>(400, ex)
  383. );
  384. xs.Subscriptions.AssertEqual(
  385. Subscribe(200, 400)
  386. );
  387. }
  388. [Fact]
  389. public void SkipWhile_Index_SelectorThrows()
  390. {
  391. var scheduler = new TestScheduler();
  392. var ex = new Exception();
  393. var xs = scheduler.CreateHotObservable(
  394. OnNext(90, -1),
  395. OnNext(110, -1),
  396. OnNext(205, 100),
  397. OnNext(210, 2),
  398. OnNext(260, 5),
  399. OnNext(290, 13),
  400. OnNext(320, 3),
  401. OnNext(350, 7),
  402. OnNext(390, 4),
  403. OnCompleted<int>(400)
  404. );
  405. var res = scheduler.Start(() =>
  406. xs.SkipWhile((x, i) => { if (i < 5) { return true; } throw ex; })
  407. );
  408. res.Messages.AssertEqual(
  409. OnError<int>(350, ex)
  410. );
  411. xs.Subscriptions.AssertEqual(
  412. Subscribe(200, 350)
  413. );
  414. }
  415. private static bool IsPrime(int i)
  416. {
  417. if (i <= 1)
  418. {
  419. return false;
  420. }
  421. var max = (int)Math.Sqrt(i);
  422. for (var j = 2; j <= max; ++j)
  423. {
  424. if (i % j == 0)
  425. {
  426. return false;
  427. }
  428. }
  429. return true;
  430. }
  431. }
  432. }