public async Task AsyncExceptionIsPropagatedToGetNextCall()
{
TaskCompletionSource <int> tcs = new TaskCompletionSource <int>();
IAsyncEnumerator <int> enumerator = AwaitAndYieldItem();
AsyncItem <int> itemTask = enumerator.GetNextAsync();
tcs.SetException(new Exception("The punch has been spiked"));
Func <Task <Optional <int> > > awaitFaultyItemAct = async() => await itemTask;
awaitFaultyItemAct.ShouldThrow <Exception>();
Optional <int> itemAfterException = await enumerator.GetNextAsync().WithSyncContext();
itemAfterException.HasValue.Should().BeFalse();
async IAsyncEnumerator <int> AwaitAndYieldItem()
{
int awaitedItem = await tcs.Task.ConfigureAwait(false);
await AsyncYield.Item(awaitedItem);
return(await AsyncYield.Break <int>());
}
}
public async Task YieldBreakCanInterruptEnumerator()
{
bool doNotInlineMePlease = true;
int[] itemsBeforeYield = Enumerable.Range(42, 10).ToArray();
int[] itemsAfterYield = Enumerable.Range(666, 10).ToArray();
int[] enumeratedItems = await ForeachBreakAndForeach().ToArray().ConfigureAwait(true);
enumeratedItems.Should().HaveEquivalentItems(itemsBeforeYield);
async IAsyncEnumerator <int> ForeachBreakAndForeach()
{
foreach (int item in itemsBeforeYield)
{
await AsyncYield.Item(item);
}
if (doNotInlineMePlease)
{
return(await AsyncYield.Break <int>());
}
foreach (int item in itemsAfterYield)
{
await AsyncYield.Item(item);
}
return(await AsyncYield.Break <int>());
}
}
public async Task TrueAwaitsAreSupported()
{
const int expectedValue = 42;
TaskCompletionSource <int> tcs = new TaskCompletionSource <int>();
IAsyncEnumerator <int> enumerator = AwaitTaskThenYieldItem();
AsyncItem <int> nextItemTask = enumerator.GetNextAsync();
nextItemTask.IsCompleted.Should().BeFalse();
tcs.SetResult(expectedValue);
Optional <int> yieldedItem = await nextItemTask.WithSyncContext();
yieldedItem.HasValue.Should().BeTrue();
yieldedItem.Value.Should().Be(expectedValue);
async IAsyncEnumerator <int> AwaitTaskThenYieldItem()
{
int awaitedValue = await tcs.Task.ConfigureAwait(false);
await AsyncYield.Item(awaitedValue);
return(await AsyncYield.Break <int>());
}
}
public static async IAsyncEnumerator <T> Create <T>(params Task <T>[] tasks)
{
foreach (Task <T> task in tasks)
{
T item = await task.ConfigureAwait(false);
await AsyncYield.Item(item);
}
return(await AsyncYield.Break <T>());
}
public async Task ItemsWithoutTrueAwaitsAreReturnedCorrectly()
{
int[] items = Enumerable.Range(42, 10).ToArray();
int[] enumeratedItems = await ForeachThroughItems().ToArray().ConfigureAwait(true);
enumeratedItems.Should().HaveEquivalentItems(items);
async IAsyncEnumerator <int> ForeachThroughItems()
{
foreach (int item in items)
{
await AsyncYield.Item(item);
}
return(await AsyncYield.Break <int>());
}
}
public async Task ExecutionDoesNotStartUntilNextAsyncIsCalled()
{
bool executionStarted = false;
IAsyncEnumerator <int> enumerator = SideEffectEnumerator();
executionStarted.Should().BeFalse();
Optional <int> nextItem = await enumerator.GetNextAsync().WithSyncContext();
executionStarted.Should().BeTrue();
async IAsyncEnumerator <int> SideEffectEnumerator()
{
executionStarted = true;
await AsyncYield.Item(42);
return(await AsyncYield.Break <int>());
}
}
public void GetNextThrowsIfCalledBeforePreviousGetNextCompletes()
{
TaskCompletionSource <int>[] taskSources = Enumerable.Range(0, 2).Select(_ => new TaskCompletionSource <int>()).ToArray();
IAsyncEnumerator <int> enumerator = TaskDependentEnumerator();
AsyncItem <int> firstItemTask = enumerator.GetNextAsync();
Func <Task <Optional <int> > > despicableAct = async() => await enumerator.GetNextAsync();
despicableAct.ShouldThrow <PreviousItemNotCompletedException>();
async IAsyncEnumerator <int> TaskDependentEnumerator()
{
foreach (TaskCompletionSource <int> tcs in taskSources)
{
int item = await tcs.Task.ConfigureAwait(false);
await AsyncYield.Item(item);
}
return(await AsyncYield.Break <int>());
}
}
public async Task AsyncLocalsAreFlowedCorrectlyBetweenYields()
{
int[] yieldedItems = await AsyncLocalEnumerator().ToArray().ConfigureAwait(true);
yieldedItems.Should().HaveEquivalentItems(new int[] { 42, 43, 44 });
async IAsyncEnumerator <int> AsyncLocalEnumerator()
{
AsyncLocal <int> asyncLocal = new AsyncLocal <int>();
asyncLocal.Value = 42;
await AsyncYield.Item(asyncLocal.Value);
asyncLocal.Value++;
await AsyncYield.Item(asyncLocal.Value);
await Task.Delay(100).ConfigureAwait(false);
asyncLocal.Value++;
await AsyncYield.Item(asyncLocal.Value);
return(await AsyncYield.Break <int>());
}
}
public async Task SynchronousExceptionIsPropagatedToGetNextCall()
{
const int expectedValue = 42;
IAsyncEnumerator <int> enumerator = YieldItemThenThrow();
Optional <int> item = await enumerator.GetNextAsync().WithSyncContext();
item.Value.Should().Be(expectedValue);
Func <Task <Optional <int> > > nextItemAct = async() => await enumerator.GetNextAsync();
nextItemAct.ShouldThrow <Exception>();
Optional <int> itemAfterException = await enumerator.GetNextAsync().WithSyncContext();
itemAfterException.HasValue.Should().BeFalse();
async IAsyncEnumerator <int> YieldItemThenThrow()
{
await AsyncYield.Item(expectedValue);
throw new Exception("All you enumerator are belong to us");
}
}