如果使用设置，则 Moq 模拟调用返回 null

Question

我正在使用 Moq 为 C# 应用程序编写测试。我的测试初始化​​程序有以下代码：

UnityContainer unityContainer = new UnityContainer();

_serviceMock = new Mock<IService>();
_serviceMock.Setup(mock => mock.GetSearchInfoAsync(It.IsAny<CancellationToken>(), It.IsAny<IEnumerable<string>>(), It.IsAny<identifierType>(), It.IsAny<bool>())).Callback(() => _count++);
unityContainer.RegisterInstance(typeof(IService), _serviceMock.Object, new ContainerControlledLifetimeManager());

我想测试一次调用是否只进行了一次。我正在尝试这样：

int _count = 0;

[TestMethod]
public void Properties_Test()
{
    _serviceMock.Verify(mock => mock.GetSearchInfoAsync(It.IsAny<CancellationToken>(), It.IsAny<IEnumerable<string>>(), It.IsAny<identifierType>(), It.IsAny<bool>()), Times.Exactly(1), "Invocation was performed " + _count + " times but was expected only once!");
}

这是它实际被调用的方法：

private void Search(string queryValue, identifierType identifierType)
{
    CancellationToken cancellationToken;

    lock (_syncLock)
    {
        _cancellationTokenSource.Cancel();
        _cancellationTokenSource = new CancellationTokenSource();
        cancellationToken = _cancellationTokenSource.Token;
    }

    IService Service = ServiceLocator.Current.GetInstance<IService>();

    Service.GetSearchInfoAsync(cancellationToken, new[] {queryValue}, identifierType)
        .ContinueWith(
            task =>
            {
                // Do stuff
            }, CancellationToken.None, TaskContinuationOptions.None, TaskScheduler.Default);
}

问题是如果我使用上面详述的这条线，

_serviceMock.Setup(mock => mock.GetSearchInfoAsync(It.IsAny<CancellationToken>(), It.IsAny<IEnumerable<string>>(), It.IsAny<identifierType>(), It.IsAny<bool>())).Callback(() => _count++);

这会返回 null 并生成 NullPointerException：

Service.GetSearchInfoAsync(cancellationToken, new[] {queryValue}, identifierType)

但是，如果我注释掉该行，测试运行良好（尽管不计算调用次数）。

我做错了什么？这是我第一次为此使用起订量，据我所知，我已经正确实现了计数功能。

编辑：按照 Chris Sinclair 的建议，我已将初始化程序更改为此，从而解决了问题：

UnityContainer unityContainer = new UnityContainer();

_serviceMock = new Mock<IService>();
Task<IEnumerable<ISearchResult>> task = new Task<IEnumerable<ISearchResult>>(Enumerable.Empty<ISearchResult>);
_serviceMock.Setup(mock => mock.GetSearchInfoAsync(It.IsAny<CancellationToken>(), It.IsAny<IEnumerable<string>>(), It.IsAny<identifierType>(), It.IsAny<bool>())).Returns(task).Callback(() => _count++);
unityContainer.RegisterInstance(typeof(IService), _serviceMock.Object, new ContainerControlledLifetimeManager());

Answer 1

当您“设置”该方法时，您设置了一个回调，但您不提供返回值。因此，当调用模拟方法时，它将返回返回类型的默认值（在这种情况下，Task<> 类型将导致 null 返回值）。因此，当您的 Search 方法调用您模拟的 GetSearchInfoAsync 方法时，它会收到一个 null 引用，当它稍后尝试在其上调用 .ContinueWith 时自然会失败。

尝试添加一个.Returns()，它会为您的模拟方法提供一个虚拟Task<>：

_serviceMock.Setup(mock => mock.GetSearchInfoAsync(It.IsAny<CancellationToken>(), It.IsAny<IEnumerable<string>>(), It.IsAny<identifierType>(), It.IsAny<bool>()))
    .Returns(new Task<IEnumerable<ISearchResult>>(Enumerable.Empty<ISearchResult>))
    .Callback(() => _count++);