临时锁定端口以进行测试

Question

在我的项目中，我有一个类可以检查 localhost 的所有端口，如果端口空闲则返回 true，否则返回 false。解决方案基于this answer

我需要以某种方式对其进行测试。有没有办法暂时使用该端口，然后将其释放回来？

我尝试使用 TcpListener Start 没有成功。如果我使用 TcpClient Connect 我得到 System.Net.Sockets.SocketException 无法建立连接，因为目标机器主动拒绝了它 127.0.0.1:myPort。

这是我使用的代码：

class TestPortLocker
{
    public TcpClient client;
    public TcpListener listener;

    public void Lock(int port)
    {
        client = new TcpClient();
        client.Connect("localhost", port);
        //listener = new TcpListener(port);
        //listener.Start();
    }

    public void Stop()
    {
        client.Close();
        //listener.Stop();
    }
}

Answer 1

您可以编写单元测试来检查您的代码。这是您的单元测试方法和被测方法的一些伪代码。

假设您有一个要测试的此类的方法：

public class PortManager 
{
    private _portChecker = portChecker;
    public PortManager(IPortChecker portChecker)
    {
       _portChecker = portChecker;
    }

    public bool PortsAreFree(int start, int end)
    {
        for(int i = start, i <= end; i++)
            if(!_portChecker.IsPortOpen(i))
                 return false;
        return true;
    }
}

现在要测试此方法，您可以使用 IPortChecker 模拟：

[TestFixture]
public class PortManagerTests
{
    [Test]
    public void PortsAreFree_APortIsLocked_ReturnsFalse()
    {
          var mock = new Mock<IPortChecker>();
          mock.Setup(x=> x.IsPortOpen(77)).Returns(false);

          var portManager = new PortManager(mock.Object);

          Assert.Equal(true, portManager.PortsAreFree(1, 76));
          Assert.Equal(true, portManager.PortsAreFree(78, 200));
          Assert.Equal(false, portManager.PortsAreFree(1, 200));
    }
}

}