【Java】Mockito捕获多次调用的方法的参数

整理Mockito结合ArgumentCaptor捕获多次调用的方法的参数

今天在对业务代码进行开发时,发现需要捕获多次调用的方法的参数,验证参数传递的正确性,这里对获取方法进行总结。

首先,模拟一个简单的测试场景,有一个对外提供服务的类:

@Component
public class CaptorLearning {

    private DalService dalService;

    @Autowired
    public CaptorLearning(DalService dalService) {
        this.dalService = dalService;
    }

    public int getResult(int num1, int num2, int num3) {
        int result1 = dalService.doHandle(num1, num2);
        int result2 = dalService.doHandle(num1, num3);
        return result1 + result2;
    }
}

有一个内部的业务处理类:

@Component
public class DalService {
    public int doHandle(int num1, int num2) {
        return num1 + num2;
    }
}

这里要对CaptorLearning进行单元测试覆盖,因此需要判断dalService两次调用了doHandle方法,并获取相应的参数,判断是否参数调用正确。

首先是两次调用,可以利用Mockito提供的verify方法,来判断调用次数,代码如下:

	@Test
    public void testByVerify() {
        captorLearning.getResult(num1, num2, num3);
        Mockito.verify(dalService, times(2)).doHandle(Mockito.anyInt(), Mockito.anyInt());
    }

如果此时,要判断两次调用doHandle方法的参数,可以利用ArgumentCaptorcapture方法进行模拟,再通过getAllValues方法来获取所有的参数值,代码如下:

    @Test
    public void testWithArgumentCaptor() {
        captorLearning.getResult(num1, num2, num3);
        ArgumentCaptor<Integer> integerArgumentCaptor = ArgumentCaptor.forClass(Integer.class);
        Mockito.verify(dalService, times(2)).doHandle(Mockito.anyInt(), integerArgumentCaptor.capture());
        
        // 拿到ArgumentCaptor捕获的参数List,通过下标,按照调用时机,依次取出
        List<Integer> list = integerArgumentCaptor.getAllValues();
        Assert.assertEquals(num2, list.get(0).intValue());
        Assert.assertEquals(num3, list.get(1).intValue());
    }

有时候可能我们捕获的虽然是方法调用的同一个参数,但有可能参数类型会不同,例如还是上面的场景,两个类都新增了方法:

@Component
public class DalService {
    public int doHandle(int num1, int num2) {
        return num1 + num2;
    }

    public <T> void doHandle(T data1, T data2, T data3, Class<T> clazz) {
        // doing ...
    }
}

@Component
public class CaptorLearning {
    private DalService dalService;

    @Autowired
    public CaptorLearning(DalService dalService) {
        this.dalService = dalService;
    }

    public int getResult(int num1, int num2, int num3) {
        int result1 = dalService.doHandle(num1, num2);
        int result2 = dalService.doHandle(num1, num3);
        return result1 + result2;
    }

    public void getResult(int num1, int num2, int num3, String string1, String string2, String string3) {
        dalService.doHandle(num1, num2, num3, Integer.class);
        dalService.doHandle(string1, string2, string3, String.class);
    }
}

这时候如果还是使用上面的测试方式,则会报错,因为String类型没办法转成Integer类型。

这里一种简单的解决办法,就是创建两个ArgumentCaptor,分别进行获取,再根据特定的下标,来获取某次调用时的捕获的参数,代码如下:

    @Test
    public void testWithTwoArgumentCaptor() {
        captorLearning.getResult(num1, num2, num3, string1, string2, string3);
        // 参数捕获器1,用于捕获参数类型为Integer类型的参数
        ArgumentCaptor<Integer> integerArgumentCaptor = ArgumentCaptor.forClass(Integer.class);
        Mockito.verify(dalService, times(2)).doHandle(integerArgumentCaptor.capture(), any(), any(), any());
        // 第一次调用时,参数类型为Integer类型,取出值进行校验
        Assert.assertEquals(num1, integerArgumentCaptor.getAllValues().get(0).intValue());

        // 参数捕获器2,用于捕获参数类型为String类型的参数
        ArgumentCaptor<String> stringArgumentCaptor = ArgumentCaptor.forClass(String.class);
        Mockito.verify(dalService, times(2)).doHandle(stringArgumentCaptor.capture(), any(), any(), any());
        // 第二次调用时,参数类型为String类型,取出值进行校验
        Assert.assertEquals(string1, stringArgumentCaptor.getAllValues().get(1));
    }

总结:

利用ArgumentCaptorcapture方法进行mock,再通过getAllValues方法,可以捕获多次调用的方法的参数。

  • 4
    点赞
  • 6
    收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
©️2022 CSDN 皮肤主题:数字20 设计师:CSDN官方博客 返回首页
评论

打赏作者

卤小蛋学编程

你的鼓励将是我创作的最大动力

¥2 ¥4 ¥6 ¥10 ¥20
输入1-500的整数
余额支付 (余额:-- )
扫码支付
扫码支付:¥2
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值