java - Mockito/PowerMock - 可以模拟 new MyClass() 吗？

Question

public void doSomething(){
    //...
    HibernateCursorItemReader<Object[]> reader = new HibernateCursorItemReader<>();
    //...
}

如何模拟阅读器，或创建新的HibernateCursorItemReader<>()返回一个模拟对象？

Answer 1

你可以包裹new HibernateCursorItemReader<>();使用方法并模拟它，例如:

 HibernateCursorItemReader<Object[]> getReader() {
      return new HibernateCursorItemReader<>();
 }

 public void doSomething(){
     //...
     HibernateCursorItemReader<Object[]> reader = getReader(); // use method invocation to get a reader
     //...
 }

现在，当你想 mock reader 时，模拟getReader()方法并返回您想要的任何内容:

MyObject object = ...; //your original object

MyObject spyObject = Mockito.spy(object);
Mockito.doReturn(/*your value*/).when(spyObject).getReader();