等待，直到所有任务在单元测试中完成

本文关键字：单元测试 任务 等待 | 更新日期: 2023-09-27 18:24:41

我有一个类要进行单元测试：

public class SomeClass
{
    public void Foo()
    {
        Bar();
    }
    private void Bar()
    {
        Task.Factory.StartNew(() =>
        {
            // Do something that takes some time (e.g. an HTTP request)
        });
    }
}

这就是我的单元测试的样子：

[TestMethod]
public void TestFoo()
{
    // Arrange
    var obj = new SomeClass();
    // Act
    obj.Foo();
    obj.Foo();
    obj.Foo();
    // Assert
    /* I need something to wait on all tasks to finish */
    Assert.IsTrue(...);
}

因此，在开始断言之前，我需要让单元测试线程等待Bar方法中启动的所有任务都完成了它们的工作。

重要：我无法更改SomeClass

我该怎么做

等待，直到所有任务在单元测试中完成

解决此问题的一种方法是定义自己的任务调度程序，以便跟踪嵌套任务的完成情况。例如，您可以定义一个同步执行任务的调度器，如下所示：

class SynchronousTaskScheduler : TaskScheduler
{
    protected override void QueueTask(Task task)
    {
        this.TryExecuteTask(task);
    }
    protected override bool TryExecuteTaskInline(Task task, bool wasPreviouslyQueued)
    {
        return this.TryExecuteTask(task);
    }
    protected override IEnumerable<Task> GetScheduledTasks()
    {
        yield break;
    }
}

随后，创建一个同步任务调度程序的实例，并使用它来执行一个根任务，该根任务又会生成所有"隐藏"任务。由于嵌套任务从其父任务继承了当前任务调度程序，因此您的所有内部任务也将在我们的同步调度程序上运行，这意味着我们最外层的StartNew调用只有在所有任务完成时才会返回。

TaskScheduler scheduler = new SynchronousTaskScheduler();
Task.Factory.StartNew(() =>
{
    // Arrange
    var obj = new SomeClass();
    // Act
    obj.Foo();
    obj.Foo();
    obj.Foo();
}, 
    CancellationToken.None,
    TaskCreationOptions.None,
    scheduler);
// Assert
/* I need something to wait on all tasks to finish */
Assert.IsTrue(...);

这种方法的一个缺点是，您将失去任务的所有并发性；但是，您可以通过将自定义调度程序增强为一个并发的调度程序来解决这个问题，但它仍然允许您跟踪正在执行的任务。

Task.WaitAll(the, list, of, task, objects, you, need, to, wait, on);

如果这是void async方法，那么你就做不到。设计已经坏了。它们只适用于火和遗忘。

不确定是否允许您进行此更改，但我可以这样做：

namespace ParallelProgramming.Playground
{
    public class SomeClass
    {
        public Task Foo()
        {
            return Bar();
        }
        private static Task Bar()
        {
            return Task.Factory.StartNew(() =>
                {
                    Console.WriteLine("I fired off. Thread ID: {0}", Thread.CurrentThread.ManagedThreadId);
                    Thread.Sleep(5000);
                    return true; //or whatever else you want.
                });
        }
    }
    [TestClass]
    public class StackOverflow
    {
        [TestMethod]
        public void TestFoo()
        {
            // Arrange
            var obj = new SomeClass();
            var results = new ConcurrentBag<Task>(); 
            var waitForMe = Task.Factory.StartNew(() =>
                {
                    // Act
                    results.Add(obj.Foo());
                    results.Add(obj.Foo());
                    results.Add(obj.Foo());
                    return true;
                });

            Task.WaitAll(waitForMe);
            // Assert
            /* I need something to wait on all tasks to finish */
            Assert.IsTrue(waitForMe.Result);
            Assert.AreEqual(3, results.Count);
        }
    }
}
相关文章：