模拟 UdpClient 进行单元测试

Question

我正在研究一个利用 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
}

。

重构我的 UdpCommsChannel 类以要求通过构造函数注入的 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());
}

。

欢迎任何进一步的cmets。

Answer 1

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

在您的类中，注入包装器并使用 is 而不是 tf UdpClient。

这样，您可以在测试中模拟 ISender 并进行测试。

当然，你不会对包装器本身进行测试，但如果它只是一个普通的呼叫转移，你就不需要这个。

基本上，您已经朝着正确的方向开始，但是您添加了一些自定义逻辑，无法通过这种方式进行测试-因此您需要将自定义逻辑和comm部分分开。

也就是说，你的班级变成这样：

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），那么您可以像上面解释的那样创建一个普通的包装器。

除了可测试性改进之外，这种方法将使您将来更容易修改代码 - 即当您需要其他一些通信方法时等。