1

我正在使用一些遗留代码，我必须在其中实现一个新的处理程序。在这个处理程序中，不幸的是，框架使用一些硬编码的属性文件初始化了一个对象，如下所示：

// new code
public class NewHandler extends RootHandler {
  Util util = Util.getInstance(); // !!!Problem: throws NPE in unit-testing env

  // defined in the legacy framework, can't change its signature
  @Override
  public void doSth() {

  }
}

// legacy code
public class Util {
    static public synchronized Util getInstance() { 
        if (_instance == null) {
            _instance = new Util();
        }
        return _instance;
    }

    private Util() {
    MyObj obj = LegacyCode.load("myConfig.cfg"); // !!!Problem: throws NPE in unit-testing env
    ...

}
}

现在的问题
是：Util()它总是从固定位置加载配置文件，因此当我运行单元测试时它总是抛出 NPE，因为单元测试类路径上没有“myConfig.cfg”。

一种解决方法是将业务逻辑提取到一个单独的方法中，传入一个 Util 对象，以便我可以在测试期间传入模拟对象：

// new code
public class NewHandler extends RootHandler {
  Util util = null;

  // defined in the legacy framework, can't change its signature
  @Override
  public void handle() {
    Util util = util.getInstance();
    doHandle(util);
  }

  public void doHandle(Util util) {
    ...
  }
}

// legacy code
public class Util {
  ...
}

我的问题是：
有没有其他方法可以解决这个问题，而不添加 doHandle() 方法？我使用 Mockito 尝试了以下代码：

Util myUtil = mock(Util.class);
when(Util.getInstance()).thenReturn(myUtil);

但它没有用，Util.getInstance() 没有被 myUtil 替换。

有什么想法吗？

更新：事情变得有点混乱，因为我发现 Util.getInstance() 在遗留代码库中的任何地方都被调用。因此，即使我的新代码中有一个模拟的 Util（如下面的 Stas 所述），当在其他地方调用 Util.getInstance() 时，事情仍然会中断。
显然，我不能在调用 Util.getInstance() 的遗留代码中的每个类中添加新的构造函数。


这就是我问“是否有可能在 Mockito 中有类似 when(Sth.getInstance()).thenReturn(myMock) 的东西”
的原因 ，如果是，那么对 Util.getInstance() 的所有调用都可以被模拟并获胜不再引起问题。
嗯....有什么想法吗？

****************************************************** *********************
解决方案：我认为我的解决方案是PowerMock

4

1 回答 1

1

你应该试试 smt like

 public class NewHandler extends RootHandler {
     Util util;
     //Pass instance
     NewHandler(Util util) {this.util = util;}
 }

比您能够使用它new NewHandler(Util.getInstance())或new NewHandler(mock(Util.class))在您的测试中使用它。

PS 了解更多信息，您可以阅读有关服务定位器和DI的信息

于 2011-04-19T09:50:29.510 回答