java - 使用 PowerMockito.whenNew() 不会被模拟并调用原始方法

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)