AsyncTests.Multiple.cs 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819
  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.Collections.Generic;
  6. using System.Diagnostics;
  7. using System.Linq;
  8. using System.Threading;
  9. using System.Threading.Tasks;
  10. using Xunit;
  11. namespace Tests
  12. {
  13. public partial class AsyncTests
  14. {
  15. [Fact]
  16. public void Concat_Null()
  17. {
  18. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.Concat<int>(null, AsyncEnumerable.Return(42)));
  19. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.Concat<int>(AsyncEnumerable.Return(42), null));
  20. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.Concat<int>(default(IAsyncEnumerable<int>[])));
  21. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.Concat<int>(default(IEnumerable<IAsyncEnumerable<int>>)));
  22. }
  23. [Fact]
  24. public void Concat1()
  25. {
  26. var ys = new[] { 1, 2, 3 }.ToAsyncEnumerable().Concat(new[] { 4, 5, 6 }.ToAsyncEnumerable());
  27. var e = ys.GetAsyncEnumerator();
  28. HasNext(e, 1);
  29. HasNext(e, 2);
  30. HasNext(e, 3);
  31. HasNext(e, 4);
  32. HasNext(e, 5);
  33. HasNext(e, 6);
  34. NoNext(e);
  35. }
  36. [Fact]
  37. public void Concat2()
  38. {
  39. var ex = new Exception("Bang");
  40. var ys = new[] { 1, 2, 3 }.ToAsyncEnumerable().Concat(AsyncEnumerable.Throw<int>(ex));
  41. var e = ys.GetAsyncEnumerator();
  42. HasNext(e, 1);
  43. HasNext(e, 2);
  44. HasNext(e, 3);
  45. AssertThrows(() => e.MoveNextAsync().Wait(WaitTimeoutMs), (Exception ex_) => ((AggregateException)ex_).Flatten().InnerExceptions.Single() == ex);
  46. }
  47. [Fact]
  48. public void Concat3()
  49. {
  50. var ex = new Exception("Bang");
  51. var ys = AsyncEnumerable.Throw<int>(ex).Concat(new[] { 4, 5, 6 }.ToAsyncEnumerable());
  52. var e = ys.GetAsyncEnumerator();
  53. AssertThrows(() => e.MoveNextAsync().Wait(WaitTimeoutMs), (Exception ex_) => ((AggregateException)ex_).Flatten().InnerExceptions.Single() == ex);
  54. }
  55. [Fact]
  56. public void Concat4()
  57. {
  58. var xs = new[] { 1, 2, 3 }.ToAsyncEnumerable();
  59. var ys = new[] { 4, 5 }.ToAsyncEnumerable();
  60. var zs = new[] { 6, 7, 8 }.ToAsyncEnumerable();
  61. var res = AsyncEnumerable.Concat(xs, ys, zs);
  62. var e = res.GetAsyncEnumerator();
  63. HasNext(e, 1);
  64. HasNext(e, 2);
  65. HasNext(e, 3);
  66. HasNext(e, 4);
  67. HasNext(e, 5);
  68. HasNext(e, 6);
  69. HasNext(e, 7);
  70. HasNext(e, 8);
  71. NoNext(e);
  72. }
  73. [Fact]
  74. public void Concat5()
  75. {
  76. var ex = new Exception("Bang");
  77. var xs = new[] { 1, 2, 3 }.ToAsyncEnumerable();
  78. var ys = new[] { 4, 5 }.ToAsyncEnumerable();
  79. var zs = AsyncEnumerable.Throw<int>(ex);
  80. var res = AsyncEnumerable.Concat(xs, ys, zs);
  81. var e = res.GetAsyncEnumerator();
  82. HasNext(e, 1);
  83. HasNext(e, 2);
  84. HasNext(e, 3);
  85. HasNext(e, 4);
  86. HasNext(e, 5);
  87. AssertThrows(() => e.MoveNextAsync().Wait(WaitTimeoutMs), (Exception ex_) => ((AggregateException)ex_).Flatten().InnerExceptions.Single() == ex);
  88. }
  89. [Fact]
  90. public void Concat6()
  91. {
  92. var res = AsyncEnumerable.Concat(ConcatXss());
  93. var e = res.GetAsyncEnumerator();
  94. HasNext(e, 1);
  95. HasNext(e, 2);
  96. HasNext(e, 3);
  97. HasNext(e, 4);
  98. HasNext(e, 5);
  99. AssertThrows(() => e.MoveNextAsync().Wait(WaitTimeoutMs), (Exception ex_) => ((AggregateException)ex_).Flatten().InnerExceptions.Single().Message == "Bang!");
  100. }
  101. [Fact]
  102. public void Concat7()
  103. {
  104. var ws = new[] { 1, 2, 3 }.ToAsyncEnumerable();
  105. var xs = new[] { 4, 5 }.ToAsyncEnumerable();
  106. var ys = new[] { 6, 7, 8 }.ToAsyncEnumerable();
  107. var zs = new[] { 9, 10, 11 }.ToAsyncEnumerable();
  108. var res = ws.Concat(xs).Concat(ys).Concat(zs);
  109. var e = res.GetAsyncEnumerator();
  110. HasNext(e, 1);
  111. HasNext(e, 2);
  112. HasNext(e, 3);
  113. HasNext(e, 4);
  114. HasNext(e, 5);
  115. HasNext(e, 6);
  116. HasNext(e, 7);
  117. HasNext(e, 8);
  118. HasNext(e, 9);
  119. HasNext(e, 10);
  120. HasNext(e, 11);
  121. NoNext(e);
  122. }
  123. [Fact]
  124. public async Task Concat8()
  125. {
  126. var ws = new[] { 1, 2, 3 }.ToAsyncEnumerable();
  127. var xs = new[] { 4, 5 }.ToAsyncEnumerable();
  128. var ys = new[] { 6, 7, 8 }.ToAsyncEnumerable();
  129. var zs = new[] { 9, 10, 11 }.ToAsyncEnumerable();
  130. var res = ws.Concat(xs).Concat(ys).Concat(zs);
  131. await SequenceIdentity(res);
  132. }
  133. [Fact]
  134. public async Task Concat9()
  135. {
  136. var xs = new[] { 1, 2, 3 }.ToAsyncEnumerable();
  137. var ys = new[] { 4, 5 }.ToAsyncEnumerable();
  138. var zs = new[] { 6, 7, 8 }.ToAsyncEnumerable();
  139. var res = AsyncEnumerable.Concat(xs, ys, zs);
  140. await SequenceIdentity(res);
  141. }
  142. [Fact]
  143. public async Task Concat10()
  144. {
  145. var xs = new[] { 1, 2, 3 }.ToAsyncEnumerable();
  146. var ys = new[] { 4, 5 }.ToAsyncEnumerable();
  147. var zs = new[] { 6, 7, 8 }.ToAsyncEnumerable();
  148. var c = xs.Concat(ys).Concat(zs);
  149. var res = new[] { 1, 2, 3, 4, 5, 6, 7, 8 };
  150. Assert.True(res.SequenceEqual(await c.ToArray()));
  151. }
  152. [Fact]
  153. public async Task Concat11()
  154. {
  155. var xs = new[] { 1, 2, 3 }.ToAsyncEnumerable();
  156. var ys = new[] { 4, 5 }.ToAsyncEnumerable();
  157. var zs = new[] { 6, 7, 8 }.ToAsyncEnumerable();
  158. var c = xs.Concat(ys).Concat(zs);
  159. var res = new List<int> { 1, 2, 3, 4, 5, 6, 7, 8 };
  160. Assert.True(res.SequenceEqual(await c.ToList()));
  161. }
  162. [Fact]
  163. public async Task Concat12()
  164. {
  165. var xs = new[] { 1, 2, 3 }.ToAsyncEnumerable();
  166. var ys = new[] { 4, 5 }.ToAsyncEnumerable();
  167. var zs = new[] { 6, 7, 8 }.ToAsyncEnumerable();
  168. var c = xs.Concat(ys).Concat(zs);
  169. Assert.Equal(8, await c.Count());
  170. }
  171. static IEnumerable<IAsyncEnumerable<int>> ConcatXss()
  172. {
  173. yield return new[] { 1, 2, 3 }.ToAsyncEnumerable();
  174. yield return new[] { 4, 5 }.ToAsyncEnumerable();
  175. throw new Exception("Bang!");
  176. }
  177. [Fact]
  178. public async Task SequenceEqual_Null()
  179. {
  180. await Assert.ThrowsAsync<ArgumentNullException>(() => AsyncEnumerable.SequenceEqual<int>(null, AsyncEnumerable.Return(42)));
  181. await Assert.ThrowsAsync<ArgumentNullException>(() => AsyncEnumerable.SequenceEqual<int>(AsyncEnumerable.Return(42), null));
  182. await Assert.ThrowsAsync<ArgumentNullException>(() => AsyncEnumerable.SequenceEqual<int>(null, AsyncEnumerable.Return(42), new Eq()));
  183. await Assert.ThrowsAsync<ArgumentNullException>(() => AsyncEnumerable.SequenceEqual<int>(AsyncEnumerable.Return(42), null, new Eq()));
  184. await Assert.ThrowsAsync<ArgumentNullException>(() => AsyncEnumerable.SequenceEqual<int>(AsyncEnumerable.Return(42), AsyncEnumerable.Return(42), null));
  185. await Assert.ThrowsAsync<ArgumentNullException>(() => AsyncEnumerable.SequenceEqual<int>(null, AsyncEnumerable.Return(42), CancellationToken.None));
  186. await Assert.ThrowsAsync<ArgumentNullException>(() => AsyncEnumerable.SequenceEqual<int>(AsyncEnumerable.Return(42), null, CancellationToken.None));
  187. await Assert.ThrowsAsync<ArgumentNullException>(() => AsyncEnumerable.SequenceEqual<int>(null, AsyncEnumerable.Return(42), new Eq(), CancellationToken.None));
  188. await Assert.ThrowsAsync<ArgumentNullException>(() => AsyncEnumerable.SequenceEqual<int>(AsyncEnumerable.Return(42), null, new Eq(), CancellationToken.None));
  189. await Assert.ThrowsAsync<ArgumentNullException>(() => AsyncEnumerable.SequenceEqual<int>(AsyncEnumerable.Return(42), AsyncEnumerable.Return(42), null, CancellationToken.None));
  190. }
  191. [Fact]
  192. public void SequenceEqual1()
  193. {
  194. var xs = new[] { 1, 2, 3 }.ToAsyncEnumerable();
  195. var res = xs.SequenceEqual(xs);
  196. Assert.True(res.Result);
  197. }
  198. [Fact]
  199. public void SequenceEqual2()
  200. {
  201. var xs = AsyncEnumerable.Empty<int>();
  202. var res = xs.SequenceEqual(xs);
  203. Assert.True(res.Result);
  204. }
  205. [Fact]
  206. public void SequenceEqual3()
  207. {
  208. var xs = new[] { 1, 2, 3 }.ToAsyncEnumerable();
  209. var ys = new[] { 1, 3, 2 }.ToAsyncEnumerable();
  210. var res = xs.SequenceEqual(ys);
  211. Assert.False(res.Result);
  212. }
  213. [Fact]
  214. public void SequenceEqual4()
  215. {
  216. var xs = new[] { 1, 2, 3, 4 }.ToAsyncEnumerable();
  217. var ys = new[] { 1, 2, 3 }.ToAsyncEnumerable();
  218. var res = xs.SequenceEqual(ys);
  219. Assert.False(res.Result);
  220. }
  221. [Fact]
  222. public void SequenceEqual5()
  223. {
  224. var xs = new[] { 1, 2, 3 }.ToAsyncEnumerable();
  225. var ys = new[] { 1, 2, 3, 4 }.ToAsyncEnumerable();
  226. var res = xs.SequenceEqual(ys);
  227. Assert.False(res.Result);
  228. }
  229. [Fact]
  230. public void SequenceEqual6()
  231. {
  232. var ex = new Exception("Bang!");
  233. var xs = new[] { 1, 2, 3, 4 }.ToAsyncEnumerable();
  234. var ys = AsyncEnumerable.Throw<int>(ex);
  235. var res = xs.SequenceEqual(ys);
  236. AssertThrows<Exception>(() => res.Wait(WaitTimeoutMs), ex_ => ((AggregateException)ex_).Flatten().InnerExceptions.Single() == ex);
  237. }
  238. [Fact]
  239. public void SequenceEqual7()
  240. {
  241. var ex = new Exception("Bang!");
  242. var xs = AsyncEnumerable.Throw<int>(ex);
  243. var ys = new[] { 1, 2, 3, 4 }.ToAsyncEnumerable();
  244. var res = xs.SequenceEqual(ys);
  245. AssertThrows<Exception>(() => res.Wait(WaitTimeoutMs), ex_ => ((AggregateException)ex_).Flatten().InnerExceptions.Single() == ex);
  246. }
  247. [Fact]
  248. public void SequenceEqual8()
  249. {
  250. var xs = new[] { 1, 2, 3 }.ToAsyncEnumerable();
  251. var res = xs.SequenceEqual(xs, new Eq());
  252. Assert.True(res.Result);
  253. }
  254. [Fact]
  255. public void SequenceEqual9()
  256. {
  257. var xs = AsyncEnumerable.Empty<int>();
  258. var res = xs.SequenceEqual(xs, new Eq());
  259. Assert.True(res.Result);
  260. }
  261. [Fact]
  262. public void SequenceEqual10()
  263. {
  264. var xs = new[] { 1, 2, 3 }.ToAsyncEnumerable();
  265. var ys = new[] { 1, 3, 2 }.ToAsyncEnumerable();
  266. var res = xs.SequenceEqual(ys, new Eq());
  267. Assert.False(res.Result);
  268. }
  269. [Fact]
  270. public void SequenceEqual11()
  271. {
  272. var xs = new[] { 1, 2, 3, 4 }.ToAsyncEnumerable();
  273. var ys = new[] { 1, 2, 3 }.ToAsyncEnumerable();
  274. var res = xs.SequenceEqual(ys, new Eq());
  275. Assert.False(res.Result);
  276. }
  277. [Fact]
  278. public void SequenceEqual12()
  279. {
  280. var xs = new[] { 1, 2, 3 }.ToAsyncEnumerable();
  281. var ys = new[] { 1, 2, 3, 4 }.ToAsyncEnumerable();
  282. var res = xs.SequenceEqual(ys, new Eq());
  283. Assert.False(res.Result);
  284. }
  285. [Fact]
  286. public void SequenceEqual13()
  287. {
  288. var ex = new Exception("Bang!");
  289. var xs = new[] { 1, 2, 3, 4 }.ToAsyncEnumerable();
  290. var ys = AsyncEnumerable.Throw<int>(ex);
  291. var res = xs.SequenceEqual(ys, new Eq());
  292. AssertThrows<Exception>(() => res.Wait(WaitTimeoutMs), ex_ => ((AggregateException)ex_).Flatten().InnerExceptions.Single() == ex);
  293. }
  294. [Fact]
  295. public void SequenceEqual14()
  296. {
  297. var ex = new Exception("Bang!");
  298. var xs = AsyncEnumerable.Throw<int>(ex);
  299. var ys = new[] { 1, 2, 3, 4 }.ToAsyncEnumerable();
  300. var res = xs.SequenceEqual(ys, new Eq());
  301. AssertThrows<Exception>(() => res.Wait(WaitTimeoutMs), ex_ => ((AggregateException)ex_).Flatten().InnerExceptions.Single() == ex);
  302. }
  303. [Fact]
  304. public void SequenceEqual15()
  305. {
  306. var xs = new[] { 1, 2, -3, 4 }.ToAsyncEnumerable();
  307. var ys = new[] { 1, -2, 3, 4 }.ToAsyncEnumerable();
  308. var res = xs.SequenceEqual(ys, new Eq());
  309. Assert.True(res.Result);
  310. }
  311. [Fact]
  312. public void SequenceEqual16()
  313. {
  314. var xs = new[] { 1, 2, -3, 4 }.ToAsyncEnumerable();
  315. var ys = new[] { 1, -2, 3, 4 }.ToAsyncEnumerable();
  316. var res = xs.SequenceEqual(ys, new EqEx());
  317. AssertThrows<Exception>(() => res.Wait(WaitTimeoutMs), ex_ => ((AggregateException)ex_).Flatten().InnerExceptions.Single() is NotImplementedException);
  318. }
  319. class EqEx : IEqualityComparer<int>
  320. {
  321. public bool Equals(int x, int y)
  322. {
  323. throw new NotImplementedException();
  324. }
  325. public int GetHashCode(int obj)
  326. {
  327. throw new NotImplementedException();
  328. }
  329. }
  330. [Fact]
  331. public void GroupJoin_Null()
  332. {
  333. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.GroupJoin<int, int, int, int>(null, AsyncEnumerable.Return(42), x => x, x => x, (x, y) => x));
  334. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.GroupJoin<int, int, int, int>(AsyncEnumerable.Return(42), null, x => x, x => x, (x, y) => x));
  335. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.GroupJoin<int, int, int, int>(AsyncEnumerable.Return(42), AsyncEnumerable.Return(42), null, x => x, (x, y) => x));
  336. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.GroupJoin<int, int, int, int>(AsyncEnumerable.Return(42), AsyncEnumerable.Return(42), x => x, null, (x, y) => x));
  337. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.GroupJoin<int, int, int, int>(AsyncEnumerable.Return(42), AsyncEnumerable.Return(42), x => x, x => x, null));
  338. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.GroupJoin<int, int, int, int>(null, AsyncEnumerable.Return(42), x => x, x => x, (x, y) => x, EqualityComparer<int>.Default));
  339. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.GroupJoin<int, int, int, int>(AsyncEnumerable.Return(42), null, x => x, x => x, (x, y) => x, EqualityComparer<int>.Default));
  340. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.GroupJoin<int, int, int, int>(AsyncEnumerable.Return(42), AsyncEnumerable.Return(42), null, x => x, (x, y) => x, EqualityComparer<int>.Default));
  341. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.GroupJoin<int, int, int, int>(AsyncEnumerable.Return(42), AsyncEnumerable.Return(42), x => x, null, (x, y) => x, EqualityComparer<int>.Default));
  342. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.GroupJoin<int, int, int, int>(AsyncEnumerable.Return(42), AsyncEnumerable.Return(42), x => x, x => x, null, EqualityComparer<int>.Default));
  343. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.GroupJoin<int, int, int, int>(AsyncEnumerable.Return(42), AsyncEnumerable.Return(42), x => x, x => x, (x, y) => x, default(IEqualityComparer<int>)));
  344. }
  345. [Fact]
  346. public void GroupJoin1()
  347. {
  348. var xs = new[] { 0, 1, 2 }.ToAsyncEnumerable();
  349. var ys = new[] { 4, 7, 6, 2, 3, 4, 8, 9 }.ToAsyncEnumerable();
  350. var res = xs.GroupJoin(ys, x => x % 3, y => y % 3, (x, i) => x + " - " + i.Aggregate("", (s, j) => s + j).Result);
  351. var e = res.GetAsyncEnumerator();
  352. HasNext(e, "0 - 639");
  353. HasNext(e, "1 - 474");
  354. HasNext(e, "2 - 28");
  355. NoNext(e);
  356. }
  357. [Fact]
  358. public void GroupJoin2()
  359. {
  360. var xs = new[] { 0, 1, 2 }.ToAsyncEnumerable();
  361. var ys = new[] { 3, 6, 4 }.ToAsyncEnumerable();
  362. var res = xs.GroupJoin(ys, x => x % 3, y => y % 3, (x, i) => x + " - " + i.Aggregate("", (s, j) => s + j).Result);
  363. var e = res.GetAsyncEnumerator();
  364. HasNext(e, "0 - 36");
  365. HasNext(e, "1 - 4");
  366. HasNext(e, "2 - ");
  367. NoNext(e);
  368. }
  369. [Fact]
  370. public void GroupJoin3()
  371. {
  372. var ex = new Exception("Bang!");
  373. var xs = AsyncEnumerable.Throw<int>(ex);
  374. var ys = new[] { 3, 6, 4 }.ToAsyncEnumerable();
  375. var res = xs.GroupJoin(ys, x => x % 3, y => y % 3, (x, i) => x + " - " + i.Aggregate("", (s, j) => s + j).Result);
  376. var e = res.GetAsyncEnumerator();
  377. AssertThrows(() => e.MoveNextAsync().Wait(WaitTimeoutMs), (Exception ex_) => ((AggregateException)ex_).Flatten().InnerExceptions.Single() == ex);
  378. }
  379. [Fact]
  380. public void GroupJoin4()
  381. {
  382. var ex = new Exception("Bang!");
  383. var xs = new[] { 0, 1, 2 }.ToAsyncEnumerable();
  384. var ys = AsyncEnumerable.Throw<int>(ex);
  385. var res = xs.GroupJoin(ys, x => x % 3, y => y % 3, (x, i) => x + " - " + i.Aggregate("", (s, j) => s + j).Result);
  386. var e = res.GetAsyncEnumerator();
  387. AssertThrows(() => e.MoveNextAsync().Wait(WaitTimeoutMs), (Exception ex_) => ((AggregateException)ex_).Flatten().InnerExceptions.Single() == ex);
  388. }
  389. [Fact]
  390. public void GroupJoin5()
  391. {
  392. var ex = new Exception("Bang!");
  393. var xs = new[] { 0, 1, 2 }.ToAsyncEnumerable();
  394. var ys = new[] { 3, 6, 4 }.ToAsyncEnumerable();
  395. var res = xs.GroupJoin(ys, x => { throw ex; }, y => y % 3, (x, i) => x + " - " + i.Aggregate("", (s, j) => s + j).Result);
  396. var e = res.GetAsyncEnumerator();
  397. AssertThrows(() => e.MoveNextAsync().Wait(WaitTimeoutMs), (Exception ex_) => ((AggregateException)ex_).Flatten().InnerExceptions.Single() == ex);
  398. }
  399. [Fact]
  400. public void GroupJoin6()
  401. {
  402. var ex = new Exception("Bang!");
  403. var xs = new[] { 0, 1, 2 }.ToAsyncEnumerable();
  404. var ys = new[] { 3, 6, 4 }.ToAsyncEnumerable();
  405. var res = xs.GroupJoin(ys, x => x % 3, y => { throw ex; }, (x, i) => x + " - " + i.Aggregate("", (s, j) => s + j).Result);
  406. var e = res.GetAsyncEnumerator();
  407. AssertThrows(() => e.MoveNextAsync().Wait(WaitTimeoutMs), (Exception ex_) => ((AggregateException)ex_).Flatten().InnerExceptions.Single() == ex);
  408. }
  409. [Fact]
  410. public void GroupJoin7()
  411. {
  412. var ex = new Exception("Bang!");
  413. var xs = new[] { 0, 1, 2 }.ToAsyncEnumerable();
  414. var ys = new[] { 3, 6, 4 }.ToAsyncEnumerable();
  415. var res = xs.GroupJoin(ys, x => x % 3, y => y % 3, (x, i) =>
  416. {
  417. if (x == 1)
  418. throw ex;
  419. return x + " - " + i.Aggregate("", (s, j) => s + j).Result;
  420. });
  421. var e = res.GetAsyncEnumerator();
  422. HasNext(e, "0 - 36");
  423. AssertThrows(() => e.MoveNextAsync().Wait(WaitTimeoutMs), (Exception ex_) => ((AggregateException)ex_).Flatten().InnerExceptions.Single() == ex);
  424. }
  425. [Fact]
  426. public void Join_Null()
  427. {
  428. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.Join<int, int, int, int>(null, AsyncEnumerable.Return(42), x => x, x => x, (x, y) => x));
  429. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.Join<int, int, int, int>(AsyncEnumerable.Return(42), null, x => x, x => x, (x, y) => x));
  430. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.Join<int, int, int, int>(AsyncEnumerable.Return(42), AsyncEnumerable.Return(42), null, x => x, (x, y) => x));
  431. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.Join<int, int, int, int>(AsyncEnumerable.Return(42), AsyncEnumerable.Return(42), x => x, null, (x, y) => x));
  432. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.Join<int, int, int, int>(AsyncEnumerable.Return(42), AsyncEnumerable.Return(42), x => x, x => x, null));
  433. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.Join<int, int, int, int>(null, AsyncEnumerable.Return(42), x => x, x => x, (x, y) => x, EqualityComparer<int>.Default));
  434. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.Join<int, int, int, int>(AsyncEnumerable.Return(42), null, x => x, x => x, (x, y) => x, EqualityComparer<int>.Default));
  435. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.Join<int, int, int, int>(AsyncEnumerable.Return(42), AsyncEnumerable.Return(42), null, x => x, (x, y) => x, EqualityComparer<int>.Default));
  436. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.Join<int, int, int, int>(AsyncEnumerable.Return(42), AsyncEnumerable.Return(42), x => x, null, (x, y) => x, EqualityComparer<int>.Default));
  437. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.Join<int, int, int, int>(AsyncEnumerable.Return(42), AsyncEnumerable.Return(42), x => x, x => x, null, EqualityComparer<int>.Default));
  438. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.Join<int, int, int, int>(AsyncEnumerable.Return(42), AsyncEnumerable.Return(42), x => x, x => x, (x, y) => x, default(IEqualityComparer<int>)));
  439. }
  440. [Fact]
  441. public void Join1()
  442. {
  443. var xs = new[] { 0, 1, 2 }.ToAsyncEnumerable();
  444. var ys = new[] { 3, 6, 4 }.ToAsyncEnumerable();
  445. var res = xs.Join(ys, x => x % 3, y => y % 3, (x, y) => x + y);
  446. var e = res.GetAsyncEnumerator();
  447. HasNext(e, 0 + 3);
  448. HasNext(e, 0 + 6);
  449. HasNext(e, 1 + 4);
  450. NoNext(e);
  451. }
  452. [Fact]
  453. public void Join2()
  454. {
  455. var xs = new[] { 3, 6, 4 }.ToAsyncEnumerable();
  456. var ys = new[] { 0, 1, 2 }.ToAsyncEnumerable();
  457. var res = xs.Join(ys, x => x % 3, y => y % 3, (x, y) => x + y);
  458. var e = res.GetAsyncEnumerator();
  459. HasNext(e, 3 + 0);
  460. HasNext(e, 6 + 0);
  461. HasNext(e, 4 + 1);
  462. NoNext(e);
  463. }
  464. [Fact]
  465. public void Join3()
  466. {
  467. var xs = new[] { 0, 1, 2 }.ToAsyncEnumerable();
  468. var ys = new[] { 3, 6 }.ToAsyncEnumerable();
  469. var res = xs.Join(ys, x => x % 3, y => y % 3, (x, y) => x + y);
  470. var e = res.GetAsyncEnumerator();
  471. HasNext(e, 0 + 3);
  472. HasNext(e, 0 + 6);
  473. NoNext(e);
  474. }
  475. [Fact]
  476. public void Join4()
  477. {
  478. var xs = new[] { 3, 6 }.ToAsyncEnumerable();
  479. var ys = new[] { 0, 1, 2 }.ToAsyncEnumerable();
  480. var res = xs.Join(ys, x => x % 3, y => y % 3, (x, y) => x + y);
  481. var e = res.GetAsyncEnumerator();
  482. HasNext(e, 3 + 0);
  483. HasNext(e, 6 + 0);
  484. NoNext(e);
  485. }
  486. [Fact]
  487. public void Join5()
  488. {
  489. var ex = new Exception("Bang!");
  490. var xs = AsyncEnumerable.Throw<int>(ex);
  491. var ys = new[] { 0, 1, 2 }.ToAsyncEnumerable();
  492. var res = xs.Join(ys, x => x % 3, y => y % 3, (x, y) => x + y);
  493. var e = res.GetAsyncEnumerator();
  494. AssertThrows(() => e.MoveNextAsync().Wait(WaitTimeoutMs), (Exception ex_) => ((AggregateException)ex_).Flatten().InnerExceptions.Single() == ex);
  495. }
  496. [Fact]
  497. public void Join6()
  498. {
  499. var ex = new Exception("Bang!");
  500. var xs = new[] { 0, 1, 2 }.ToAsyncEnumerable();
  501. var ys = AsyncEnumerable.Throw<int>(ex);
  502. var res = xs.Join(ys, x => x % 3, y => y % 3, (x, y) => x + y);
  503. var e = res.GetAsyncEnumerator();
  504. AssertThrows(() => e.MoveNextAsync().Wait(WaitTimeoutMs), (Exception ex_) => ((AggregateException)ex_).Flatten().InnerExceptions.Single() == ex);
  505. }
  506. [Fact]
  507. public void Join7()
  508. {
  509. var ex = new Exception("Bang!");
  510. var xs = new[] { 0, 1, 2 }.ToAsyncEnumerable();
  511. var ys = new[] { 0, 1, 2 }.ToAsyncEnumerable();
  512. var res = xs.Join(ys, x => { throw ex; }, y => y, (x, y) => x + y);
  513. var e = res.GetAsyncEnumerator();
  514. AssertThrows(() => e.MoveNextAsync().Wait(WaitTimeoutMs), (Exception ex_) => ((AggregateException)ex_).Flatten().InnerExceptions.Single() == ex);
  515. }
  516. [Fact]
  517. public void Join8()
  518. {
  519. var ex = new Exception("Bang!");
  520. var xs = new[] { 0, 1, 2 }.ToAsyncEnumerable();
  521. var ys = new[] { 0, 1, 2 }.ToAsyncEnumerable();
  522. var res = xs.Join(ys, x => x, y => { throw ex; }, (x, y) => x + y);
  523. var e = res.GetAsyncEnumerator();
  524. AssertThrows(() => e.MoveNextAsync().Wait(WaitTimeoutMs), (Exception ex_) => ((AggregateException)ex_).Flatten().InnerExceptions.Single() == ex);
  525. }
  526. [Fact]
  527. public void Join9()
  528. {
  529. var ex = new Exception("Bang!");
  530. var xs = new[] { 0, 1, 2 }.ToAsyncEnumerable();
  531. var ys = new[] { 0, 1, 2 }.ToAsyncEnumerable();
  532. var res = xs.Join<int, int, int, int>(ys, x => x, y => y, (x, y) => { throw ex; });
  533. var e = res.GetAsyncEnumerator();
  534. AssertThrows(() => e.MoveNextAsync().Wait(WaitTimeoutMs), (Exception ex_) => ((AggregateException)ex_).Flatten().InnerExceptions.Single() == ex);
  535. }
  536. [Fact]
  537. public async Task Join10()
  538. {
  539. var xs = new[] { 0, 1, 2 }.ToAsyncEnumerable();
  540. var ys = new[] { 3, 6, 4 }.ToAsyncEnumerable();
  541. var res = xs.Join(ys, x => x % 3, y => y % 3, (x, y) => x + y);
  542. await SequenceIdentity(res);
  543. }
  544. [Fact]
  545. public void Join11()
  546. {
  547. var customers = new List<Customer>
  548. {
  549. new Customer {CustomerId = "ALFKI"},
  550. new Customer {CustomerId = "ANANT"},
  551. new Customer {CustomerId = "FISSA"}
  552. };
  553. var orders = new List<Order>
  554. {
  555. new Order { OrderId = 1, CustomerId = "ALFKI"},
  556. new Order { OrderId = 2, CustomerId = "ALFKI"},
  557. new Order { OrderId = 3, CustomerId = "ALFKI"},
  558. new Order { OrderId = 4, CustomerId = "FISSA"},
  559. new Order { OrderId = 5, CustomerId = "FISSA"},
  560. new Order { OrderId = 6, CustomerId = "FISSA"},
  561. };
  562. var asyncResult = customers.ToAsyncEnumerable()
  563. .Join(orders.ToAsyncEnumerable(), c => c.CustomerId, o => o.CustomerId,
  564. (c, o) => new CustomerOrder { CustomerId = c.CustomerId, OrderId = o.OrderId });
  565. var e = asyncResult.GetAsyncEnumerator();
  566. HasNext(e, new CustomerOrder { CustomerId = "ALFKI", OrderId = 1 });
  567. HasNext(e, new CustomerOrder { CustomerId = "ALFKI", OrderId = 2 });
  568. HasNext(e, new CustomerOrder { CustomerId = "ALFKI", OrderId = 3 });
  569. HasNext(e, new CustomerOrder { CustomerId = "FISSA", OrderId = 4 });
  570. HasNext(e, new CustomerOrder { CustomerId = "FISSA", OrderId = 5 });
  571. HasNext(e, new CustomerOrder { CustomerId = "FISSA", OrderId = 6 });
  572. NoNext(e);
  573. }
  574. [Fact]
  575. public void Join12()
  576. {
  577. var customers = new List<Customer>
  578. {
  579. new Customer {CustomerId = "ANANT"},
  580. new Customer {CustomerId = "ALFKI"},
  581. new Customer {CustomerId = "FISSA"}
  582. };
  583. var orders = new List<Order>
  584. {
  585. new Order { OrderId = 1, CustomerId = "ALFKI"},
  586. new Order { OrderId = 2, CustomerId = "ALFKI"},
  587. new Order { OrderId = 3, CustomerId = "ALFKI"},
  588. new Order { OrderId = 4, CustomerId = "FISSA"},
  589. new Order { OrderId = 5, CustomerId = "FISSA"},
  590. new Order { OrderId = 6, CustomerId = "FISSA"},
  591. };
  592. var asyncResult = customers.ToAsyncEnumerable()
  593. .Join(orders.ToAsyncEnumerable(), c => c.CustomerId, o => o.CustomerId,
  594. (c, o) => new CustomerOrder { CustomerId = c.CustomerId, OrderId = o.OrderId });
  595. var e = asyncResult.GetAsyncEnumerator();
  596. HasNext(e, new CustomerOrder { CustomerId = "ALFKI", OrderId = 1 });
  597. HasNext(e, new CustomerOrder { CustomerId = "ALFKI", OrderId = 2 });
  598. HasNext(e, new CustomerOrder { CustomerId = "ALFKI", OrderId = 3 });
  599. HasNext(e, new CustomerOrder { CustomerId = "FISSA", OrderId = 4 });
  600. HasNext(e, new CustomerOrder { CustomerId = "FISSA", OrderId = 5 });
  601. HasNext(e, new CustomerOrder { CustomerId = "FISSA", OrderId = 6 });
  602. NoNext(e);
  603. }
  604. [Fact]
  605. public void SelectManyMultiple_Null()
  606. {
  607. AssertThrows<ArgumentNullException>(() => AsyncEnumerableEx.SelectMany<int, int>(default(IAsyncEnumerable<int>), AsyncEnumerable.Return(42)));
  608. AssertThrows<ArgumentNullException>(() => AsyncEnumerableEx.SelectMany<int, int>(AsyncEnumerable.Return(42), default(IAsyncEnumerable<int>)));
  609. }
  610. [Fact]
  611. public void SelectManyMultiple1()
  612. {
  613. var xs = new[] { 0, 1, 2 }.ToAsyncEnumerable();
  614. var ys = new[] { 3, 4 }.ToAsyncEnumerable();
  615. var res = xs.SelectMany(ys);
  616. var e = res.GetAsyncEnumerator();
  617. HasNext(e, 3);
  618. HasNext(e, 4);
  619. HasNext(e, 3);
  620. HasNext(e, 4);
  621. HasNext(e, 3);
  622. HasNext(e, 4);
  623. NoNext(e);
  624. }
  625. public class Customer
  626. {
  627. public string CustomerId { get; set; }
  628. }
  629. public class Order
  630. {
  631. public int OrderId { get; set; }
  632. public string CustomerId { get; set; }
  633. }
  634. [DebuggerDisplay("CustomerId = {CustomerId}, OrderId = {OrderId}")]
  635. public class CustomerOrder : IEquatable<CustomerOrder>
  636. {
  637. public bool Equals(CustomerOrder other)
  638. {
  639. if (ReferenceEquals(null, other)) return false;
  640. if (ReferenceEquals(this, other)) return true;
  641. return OrderId == other.OrderId && string.Equals(CustomerId, other.CustomerId);
  642. }
  643. public override bool Equals(object obj)
  644. {
  645. if (ReferenceEquals(null, obj)) return false;
  646. if (ReferenceEquals(this, obj)) return true;
  647. if (obj.GetType() != this.GetType()) return false;
  648. return Equals((CustomerOrder)obj);
  649. }
  650. public override int GetHashCode()
  651. {
  652. unchecked
  653. {
  654. return (OrderId * 397) ^ (CustomerId != null ? CustomerId.GetHashCode() : 0);
  655. }
  656. }
  657. public static bool operator ==(CustomerOrder left, CustomerOrder right)
  658. {
  659. return Equals(left, right);
  660. }
  661. public static bool operator !=(CustomerOrder left, CustomerOrder right)
  662. {
  663. return !Equals(left, right);
  664. }
  665. public int OrderId { get; set; }
  666. public string CustomerId { get; set; }
  667. }
  668. private sealed class Eq : IEqualityComparer<int>
  669. {
  670. public bool Equals(int x, int y)
  671. {
  672. return EqualityComparer<int>.Default.Equals(Math.Abs(x), Math.Abs(y));
  673. }
  674. public int GetHashCode(int obj)
  675. {
  676. return EqualityComparer<int>.Default.GetHashCode(Math.Abs(obj));
  677. }
  678. }
  679. }
  680. }