c# – 模拟Task.Delay

我有一个方法,使用以下行：await Task.Delay(waitTime).ConfigureAwait(false);

我有一个很好的策略,可以避免在单元测试时等待几秒钟,而是确认我们试图等待特定的秒数.

例如,有没有办法在我的方法中注入一个额外的参数,就像在这个(人为的)示例中我注入一个虚构的ITaskWaiter接口的模拟对象：

// Arrange
var mockWait = new Mock<ITaskWaiter>(MockBehavior.Strict);
mockWait.Setup(w => w.Delay(It.Is<TimeSpan>(t => t.TotalSeconds == 2)));

// Act
myObject.MyMethod(mockWait.Object);

// Assert
mockWait.Verify();

解决方法

您可以像这样定义“delayer”界面：

public interface IAsyncDelayer
{
    Task Delay(TimeSpan timeSpan);
}

然后,您可以为生产代码提供以下实现：

public class AsyncDelayer : IAsyncDelayer
{
    public Task Delay(TimeSpan timeSpan)
    {
        return Task.Delay(timeSpan);
    }
}

现在,你的课程看起来像这样：

public class ClassUnderTest
{
    private readonly IAsyncDelayer asyncDelayer;

    public ClassUnderTest(IAsyncDelayer asyncDelayer)
    {
        this.asyncDelayer = asyncDelayer;
    }

    public async Task<int> MethodUndertest()
    {
        await asyncDelayer.Delay(TimeSpan.FromSeconds(2));

        return 5;
    }
}

这是Dependency Injection的基本应用.基本上,我们提取了异步等待不同类的逻辑,并为其创建了一个接口,以启用polymorphism.

在制作中,你会像这样组成你的对象：

var myClass = new ClassUnderTest(new AsyncDelayer());

现在,在您的测试中,您可以创建一个伪造的delayer,它会立即返回：

[TestMethod]
public async Task TestMethod1()
{
    var mockWait = new Mock<IAsyncDelayer>();

    mockWait.Setup(m => m.Delay(It.IsAny<TimeSpan>())).Returns(Task.Fromresult(0));

    var sut = new ClassUnderTest(mockWait.Object);

    var result = await sut.MethodUndertest();

    Assert.AreEqual(5,result);
}