如何对使用Task.WhenAll的方法进行单元测试

时间:2016-01-28 23:22:31

标签: c# unit-testing nunit task

我们说我们有以下课程:

class SampleClass
{
    private readonly IList<int> _numberList = new List<int> {1, 2, 3, 4, 5, 6, 7, 8, 9, 10};

    public void MethodToBeTested()
    {
        Task.WhenAll(_numberList.Select(async number => { await Task.Run(() => { ProcessNumber(number); }); }));
    }

    private static void ProcessNumber(int number)
    {
        Console.WriteLine($"Thread: {Thread.CurrentThread.ManagedThreadId} Number:{number}");
        Thread.Sleep(1000);
    }
}

我们添加以下单元测试:

[Test]
public void TestMethodToBeTested()
{
    var sampleClass = new SampleClass();
    _sampleClass.MethodToBeTested();
}

问题是当测试运行时,它不会等待MethodToBeTested完成执行,因此输出会有所不同。有没有办法使用NUnit测试此方法的完整运行,而不更改方法签名(从void到Task)?

1 个答案:

答案 0 :(得分:1)

使用GetAwaiter().GetResult()将确保异步调用运行完成。详细了解approach vs await here

public void MethodToBeTested()
{
    Task.WhenAll(_numberList.Select(async number => { await Task.Run(() => { ProcessNumber(number); }); })).GetAwaiter().GetResult();
}