Hello! 欢迎来到小浪资源网!


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


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

如何在 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() 抛出异常,模拟失败的情况。通过在每次操作之间清除记录,我们可以确保不会影响后续测试。

相关阅读