RetryTest.cs 1.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455
  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 Xunit;
  8. namespace Tests
  9. {
  10. public class RetryTest : Tests
  11. {
  12. [Fact]
  13. public void Retry_Arguments()
  14. {
  15. AssertThrows<ArgumentNullException>(() => EnumerableEx.Retry<int>(null));
  16. AssertThrows<ArgumentNullException>(() => EnumerableEx.Retry<int>(null, 5));
  17. AssertThrows<ArgumentOutOfRangeException>(() => EnumerableEx.Retry<int>(new[] { 1 }, -1));
  18. }
  19. [Fact]
  20. public void Retry1()
  21. {
  22. var xs = Enumerable.Range(0, 10);
  23. var res = xs.Retry();
  24. Assert.True(Enumerable.SequenceEqual(res, xs));
  25. }
  26. [Fact]
  27. public void Retry2()
  28. {
  29. var xs = Enumerable.Range(0, 10);
  30. var res = xs.Retry(2);
  31. Assert.True(Enumerable.SequenceEqual(res, xs));
  32. }
  33. [Fact]
  34. public void Retry3()
  35. {
  36. var ex = new MyException();
  37. var xs = Enumerable.Range(0, 2).Concat(EnumerableEx.Throw<int>(ex));
  38. var res = xs.Retry(2);
  39. var e = res.GetEnumerator();
  40. HasNext(e, 0);
  41. HasNext(e, 1);
  42. HasNext(e, 0);
  43. HasNext(e, 1);
  44. AssertThrows<MyException>(() => e.MoveNext(), ex_ => ex == ex_);
  45. }
  46. }
  47. }