java - 使用 PowerMockito.whenNew() 不会被 mock ，而是调用原始方法

Question

我有一个类似下面的代码:

Class A {
  public boolean myMethod(someargs) {
    MyQueryClass query = new MyQueryClass();
    Long id = query.getNextId();
    // some more code
  }
}
Class MyQueryClass     {
  ....
  public Long getNextId() {
    //lot of DB code, execute some DB query
    return id;
  }
}

现在我正在为 A.myMethod(someargs) 编写测试。我想跳过真正的方法 query.getNextId() 而是返回一个 stub 值。基本上，我想模拟 MyQueryClass。

所以在我的测试用例中，我使用了:

MyQueryClass query = PowerMockito.mock(MyQueryClass.class);
PowerMockito.whenNew(MyQueryClass.class).withNoArguments().thenReturn(query);
when(query.getNextId()).thenReturn(1000000L);

boolean b = A.getInstance().myMethod(args);

//asserts

我在测试类的开头使用了 @RunWith(PowerMockRunner.class) 和 @PrepareForTest({MyQueryClass.class})。

但是我调试测试的时候，还是调用了MyQueryClass类的真实方法getNextId()。

我在这里缺少什么？任何人都可以提供帮助，因为我是 Mockito 和 PowerMockito 的新手。

Answer 1

您需要将调用 构造函数的类放入 @PrepareForTest 注释而不是正在构造的类 - 参见 Mock construction of new objects .

在你的情况下:

✗ @PrepareForTest(MyQueryClass.class)

✓ @PrepareForTest(A.class)

更笼统的:

✗ @PrepareForTest(NewInstanceClass.class)

✓ @PrepareForTest(ClassThatCreatesTheNewInstance.class)