我有一个类似下面的代码:
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(),而是返回一个存根值.基本上,我想模拟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的新手.
解决方法:
您需要将调用构造函数的类放入@PrepareForTest注释而不是正在构造的类中 – 请参阅Mock construction of new objects.
在你的情况下:
✗@ PrepareForTest(MyQueryClass.class)
✓@ PrepareForTest(A.class)
更一般:
✗@ PrepareForTest(NewInstanceClass.class)
✓@ PrepareForTest(ClassThatCreatesTheNewInstance.class)