java、mock的单元测试，在某个测试方法里，能否对同一个mock的方法做不同的操作？

Question

1、后端开发代码
public void methodA(){

.......
.......
try {
    mapper.insert(user);
} catch (Exception e) {
    logger.error();
    ......
}

}

2、测试方法代码
public void methodATest(){

Mockito.when(mapper.insert(Mockito.any())).thenReturn(1);
xxx.methodA();

Mockito.when(mapper.insert(Mockito.any())).thenThrow(RuntimeException.class);
xxx.methodA();
}

3、问题
我现在想要覆盖代码里try……catch……的内容（执行通过和执行失败）。

从现在写的代码来看好像不能在同个测试方法里，第二次调用methodA()会报一个奇怪、不应该出现的错。如果把测试方法拆成两个，一个测试通过，一个测试失败，就没问题。

请问mapper.insert() 这个mock能否在同个测试方法里打两次或以上？如果可以要怎么写，如果不行，那是在方法里每次走到if判断或者try catch块，都单独写个方法出来验证吗？求指教谢谢。

Answer 1

@Test
public void methodATest() {
    // 第一次
    Mockito.doReturn(1).when(mapper).insert(Mockito.any());
    xxx.methodA();

    // 清除记录
    Mockito.clearInvocations(mapper);

    // 第二次
    Mockito.doThrow(new RuntimeException("Insert failed")).when(mapper).insert(Mockito.any());
    xxx.methodA();

    // 验证
}