1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
// Copyright (c) Microsoft Corporation. All rights reserved. See License.txt in the project root for license information.
using System;
using System.Threading.Tasks;
using Xunit;
// No namespace so that these extensions are available for all test classes
public static class TaskExtensions
{
/// <summary>
/// Waits until the given task finishes executing and completes in any of the 3 states.
/// </summary>
/// <param name="task">A task</param>
public static void WaitUntilCompleted(this Task task)
{
if (task == null) throw new ArgumentNullException("task");
task.ContinueWith(prev =>
{
if (prev.IsFaulted)
{
// Observe the exception in the faulted case to avoid an unobserved exception leaking and
// killing the thread finalizer.
var e = prev.Exception;
}
}, TaskContinuationOptions.ExecuteSynchronously).Wait();
}
public static void RethrowFaultedTaskException(this Task task)
{
task.WaitUntilCompleted();
Assert.Equal(TaskStatus.Faulted, task.Status);
throw task.Exception.GetBaseException();
}
}
|
