无限轮询方法的单元测试

Question

我有一个方法，当应用程序启动时在新任务中启动，当它结束时停止。该方法每半秒执行一些逻辑。我不知道该如何测试它——有线索吗？

public async Task StartPoll(CancellationToken token)
{
  while(!token.IsCancellationRequested)
  {
    try 
    {
       var dict = dependecy.getDictionary();
       var model = Encoding.ASCII.GetBytes(JsonConvert.SerializeObject(dict));
       udpDistributer.Send(model, model.Length, _distributionEndPoint);
    }
    finally
    {
      await Task.Delay(500, Token);
    }
  }
}

其中，udpDistributer是一个来自DI的新UdpClient（0），而\u distributionEndPoint是一个iEndpoint

基本上，我想检查逻辑是否有效，以及是否以适当的间隔发送了正确的数据。如何使用xunit实现这一点？

编辑

我已经走到这一步了：

   // arrange
   ...
   cts = new CancellationTokenSource(timeout);

   //act
   await sut.StartPoll(cts.Token);

   // assert
   mockUdpDistributer.Verify(x => x.Send(expectedModel, expectedModel.length,
             It.IsAny<IPEndPoint>()), Times.Exactly(expectedTimes));

但是，这会导致测试失败，出现TaskCanceledException

Answer 1

UdpClient如果可能，应该包装在抽象中，StartPoll应该返回任务。

要测试是否实现了所需的行为，请使用令牌在已知时间后取消，并检查IUdpClient的次数。已调用发送

下面是一个过于简化的示例

[TestFixture]
public class MyTestClass {
    
    [Test]
    public async Task MyTestMethod() {
        //Arrange
        int expectedTimes = 3; 
        var timeout = TimeSpan.FromMilliseconds(1800);
        CancellationTokenSource cts = new CancellationTokenSource();
        var mockUdpDistributer = new  Mock<IUdpClient>();
        var sut = new MyClass(mockUdpDistributer.Object);

        //Act
        Task start = sut.StartPoll(cts.Token);
        await Task.Delay(timeout);
        cts.Cancel();

        //Assert
        mockUdpDistributer
            .Verify(x => 
                x.Send(It.IsAny<byte[]>(), It.IsAny<int>(), It.IsAny<IPEndPoint>())
                , Times.AtLeast(expectedTimes)
            );
    }
}

使用抽象客户端

public interface IUdpClient {
    void Send(byte[] datagram, int bytes, System.Net.IPEndPoint endPoint);
}

可以被嘲笑并注射到被测试者体内

class MyClass {
    private readonly IUdpClient udpDistributer;
    private readonly IPEndPoint _distributionEndPoint;

    public MyClass(IUdpClient udpDistributer) {
        this.udpDistributer = udpDistributer;
    }

    public async Task StartPoll(CancellationToken token) {
        while (!token.IsCancellationRequested) {
            try {
                object dict = new object();//<-- dependecy.getDictionary();
                byte[] model = Encoding.ASCII.GetBytes(JsonConvert.SerializeObject(dict));
                udpDistributer.Send(model, model.Length, _distributionEndPoint);
            } finally {
                await Task.Delay(500, token);
            }
        }
    }
}

Answer 2

此留档异步编程单元测试异步代码可能正是您正在寻找的。

实际上，它说的是使用异步测试方法并等待函数：

[TestMethod]
public async Task CorrectlyFailingTest()
{
  await SystemUnderTest.FailAsync();
}

还要尽量避免像这样的签名者private async void StartPoll。最好总是从异步代码返回Task。

作为如何测试调用的一部分，我还假设udpDistributer是实际实现的接口。因此，您可以模拟/计数/设置模拟接口，以确保其按预期工作。使用Moq模拟单元测试的异步方法

要使测试在某一点停止，您可以使用取消令牌取消异步任务后一段时间

Answer 3

您可以验证udpDistributer。Send（）（我假设您已经注入了，如果没有，那么就这样做）在您需要它的时间内被调用适当的次数，然后取消令牌。