如何在 Java 单元测试中对同一个 mock 方法进行不同的操作
在编写 java 单元测试时,有时需要对相同的 mock 方法进行不同的操作。例如,为了模拟不同的方法调用结果或行为。mockito 框架提供了方法来实现这一点。
可以使用 @test 注解定义一个测试方法,并将 mock 注解与mockito.when() 结合使用来指定在特定条件下返回特定值。但是,如果需要对同一方法执行多次不同的操作,则需要在每次操作之前清除 mock 的记录。
下面是一个示例,展示了如何在一个测试方法中对同一个 mock 方法进行不同的操作:
立即学习“Java免费学习笔记(深入)”;
@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(); // 验证 }
在这个例子中,methoda() 方法调用了 mapper 的 insert() 方法,而 mapper 是被 mock 掉的。第一次,doreturn(1) 指定 insert() 返回 1。第二次,dothrow() 指定 insert() 抛出异常,模拟失败的情况。通过在每次操作之间清除记录,我们可以确保不会影响后续测试。