Java单元测试：如何Mock同一个方法实现不同行为？

如何在 java 单元测试中对同一个 mock 方法进行不同的操作

在编写 java 单元测试时，有时需要对相同的 mock 方法进行不同的操作。例如，为了模拟不同的方法调用结果或行为。mockito 框架提供了方法来实现这一点。

可以使用 @test 注解定义一个测试方法，并将 mock 注解与mockito.when() 结合使用来指定在特定条件下返回特定值。但是，如果需要对同一方法执行多次不同的操作，则需要在每次操作之前清除 mock 的记录。

下面是一个示例，展示了如何在一个测试方法中对同一个 mock 方法进行不同的操作：

@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() 抛出异常，模拟失败的情况。通过在每次操作之间清除记录，我们可

以确保不会影响后续测试。