12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667 |
- // Licensed to the .NET Foundation under one or more agreements.
- // The .NET Foundation licenses this file to you under the MIT License.
- // See the LICENSE file in the project root for more information.
- using System;
- using System.Collections.Generic;
- using System.Linq;
- using System.Threading.Tasks;
- using Xunit;
- namespace Tests
- {
- public class Expand : AsyncEnumerableExTests
- {
- [Fact]
- public void Expand_Null()
- {
- Assert.Throws<ArgumentNullException>(() => AsyncEnumerableEx.Expand(default(IAsyncEnumerable<int>), x => default(IAsyncEnumerable<int>)));
- Assert.Throws<ArgumentNullException>(() => AsyncEnumerableEx.Expand(Return42, default(Func<int, IAsyncEnumerable<int>>)));
- }
- [Fact]
- public async Task Expand1Async()
- {
- var xs = new[] { 2, 3 }.ToAsyncEnumerable().Expand(x => AsyncEnumerableEx.Return(x - 1).Repeat(x - 1));
- var e = xs.GetAsyncEnumerator();
- await HasNextAsync(e, 2);
- await HasNextAsync(e, 3);
- await HasNextAsync(e, 1);
- await HasNextAsync(e, 2);
- await HasNextAsync(e, 2);
- await HasNextAsync(e, 1);
- await HasNextAsync(e, 1);
- await NoNextAsync(e);
- }
- [Fact]
- public async Task Expand2()
- {
- var ex = new Exception("Bang!");
- var xs = new[] { 2, 3 }.ToAsyncEnumerable().Expand(new Func<int, IAsyncEnumerable<int>>(x => { throw ex; }));
- var e = xs.GetAsyncEnumerator();
- await AssertThrowsAsync(e.MoveNextAsync(), ex);
- }
- [Fact]
- public async Task Expand3Async()
- {
- var xs = new[] { 2, 3 }.ToAsyncEnumerable().Expand(x => default(IAsyncEnumerable<int>));
- var e = xs.GetAsyncEnumerator();
- await HasNextAsync(e, 2);
- await HasNextAsync(e, 3);
- await AssertThrowsAsync<NullReferenceException>(e.MoveNextAsync().AsTask());
- }
- [Fact]
- public async Task Expand4()
- {
- var xs = new[] { 2, 3 }.ToAsyncEnumerable().Expand(x => AsyncEnumerableEx.Return(x - 1).Repeat(x - 1));
- await SequenceIdentity(xs);
- }
- }
- }
|