具有 Dns.GetHostEntry(hostNameOrIp) 依赖项的单元测试代码

Question

我有使用 Dns.GetHostEntry(hostNameOrIp) 的代码，我想检查一下场景，有一次这个函数返回实际值但是在某个时间（当我决定时）这个函数抛出异常。
目前我在 Visual Studio 2010 中使用 MSTest 框架。 有人知道实现它的最简单方法是什么？

Answer 1

实现它的最简单方法是为此静态方法创建包装器：

public class DnsWrapper : IDnsWrapper
{
    public IPHostEntry GetHostEntry(string hostNameOrAddress)
    {
         Dns.GetHostEntry(hostNameOrAddress);
    }
}

并使你的代码依赖于这个接口：

public interface IDnsWrapper
{
   IPHostEntry GetHostEntry(string hostNameOrAddress);
}

现在使用任何模拟库都可以很容易地模拟这种依赖关系。例如Moq:

Mock<IDnsWrapper> dnsMock = new Mock<IDnsWrapper>();
dnsMock.Setup(d => d.GetHostEntry(It.IsAny<string>()))
       .Throws(new SocketException());

var yourClass = new YourClass(dnsMock.Object); // inject interface implementation
yourClass.DoSomethingWhichGetHostsEntry();