模拟UdpClient进行单元测试

本文关键字:单元测试 UdpClient 模拟 | 更新日期: 2023-09-27 18:03:28

我正在研究一个使用UdpClient的类,并试图在此过程中使用NUnit和Moq学习/利用TDD方法。

到目前为止,我的类的基本部分如下:

public UdpCommsChannel(IPAddress address, int port)
{
    this._udpClient = new UdpClient();
    this._address = address;
    this._port = port;
    this._endPoint = new IPEndPoint(address, port);
}
public override void Open()
{
    if (this._disposed) throw new ObjectDisposedException(GetType().FullName);
    try
    {
        this._udpClient.Connect(this._endPoint);
    }
    catch (SocketException ex)
    {
        Debug.WriteLine(ex.Message);
    }
}
public override void Send(IPacket packet)
{
    if (this._disposed) throw new ObjectDisposedException(GetType().FullName);
    byte[] data = packet.GetBytes();
    int num = data.Length;
    try
    {
        int sent = this._udpClient.Send(data, num);
        Debug.WriteLine("sent : " + sent);
    }
    catch (SocketException ex)
    {
        Debug.WriteLine(ex.Message);
    }
}



。对于Send方法,我现在有以下单元测试:

[Test]
public void DataIsSent()
{
    const int port = 9600;
    var mock = new Mock<IPacket>(MockBehavior.Strict);
    mock.Setup(p => p.GetBytes()).Returns(new byte[] { }).Verifiable();
    using (UdpCommsChannel udp = new UdpCommsChannel(IPAddress.Loopback, port))
    {
        udp.Open();
        udp.Send(mock.Object);
    }
    mock.Verify(p => p.GetBytes(), Times.Once());
}



。我对此并不满意,因为它使用的是实际的IP地址,尽管只有本地主机,而类中的UdpClient正在物理地向它发送数据。因此,据我所知,这不是一个真正的单元测试。

问题是,我不知道该怎么做。我是否应该更改我的类并传入一个新的UdpClient作为依赖?以某种方式模拟IPAddress ?

有点挣扎,所以我需要在继续之前停下来看看我是否在正确的轨道上。任何建议感谢!

(使用NUnit 2.5.7, Moq 4.0和c# WinForms)
。.

更新:

好的,我已经重构了我的代码如下:

创建IUdpClient接口:

public interface IUdpClient
{
    void Connect(IPEndPoint endpoint);
    int Send(byte[] data, int num);
    void Close();
}

创建一个适配器类来包装UdpClient系统类:

public class UdpClientAdapter : IUdpClient
{
    private UdpClient _client;
    public UdpClientAdapter()
    {
        this._client = new UdpClient();
    }
    #region IUdpClient Members
    public void Connect(IPEndPoint endpoint)
    {
        this._client.Connect(endpoint);
    }
    public int Send(byte[] data, int num)
    {
        return this._client.Send(data, num);
    }
    public void Close()
    {
        this._client.Close();
    }
    #endregion
}

重构了我的udpcommchannel类,要求通过构造函数注入一个IUdpClient的实例:

public UdpCommsChannel(IUdpClient client, IPEndPoint endpoint)
{
    this._udpClient = client;
    this._endPoint = endpoint;
}

我的单元测试现在看起来像这样:

[Test]
public void DataIsSent()
{
    var mockClient = new Mock<IUdpClient>();
    mockClient.Setup(c => c.Send(It.IsAny<byte[]>(), It.IsAny<int>())).Returns(It.IsAny<int>());
    var mockPacket = new Mock<IPacket>(MockBehavior.Strict);
    mockPacket.Setup(p => p.GetBytes()).Returns(new byte[] { }).Verifiable();
    using (UdpCommsChannel udp = new UdpCommsChannel(mockClient.Object, It.IsAny<IPEndPoint>()))
    {
        udp.Open();
        udp.Send(mockPacket.Object);
    }
    mockPacket.Verify(p => p.GetBytes(), Times.Once());
}

欢迎进一步评论

模拟UdpClient进行单元测试

如果你只想测试你的类的功能,我将去创建一个接口iccommunicator,然后创建一个类UdpCommunicator,它直接包装所需的UdpClient属性和方法,不需要任何检查和条件。

在你的类中,注入包装器并使用is代替tf UdpClient。

这样,在测试中您可以模拟ISender,并测试。

当然,您将不需要对包装器本身进行测试,但如果它只是一个普通的调用转发,则不需要这样做。

基本上,你已经开始在正确的方向,但你已经添加了一些自定义逻辑,这是无法测试的方式-所以你需要拆分自定义逻辑和通信部分。

。,你的类变成这样:

public MyClass(ICommunicator comm)
{
     public void Method1(someparam)
     {
         //do some work with ICommunicator
     }
     ....
}

你的测试将看起来像:

var mockComm = Mock.GetMock<ICommunicator>();
mockComm.Setup ....
var myTestObj = new MyClass(mock.Object);
MyClass.Method1(something);
mock.Verify....

因此,在几个验证语句中,您可以检查通信器上的Open是否被调用,是否传递了正确的数据,等等。

一般-你不需要测试系统或第三方类,只需要测试你自己的。

如果你的代码使用这样的类,让它们是可注入的。如果这些类没有虚拟方法(也就是说,你不能直接模拟它们),或者没有实现一些常见的接口(比如SqlConnection等),那就用它吧。实现IDbConnection),然后像上面解释的那样创建一个普通包装器。

除了可测试性的改进,这样的方法将使它更容易修改你的代码在未来-即当你需要一些其他的通信方法,等等