CurrentThreadTaskScheduler 未完成同步

我尝试为视图模型编写单元测试，但在尝试验证 ICommand 调用异步方法两次时遇到了困难。

我使用 Moq 作为我的依赖项。 我像这样设置了异步方法。

this.communicationServiceFake
     .Setup(x => x.WriteParameterAsync(It.IsAny<string>(), It.IsAny<object>()))
     .ReturnsAsyncIncomplete();

扩展 ReturnsAsyncIncomplete 不会在 wait 关键字处立即返回，基本上类似于此处找到的内容：Async/Await 和代码覆盖率 http://bernhard-richter.blogspot.co.at/2014/09/asyncawait-and-code-coverage.html

我使用自己的 TaskSheduler 来确保方法在 Task.Factory.StartNew 返回之前完成。

Task.Factory.StartNew(() => viewModel.Command.Execute(null), 
    CancellationToken.None, TaskCreationOptions.None, new CurrentThreadTaskScheduler ());

基本上 CurrentThreadTaskScheduler 来自这里：等待单元测试中所有任务完成 https://stackoverflow.com/questions/17844182/wait-until-all-task-finish-in-unit-test看起来确实像这样：

public class CurrentThreadTaskScheduler : 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;
    }
}

该命令确实调用了以下代码：

await this.communicationService.WriteParameterAsync("Parameter1", true);
await this.communicationService.WriteParameterAsync("Parameter2", true);

然后验证：

  this.communicationServiceFake
       .Verify(t => t.WriteParameterAsync("Parameter1", true), Times.Once);
  this.communicationServiceFake
       .Verify(t => t.WriteParameterAsync("Parameter2", true), Times.Once);

有时它说第二个电话没有发生。 如果我用 ThreadSleep 替换我的 Task.Factory.StartNew 以确保一切都完成，那么即使不必要地延迟我的单元测试似乎是正确的，一切都可以正常工作。

为什么我的 CurrentThreadTaskScheduler 允许 Task.Factory.StartNew 在 Command.Execute 完成之前返回？

The GetIncompleteTask扩展方法在链接的博客文章 http://bernhard-richter.blogspot.co.at/2014/09/asyncawait-and-code-coverage.html uses Task.Run，因此引入了另一个线程（和竞争条件）。

The CurrentThreadTaskScheduler只会为async void方法（如果所有任务都已完成），这正是您所要做的avoiding with ReturnsAsyncIncomplete.

这是发生的事情：

1. Execute跑进CurrentThreadTaskScheduler. It await正在线程池上运行的未完成任务。此时，StartNew任务完成。
2. 线程池线程完成未完成的任务，并在current线程（线程池线程）。

你想要做的是单元测试async void完全覆盖的方法（即异步）。这当然不容易。

推荐解决方案

使用某种形式的async-aware ICommand比如我在中描述的我的 MSDN 文章 http://msdn.microsoft.com/en-us/magazine/dn630647.aspx。然后你就可以使用await viewModel.Command.ExecuteAsync(null)在你的单元测试中，不要乱搞自定义任务调度程序。

此外，异步模拟确实可以被简化；不需要自定义等待程序，因为 .NET 框架已经有一个：Task.Yield。因此，您可以用一个扩展方法（未经测试）替换所有代码：

public static IReturnsResult<TMock> ReturnsIncompleteAsync<TMock, TResult>(this IReturns<TMock, Task<TResult>> mock, TResult value) where TMock : class
{
  return mock.Returns(async () =>
  {
    await Task.Yield();
    return value;
  });
}

保持异步无效

如果你真的想保留现有的ICommand实现，并想要进行单元测试async void方法，and想要完整的代码覆盖率，那么你真的需要一个自定义的SynchronizationContext而不是自定义TaskScheduler。最简单的方法是安装我的AsyncEx NuGet 包 http://www.nuget.org/packages/Nito.AsyncEx并使用AsyncContext:

// Blocks the current thread until all continuations have completed.
AsyncContext.Run(() => viewModel.Command.Execute(null));

另外，我建议您使用Task.Yield上面描述的方法而不是线程池上的自定义等待。