Retry.cs 1.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758
  1. // Licensed to the .NET Foundation under one or more agreements.
  2. // The .NET Foundation licenses this file to you under the MIT License.
  3. // See the LICENSE file in the project root for more information.
  4. using System;
  5. using System.Linq;
  6. using Xunit;
  7. namespace Tests
  8. {
  9. public class Retry : Tests
  10. {
  11. [Fact]
  12. public void Retry_Arguments()
  13. {
  14. AssertThrows<ArgumentNullException>(() => EnumerableEx.Retry<int>(null));
  15. AssertThrows<ArgumentNullException>(() => EnumerableEx.Retry<int>(null, 5));
  16. AssertThrows<ArgumentOutOfRangeException>(() => EnumerableEx.Retry<int>([1], -1));
  17. }
  18. [Fact]
  19. public void Retry1()
  20. {
  21. var xs = Enumerable.Range(0, 10);
  22. var res = xs.Retry();
  23. Assert.True(Enumerable.SequenceEqual(res, xs));
  24. }
  25. [Fact]
  26. public void Retry2()
  27. {
  28. var xs = Enumerable.Range(0, 10);
  29. var res = xs.Retry(2);
  30. Assert.True(Enumerable.SequenceEqual(res, xs));
  31. }
  32. [Fact]
  33. public void Retry3()
  34. {
  35. var ex = new MyException();
  36. var xs = Enumerable.Range(0, 2).Concat(EnumerableEx.Throw<int>(ex));
  37. var res = xs.Retry(2);
  38. var e = res.GetEnumerator();
  39. HasNext(e, 0);
  40. HasNext(e, 1);
  41. HasNext(e, 0);
  42. HasNext(e, 1);
  43. AssertThrows<MyException>(() => e.MoveNext(), ex_ => ex == ex_);
  44. }
  45. private sealed class MyException : Exception
  46. {
  47. }
  48. }
  49. }