【Mockito】Mock对象多次调用方法，返回不同结果

hf寒沨

于 2021-01-30 15:59:49 发布

阅读量1w

点赞数 1

分类专栏： Java 文章标签： java

本文链接：https://blog.youkuaiyun.com/hc1017/article/details/113436579

版权

Java 专栏收录该内容

57 篇文章

订阅专栏

这篇博客介绍了如何创建ArgumentMatcher以根据不同的参数值匹配对应的行为。通过定义Predicate来判断ProductFlowFilter对象的id1属性，实现了Mockito的when()方法中针对不同条件返回不同mock结果的功能。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

创建ArgumentMatcher，根据不同的入参匹配对应行为

    /**
     * 创建ArgumentMatcher，根据不同参数值来匹配对应的行为
     */
    private <T> T createArgumentMatcher(final Predicate<T> predicate) {
        return Mockito.argThat(new ArgumentMatcher<T>() {
            @Override
            public boolean matches(Object argument) {
                if (null == argument) {
                    return false;
                }
                return predicate.apply((T) argument);
            }
        });
    }

实现方法

        Predicate<ProductFlowFilter> one = new Predicate<ProductFlowFilter>() {
            @Override
            public boolean apply(@Nullable ProductFlowFilter filter) {
                return "0".equals(filter.getId1());
            }
        };
        Predicate<ProductFlowFilter> two = new Predicate<ProductFlowFilter>() {
            @Override
            public boolean apply(@Nullable ProductFlowFilter filter) {
                return "-999999".equals(filter.getId1());
            }
        };
        Mockito.when(productFlowDao.getViewFlowData(createArgumentMatcher(one))).thenReturn(json1);
        Mockito.when(productFlowDao.getViewFlowData(createArgumentMatcher(two))).thenReturn(json2);