AsyncTests.Multiple.cs 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640
  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 void GroupJoin_Null()
  179. {
  180. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.GroupJoin<int, int, int, int>(null, AsyncEnumerable.Return(42), x => x, x => x, (x, y) => x));
  181. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.GroupJoin<int, int, int, int>(AsyncEnumerable.Return(42), null, x => x, x => x, (x, y) => x));
  182. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.GroupJoin<int, int, int, int>(AsyncEnumerable.Return(42), AsyncEnumerable.Return(42), null, x => x, (x, y) => x));
  183. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.GroupJoin<int, int, int, int>(AsyncEnumerable.Return(42), AsyncEnumerable.Return(42), x => x, null, (x, y) => x));
  184. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.GroupJoin<int, int, int, int>(AsyncEnumerable.Return(42), AsyncEnumerable.Return(42), x => x, x => x, null));
  185. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.GroupJoin<int, int, int, int>(null, AsyncEnumerable.Return(42), x => x, x => x, (x, y) => x, EqualityComparer<int>.Default));
  186. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.GroupJoin<int, int, int, int>(AsyncEnumerable.Return(42), null, x => x, x => x, (x, y) => x, EqualityComparer<int>.Default));
  187. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.GroupJoin<int, int, int, int>(AsyncEnumerable.Return(42), AsyncEnumerable.Return(42), null, x => x, (x, y) => x, EqualityComparer<int>.Default));
  188. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.GroupJoin<int, int, int, int>(AsyncEnumerable.Return(42), AsyncEnumerable.Return(42), x => x, null, (x, y) => x, EqualityComparer<int>.Default));
  189. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.GroupJoin<int, int, int, int>(AsyncEnumerable.Return(42), AsyncEnumerable.Return(42), x => x, x => x, null, EqualityComparer<int>.Default));
  190. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.GroupJoin<int, int, int, int>(AsyncEnumerable.Return(42), AsyncEnumerable.Return(42), x => x, x => x, (x, y) => x, default(IEqualityComparer<int>)));
  191. }
  192. [Fact]
  193. public void GroupJoin1()
  194. {
  195. var xs = new[] { 0, 1, 2 }.ToAsyncEnumerable();
  196. var ys = new[] { 4, 7, 6, 2, 3, 4, 8, 9 }.ToAsyncEnumerable();
  197. var res = xs.GroupJoin(ys, x => x % 3, y => y % 3, (x, i) => x + " - " + i.Aggregate("", (s, j) => s + j).Result);
  198. var e = res.GetAsyncEnumerator();
  199. HasNext(e, "0 - 639");
  200. HasNext(e, "1 - 474");
  201. HasNext(e, "2 - 28");
  202. NoNext(e);
  203. }
  204. [Fact]
  205. public void GroupJoin2()
  206. {
  207. var xs = new[] { 0, 1, 2 }.ToAsyncEnumerable();
  208. var ys = new[] { 3, 6, 4 }.ToAsyncEnumerable();
  209. var res = xs.GroupJoin(ys, x => x % 3, y => y % 3, (x, i) => x + " - " + i.Aggregate("", (s, j) => s + j).Result);
  210. var e = res.GetAsyncEnumerator();
  211. HasNext(e, "0 - 36");
  212. HasNext(e, "1 - 4");
  213. HasNext(e, "2 - ");
  214. NoNext(e);
  215. }
  216. [Fact]
  217. public void GroupJoin3()
  218. {
  219. var ex = new Exception("Bang!");
  220. var xs = AsyncEnumerable.Throw<int>(ex);
  221. var ys = new[] { 3, 6, 4 }.ToAsyncEnumerable();
  222. var res = xs.GroupJoin(ys, x => x % 3, y => y % 3, (x, i) => x + " - " + i.Aggregate("", (s, j) => s + j).Result);
  223. var e = res.GetAsyncEnumerator();
  224. AssertThrows(() => e.MoveNextAsync().Wait(WaitTimeoutMs), (Exception ex_) => ((AggregateException)ex_).Flatten().InnerExceptions.Single() == ex);
  225. }
  226. [Fact]
  227. public void GroupJoin4()
  228. {
  229. var ex = new Exception("Bang!");
  230. var xs = new[] { 0, 1, 2 }.ToAsyncEnumerable();
  231. var ys = AsyncEnumerable.Throw<int>(ex);
  232. var res = xs.GroupJoin(ys, x => x % 3, y => y % 3, (x, i) => x + " - " + i.Aggregate("", (s, j) => s + j).Result);
  233. var e = res.GetAsyncEnumerator();
  234. AssertThrows(() => e.MoveNextAsync().Wait(WaitTimeoutMs), (Exception ex_) => ((AggregateException)ex_).Flatten().InnerExceptions.Single() == ex);
  235. }
  236. [Fact]
  237. public void GroupJoin5()
  238. {
  239. var ex = new Exception("Bang!");
  240. var xs = new[] { 0, 1, 2 }.ToAsyncEnumerable();
  241. var ys = new[] { 3, 6, 4 }.ToAsyncEnumerable();
  242. var res = xs.GroupJoin(ys, x => { throw ex; }, y => y % 3, (x, i) => x + " - " + i.Aggregate("", (s, j) => s + j).Result);
  243. var e = res.GetAsyncEnumerator();
  244. AssertThrows(() => e.MoveNextAsync().Wait(WaitTimeoutMs), (Exception ex_) => ((AggregateException)ex_).Flatten().InnerExceptions.Single() == ex);
  245. }
  246. [Fact]
  247. public void GroupJoin6()
  248. {
  249. var ex = new Exception("Bang!");
  250. var xs = new[] { 0, 1, 2 }.ToAsyncEnumerable();
  251. var ys = new[] { 3, 6, 4 }.ToAsyncEnumerable();
  252. var res = xs.GroupJoin(ys, x => x % 3, y => { throw ex; }, (x, i) => x + " - " + i.Aggregate("", (s, j) => s + j).Result);
  253. var e = res.GetAsyncEnumerator();
  254. AssertThrows(() => e.MoveNextAsync().Wait(WaitTimeoutMs), (Exception ex_) => ((AggregateException)ex_).Flatten().InnerExceptions.Single() == ex);
  255. }
  256. [Fact]
  257. public void GroupJoin7()
  258. {
  259. var ex = new Exception("Bang!");
  260. var xs = new[] { 0, 1, 2 }.ToAsyncEnumerable();
  261. var ys = new[] { 3, 6, 4 }.ToAsyncEnumerable();
  262. var res = xs.GroupJoin(ys, x => x % 3, y => y % 3, (x, i) =>
  263. {
  264. if (x == 1)
  265. throw ex;
  266. return x + " - " + i.Aggregate("", (s, j) => s + j).Result;
  267. });
  268. var e = res.GetAsyncEnumerator();
  269. HasNext(e, "0 - 36");
  270. AssertThrows(() => e.MoveNextAsync().Wait(WaitTimeoutMs), (Exception ex_) => ((AggregateException)ex_).Flatten().InnerExceptions.Single() == ex);
  271. }
  272. [Fact]
  273. public void Join_Null()
  274. {
  275. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.Join<int, int, int, int>(null, AsyncEnumerable.Return(42), x => x, x => x, (x, y) => x));
  276. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.Join<int, int, int, int>(AsyncEnumerable.Return(42), null, x => x, x => x, (x, y) => x));
  277. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.Join<int, int, int, int>(AsyncEnumerable.Return(42), AsyncEnumerable.Return(42), null, x => x, (x, y) => x));
  278. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.Join<int, int, int, int>(AsyncEnumerable.Return(42), AsyncEnumerable.Return(42), x => x, null, (x, y) => x));
  279. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.Join<int, int, int, int>(AsyncEnumerable.Return(42), AsyncEnumerable.Return(42), x => x, x => x, null));
  280. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.Join<int, int, int, int>(null, AsyncEnumerable.Return(42), x => x, x => x, (x, y) => x, EqualityComparer<int>.Default));
  281. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.Join<int, int, int, int>(AsyncEnumerable.Return(42), null, x => x, x => x, (x, y) => x, EqualityComparer<int>.Default));
  282. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.Join<int, int, int, int>(AsyncEnumerable.Return(42), AsyncEnumerable.Return(42), null, x => x, (x, y) => x, EqualityComparer<int>.Default));
  283. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.Join<int, int, int, int>(AsyncEnumerable.Return(42), AsyncEnumerable.Return(42), x => x, null, (x, y) => x, EqualityComparer<int>.Default));
  284. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.Join<int, int, int, int>(AsyncEnumerable.Return(42), AsyncEnumerable.Return(42), x => x, x => x, null, EqualityComparer<int>.Default));
  285. AssertThrows<ArgumentNullException>(() => AsyncEnumerable.Join<int, int, int, int>(AsyncEnumerable.Return(42), AsyncEnumerable.Return(42), x => x, x => x, (x, y) => x, default(IEqualityComparer<int>)));
  286. }
  287. [Fact]
  288. public void Join1()
  289. {
  290. var xs = new[] { 0, 1, 2 }.ToAsyncEnumerable();
  291. var ys = new[] { 3, 6, 4 }.ToAsyncEnumerable();
  292. var res = xs.Join(ys, x => x % 3, y => y % 3, (x, y) => x + y);
  293. var e = res.GetAsyncEnumerator();
  294. HasNext(e, 0 + 3);
  295. HasNext(e, 0 + 6);
  296. HasNext(e, 1 + 4);
  297. NoNext(e);
  298. }
  299. [Fact]
  300. public void Join2()
  301. {
  302. var xs = new[] { 3, 6, 4 }.ToAsyncEnumerable();
  303. var ys = new[] { 0, 1, 2 }.ToAsyncEnumerable();
  304. var res = xs.Join(ys, x => x % 3, y => y % 3, (x, y) => x + y);
  305. var e = res.GetAsyncEnumerator();
  306. HasNext(e, 3 + 0);
  307. HasNext(e, 6 + 0);
  308. HasNext(e, 4 + 1);
  309. NoNext(e);
  310. }
  311. [Fact]
  312. public void Join3()
  313. {
  314. var xs = new[] { 0, 1, 2 }.ToAsyncEnumerable();
  315. var ys = new[] { 3, 6 }.ToAsyncEnumerable();
  316. var res = xs.Join(ys, x => x % 3, y => y % 3, (x, y) => x + y);
  317. var e = res.GetAsyncEnumerator();
  318. HasNext(e, 0 + 3);
  319. HasNext(e, 0 + 6);
  320. NoNext(e);
  321. }
  322. [Fact]
  323. public void Join4()
  324. {
  325. var xs = new[] { 3, 6 }.ToAsyncEnumerable();
  326. var ys = new[] { 0, 1, 2 }.ToAsyncEnumerable();
  327. var res = xs.Join(ys, x => x % 3, y => y % 3, (x, y) => x + y);
  328. var e = res.GetAsyncEnumerator();
  329. HasNext(e, 3 + 0);
  330. HasNext(e, 6 + 0);
  331. NoNext(e);
  332. }
  333. [Fact]
  334. public void Join5()
  335. {
  336. var ex = new Exception("Bang!");
  337. var xs = AsyncEnumerable.Throw<int>(ex);
  338. var ys = new[] { 0, 1, 2 }.ToAsyncEnumerable();
  339. var res = xs.Join(ys, x => x % 3, y => y % 3, (x, y) => x + y);
  340. var e = res.GetAsyncEnumerator();
  341. AssertThrows(() => e.MoveNextAsync().Wait(WaitTimeoutMs), (Exception ex_) => ((AggregateException)ex_).Flatten().InnerExceptions.Single() == ex);
  342. }
  343. [Fact]
  344. public void Join6()
  345. {
  346. var ex = new Exception("Bang!");
  347. var xs = new[] { 0, 1, 2 }.ToAsyncEnumerable();
  348. var ys = AsyncEnumerable.Throw<int>(ex);
  349. var res = xs.Join(ys, x => x % 3, y => y % 3, (x, y) => x + y);
  350. var e = res.GetAsyncEnumerator();
  351. AssertThrows(() => e.MoveNextAsync().Wait(WaitTimeoutMs), (Exception ex_) => ((AggregateException)ex_).Flatten().InnerExceptions.Single() == ex);
  352. }
  353. [Fact]
  354. public void Join7()
  355. {
  356. var ex = new Exception("Bang!");
  357. var xs = new[] { 0, 1, 2 }.ToAsyncEnumerable();
  358. var ys = new[] { 0, 1, 2 }.ToAsyncEnumerable();
  359. var res = xs.Join(ys, x => { throw ex; }, y => y, (x, y) => x + y);
  360. var e = res.GetAsyncEnumerator();
  361. AssertThrows(() => e.MoveNextAsync().Wait(WaitTimeoutMs), (Exception ex_) => ((AggregateException)ex_).Flatten().InnerExceptions.Single() == ex);
  362. }
  363. [Fact]
  364. public void Join8()
  365. {
  366. var ex = new Exception("Bang!");
  367. var xs = new[] { 0, 1, 2 }.ToAsyncEnumerable();
  368. var ys = new[] { 0, 1, 2 }.ToAsyncEnumerable();
  369. var res = xs.Join(ys, x => x, y => { throw ex; }, (x, y) => x + y);
  370. var e = res.GetAsyncEnumerator();
  371. AssertThrows(() => e.MoveNextAsync().Wait(WaitTimeoutMs), (Exception ex_) => ((AggregateException)ex_).Flatten().InnerExceptions.Single() == ex);
  372. }
  373. [Fact]
  374. public void Join9()
  375. {
  376. var ex = new Exception("Bang!");
  377. var xs = new[] { 0, 1, 2 }.ToAsyncEnumerable();
  378. var ys = new[] { 0, 1, 2 }.ToAsyncEnumerable();
  379. var res = xs.Join<int, int, int, int>(ys, x => x, y => y, (x, y) => { throw ex; });
  380. var e = res.GetAsyncEnumerator();
  381. AssertThrows(() => e.MoveNextAsync().Wait(WaitTimeoutMs), (Exception ex_) => ((AggregateException)ex_).Flatten().InnerExceptions.Single() == ex);
  382. }
  383. [Fact]
  384. public async Task Join10()
  385. {
  386. var xs = new[] { 0, 1, 2 }.ToAsyncEnumerable();
  387. var ys = new[] { 3, 6, 4 }.ToAsyncEnumerable();
  388. var res = xs.Join(ys, x => x % 3, y => y % 3, (x, y) => x + y);
  389. await SequenceIdentity(res);
  390. }
  391. [Fact]
  392. public void Join11()
  393. {
  394. var customers = new List<Customer>
  395. {
  396. new Customer {CustomerId = "ALFKI"},
  397. new Customer {CustomerId = "ANANT"},
  398. new Customer {CustomerId = "FISSA"}
  399. };
  400. var orders = new List<Order>
  401. {
  402. new Order { OrderId = 1, CustomerId = "ALFKI"},
  403. new Order { OrderId = 2, CustomerId = "ALFKI"},
  404. new Order { OrderId = 3, CustomerId = "ALFKI"},
  405. new Order { OrderId = 4, CustomerId = "FISSA"},
  406. new Order { OrderId = 5, CustomerId = "FISSA"},
  407. new Order { OrderId = 6, CustomerId = "FISSA"},
  408. };
  409. var asyncResult = customers.ToAsyncEnumerable()
  410. .Join(orders.ToAsyncEnumerable(), c => c.CustomerId, o => o.CustomerId,
  411. (c, o) => new CustomerOrder { CustomerId = c.CustomerId, OrderId = o.OrderId });
  412. var e = asyncResult.GetAsyncEnumerator();
  413. HasNext(e, new CustomerOrder { CustomerId = "ALFKI", OrderId = 1 });
  414. HasNext(e, new CustomerOrder { CustomerId = "ALFKI", OrderId = 2 });
  415. HasNext(e, new CustomerOrder { CustomerId = "ALFKI", OrderId = 3 });
  416. HasNext(e, new CustomerOrder { CustomerId = "FISSA", OrderId = 4 });
  417. HasNext(e, new CustomerOrder { CustomerId = "FISSA", OrderId = 5 });
  418. HasNext(e, new CustomerOrder { CustomerId = "FISSA", OrderId = 6 });
  419. NoNext(e);
  420. }
  421. [Fact]
  422. public void Join12()
  423. {
  424. var customers = new List<Customer>
  425. {
  426. new Customer {CustomerId = "ANANT"},
  427. new Customer {CustomerId = "ALFKI"},
  428. new Customer {CustomerId = "FISSA"}
  429. };
  430. var orders = new List<Order>
  431. {
  432. new Order { OrderId = 1, CustomerId = "ALFKI"},
  433. new Order { OrderId = 2, CustomerId = "ALFKI"},
  434. new Order { OrderId = 3, CustomerId = "ALFKI"},
  435. new Order { OrderId = 4, CustomerId = "FISSA"},
  436. new Order { OrderId = 5, CustomerId = "FISSA"},
  437. new Order { OrderId = 6, CustomerId = "FISSA"},
  438. };
  439. var asyncResult = customers.ToAsyncEnumerable()
  440. .Join(orders.ToAsyncEnumerable(), c => c.CustomerId, o => o.CustomerId,
  441. (c, o) => new CustomerOrder { CustomerId = c.CustomerId, OrderId = o.OrderId });
  442. var e = asyncResult.GetAsyncEnumerator();
  443. HasNext(e, new CustomerOrder { CustomerId = "ALFKI", OrderId = 1 });
  444. HasNext(e, new CustomerOrder { CustomerId = "ALFKI", OrderId = 2 });
  445. HasNext(e, new CustomerOrder { CustomerId = "ALFKI", OrderId = 3 });
  446. HasNext(e, new CustomerOrder { CustomerId = "FISSA", OrderId = 4 });
  447. HasNext(e, new CustomerOrder { CustomerId = "FISSA", OrderId = 5 });
  448. HasNext(e, new CustomerOrder { CustomerId = "FISSA", OrderId = 6 });
  449. NoNext(e);
  450. }
  451. [Fact]
  452. public void SelectManyMultiple_Null()
  453. {
  454. AssertThrows<ArgumentNullException>(() => AsyncEnumerableEx.SelectMany<int, int>(default(IAsyncEnumerable<int>), AsyncEnumerable.Return(42)));
  455. AssertThrows<ArgumentNullException>(() => AsyncEnumerableEx.SelectMany<int, int>(AsyncEnumerable.Return(42), default(IAsyncEnumerable<int>)));
  456. }
  457. [Fact]
  458. public void SelectManyMultiple1()
  459. {
  460. var xs = new[] { 0, 1, 2 }.ToAsyncEnumerable();
  461. var ys = new[] { 3, 4 }.ToAsyncEnumerable();
  462. var res = xs.SelectMany(ys);
  463. var e = res.GetAsyncEnumerator();
  464. HasNext(e, 3);
  465. HasNext(e, 4);
  466. HasNext(e, 3);
  467. HasNext(e, 4);
  468. HasNext(e, 3);
  469. HasNext(e, 4);
  470. NoNext(e);
  471. }
  472. public class Customer
  473. {
  474. public string CustomerId { get; set; }
  475. }
  476. public class Order
  477. {
  478. public int OrderId { get; set; }
  479. public string CustomerId { get; set; }
  480. }
  481. [DebuggerDisplay("CustomerId = {CustomerId}, OrderId = {OrderId}")]
  482. public class CustomerOrder : IEquatable<CustomerOrder>
  483. {
  484. public bool Equals(CustomerOrder other)
  485. {
  486. if (ReferenceEquals(null, other)) return false;
  487. if (ReferenceEquals(this, other)) return true;
  488. return OrderId == other.OrderId && string.Equals(CustomerId, other.CustomerId);
  489. }
  490. public override bool Equals(object obj)
  491. {
  492. if (ReferenceEquals(null, obj)) return false;
  493. if (ReferenceEquals(this, obj)) return true;
  494. if (obj.GetType() != this.GetType()) return false;
  495. return Equals((CustomerOrder)obj);
  496. }
  497. public override int GetHashCode()
  498. {
  499. unchecked
  500. {
  501. return (OrderId * 397) ^ (CustomerId != null ? CustomerId.GetHashCode() : 0);
  502. }
  503. }
  504. public static bool operator ==(CustomerOrder left, CustomerOrder right)
  505. {
  506. return Equals(left, right);
  507. }
  508. public static bool operator !=(CustomerOrder left, CustomerOrder right)
  509. {
  510. return !Equals(left, right);
  511. }
  512. public int OrderId { get; set; }
  513. public string CustomerId { get; set; }
  514. }
  515. private sealed class Eq : IEqualityComparer<int>
  516. {
  517. public bool Equals(int x, int y)
  518. {
  519. return EqualityComparer<int>.Default.Equals(Math.Abs(x), Math.Abs(y));
  520. }
  521. public int GetHashCode(int obj)
  522. {
  523. return EqualityComparer<int>.Default.GetHashCode(Math.Abs(obj));
  524. }
  525. }
  526. }
  527. }