如何模拟 OperationContext.Current（WCF 消息）

Question

目前我面临对生产代码进行单元测试的挑战。我们有一个函数可以从传入的 WCF 消息中检索 IP 地址。

public void DoSomething(){
    var ipAddressFromMessage = GetIpFromWcfMessage();

    var IpAddress = IPAddress.Parse(ipAddressFromMessage);

    if(IpAddress.IsLoopback)
    {
        // do something 
    }
    else
    {
        // do something else
    }
}

private string GetIpFromWcfMessage()
{       
    OperationContext context = OperationContext.Current;
    string ip = ...//use the IP from context.IncomingMessageProperties to extract the ip

    return ip;    
}

问题是，我应该怎么做才能测试DoSomething()中的IP检查？

[Test]
Public void DoSomethingTest()
{
    //Arrange...
    // Mock OperationContext so that we can manipulate the ip address in the message

    // Assert.
    ...
}

我是否应该改变我使用 Operation 上下文的方式，以便我可以模拟它（例如，实现一个接口并模拟该接口的实现）？

Answer 1

我会用静态助手来包装调用：

public static class MessagePropertiesHelper
{
  private static Func<MessageProperties> _current = () => OperationContext.Current.IncomingMessageProperties;


  public static MessageProperties Current
  {
      get { return _current(); }
  }

  public static void SwitchCurrent(Func<MessageProperties> messageProperties)
  {
      _current = messageProperties;
  }

}

然后在GetIpFromWcfMessage我会打电话：

private string GetIpFromWcfMessage()
{       
    var props = MessagePropertiesHelper.Current;
    string ip = ...//use the IP from MessageProperties to extract the ip

    return ip;    
}

而且我可以在测试场景中切换实现：

[Test]
Public void DoSomethingTest()
{
    //Arrange...
    // Mock MessageProperties so that we can manipulate the ip address in the message    
    MessagePropertiesHelper.SwitchCurrent(() => new MessageProperties());

    // Assert.
    ...
}

在这里你可以找到我对类似问题的回答：https://stackoverflow.com/a/27159831/2131067。