如何模拟IAsyncEnumerable？

Question

我想对一个方法进行单元测试，该方法调用返回IAsyncEnumerable<T>的服务的另一个方法。我已经为我的服务Mock<MyService>创建了一个模拟，我想要setUp这个模拟，但我不知道如何做到这一点。有可能吗？是否有其他方法来测试一个方法，该方法调用了一个重新调整IAsyncEnumerable的东西？

public async Task<List<String>> MyMethodIWantToTest()
{
  var results = new List<string>();
  await foreach(var item in _myService.CallSomethingReturningAsyncStream())
  {
    results.Add(item);
  }
  return results;
}

Answer 1

如果您不想做任何特殊的事情，例如延迟返回(通常是异步枚举的点)，那么您可以只创建一个生成器函数来为您返回值。

public static async IAsyncEnumerable<string> GetTestValues()
{
    yield return "foo";
    yield return "bar";

    await Task.CompletedTask; // to make the compiler warning go away
}

这样，您就可以为您的服务创建一个模拟并测试您的对象：

var serviceMock = new Mock<IMyService>();
serviceMock.Setup(s => s.CallSomethingReturningAsyncStream()).Returns(GetTestValues);

var thing = new Thing(serviceMock.Object);
var result = await thing.MyMethodIWantToTest();
Assert.Equal("foo", result[0]);
Assert.Equal("bar", result[1]);

当然，由于您现在使用的是生成器函数，您还可以使这个过程变得更加复杂，并添加实际的延迟，甚至还可以包括一些机制来控制产量。

Answer 2

我建议使用来自System.Linq.Async的System.Linq.Async，正如Jeroen所建议的那样。看起来您使用的是Moq，因此看起来如下：

async Task MyTest()
{
  var mock = new Mock<MyService>();
  var mockData = new[] { "first", "second" };
  mock.Setup(x => x.CallSomethingReturningAsyncStream()).Returns(mockData.ToAsyncEnumerable());

  var sut = new SystemUnderTest(mock.Object);
  var result = await sut.MyMethodIWantToTest();

  // TODO: verify `result`
}

Answer 3

这取决于您使用的是哪个模拟框架。但是，使用Moq这样的例子可能很简单

var data = new [] {1,2,3,4};
var mockSvc = new Mock<MyService>();
mockSvc.Setup(obj => obj.CallSomethingReturningAsyncStream()).Returns(data.ToAsyncEnumerable());