如何在另一个方法中模拟方法调用

Question

这是我的代码：

`public class Service {
   callSomeService(ContextTpye1 c1,Parameter p){
   //do something here
   ContextTpye2 c2 = constructContext(c1,p);      
   if(c2.timeout()){
      c2.audit();
     }
   }
 }`

在单元测试中，初始化Service、Parameter和ContextTpye1。是否可以只模拟 c2.audit()？我想让它在这里什么都不做。谢谢。

Answer 1

因此，假设constructContext(c1,p) 是public，我解决问题的方法是在测试Service 类时监视Service，允许您模拟constructContext(c1,p) 并返回@ 的模拟987654326@.

public class ServiceTest
{
    private Service service = Mockito.spy(new Service());

    private ContextType2 c2 = Mockito.mock(ContextType2.class);

    @Test
    public void testCallSomeService()
    {
        Mockito.when(service.constructContext(Mockito.eq(ContextType1.class), Mockito.eq(Parameter.class))).thenReturn(c2);
            Mockito.when(c2.timeout()).thenReturn(true);
        Mockito.when(c2.audit()).doNothing();
            service.callSomeService(new ContextType1(), new Parameter());
    }
}