Using.cs 4.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196
  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.Linq;
  7. using System.Threading.Tasks;
  8. using Xunit;
  9. namespace Tests
  10. {
  11. public class Using : AsyncEnumerableExTests
  12. {
  13. [Fact]
  14. public void Using_Null()
  15. {
  16. Assert.Throws<ArgumentNullException>(() => AsyncEnumerableEx.Using<int, IDisposable>(default, _ => default(IAsyncEnumerable<int>)));
  17. Assert.Throws<ArgumentNullException>(() => AsyncEnumerableEx.Using<int, IDisposable>(() => new MyD(null), default));
  18. }
  19. [Fact]
  20. public void Using1()
  21. {
  22. var i = 0;
  23. var d = 0;
  24. var xs = AsyncEnumerableEx.Using(
  25. () =>
  26. {
  27. i++;
  28. return new MyD(() => { d++; });
  29. },
  30. _ => Return42
  31. );
  32. Assert.Equal(0, i);
  33. Assert.Equal(0, d);
  34. var e = xs.GetAsyncEnumerator();
  35. Assert.Equal(0, i);
  36. Assert.Equal(0, d);
  37. }
  38. [Fact]
  39. public async Task Using2()
  40. {
  41. var i = 0;
  42. var d = 0;
  43. var xs = AsyncEnumerableEx.Using(
  44. () =>
  45. {
  46. i++;
  47. return new MyD(() => { d++; });
  48. },
  49. _ => Return42
  50. );
  51. Assert.Equal(0, i);
  52. Assert.Equal(0, d);
  53. var e = xs.GetAsyncEnumerator();
  54. Assert.Equal(0, i);
  55. Assert.Equal(0, d);
  56. await e.DisposeAsync();
  57. Assert.Equal(0, i);
  58. Assert.Equal(0, d);
  59. }
  60. [Fact]
  61. public async Task Using3()
  62. {
  63. var ex = new Exception("Bang!");
  64. var i = 0;
  65. var d = 0;
  66. var xs = AsyncEnumerableEx.Using<int, MyD>(
  67. () =>
  68. {
  69. i++;
  70. return new MyD(() => { d++; });
  71. },
  72. _ => { throw ex; }
  73. );
  74. Assert.Equal(0, i);
  75. Assert.Equal(0, d);
  76. var e = xs.GetAsyncEnumerator();
  77. Assert.Equal(0, i);
  78. Assert.Equal(0, d);
  79. await e.DisposeAsync();
  80. Assert.Equal(0, i);
  81. Assert.Equal(0, d);
  82. }
  83. [Fact]
  84. public async Task Using4Async()
  85. {
  86. var i = 0;
  87. var disposed = new TaskCompletionSource<bool>();
  88. var xs = AsyncEnumerableEx.Using(
  89. () =>
  90. {
  91. i++;
  92. return new MyD(() => { disposed.TrySetResult(true); });
  93. },
  94. _ => Return42
  95. );
  96. Assert.Equal(0, i);
  97. var e = xs.GetAsyncEnumerator();
  98. Assert.Equal(0, i);
  99. await HasNextAsync(e, 42);
  100. Assert.Equal(1, i);
  101. await NoNextAsync(e);
  102. Assert.True(disposed.Task.Result);
  103. }
  104. [Fact]
  105. public void Using5()
  106. {
  107. var ex = new Exception("Bang!");
  108. var i = 0;
  109. var disposed = new TaskCompletionSource<bool>();
  110. var xs = AsyncEnumerableEx.Using(
  111. () =>
  112. {
  113. i++;
  114. return new MyD(() => { disposed.TrySetResult(true); });
  115. },
  116. _ => Throw<int>(ex)
  117. );
  118. Assert.Equal(0, i);
  119. var e = xs.GetAsyncEnumerator();
  120. Assert.Equal(0, i);
  121. AssertThrows(() => e.MoveNextAsync().Wait(WaitTimeoutMs), SingleInnerExceptionMatches(ex));
  122. Assert.Equal(1, i);
  123. Assert.True(disposed.Task.Result);
  124. }
  125. [Fact]
  126. public async Task Using7()
  127. {
  128. var i = 0;
  129. var d = 0;
  130. var xs = AsyncEnumerableEx.Using(
  131. () =>
  132. {
  133. i++;
  134. return new MyD(() => { d++; });
  135. },
  136. _ => Return42
  137. );
  138. await SequenceIdentity(xs);
  139. }
  140. private sealed class MyD : IDisposable
  141. {
  142. private readonly Action _dispose;
  143. public MyD(Action dispose)
  144. {
  145. _dispose = dispose;
  146. }
  147. public void Dispose()
  148. {
  149. _dispose();
  150. }
  151. }
  152. }
  153. }