mockito 参数匹配的三种情况

最新推荐文章于 2024-06-22 21:29:26 发布

白衣若尘

最新推荐文章于 2024-06-22 21:29:26 发布

阅读量8.6k

点赞数 2

分类专栏： junit 文章标签： mock mockito junit 参数匹配 java

本文链接：https://blog.csdn.net/besto229/article/details/106102610

版权

junit 专栏收录该内容

6 篇文章 0 订阅

订阅专栏

文章目录

mock方法时，总会因场景不同，调整参数的匹配情况。比如：某个参数是为了在数据库进行模糊查询的；或则精确匹配时，引用类型参数未实现equals方法。这时候就需要mock方法时，进行不同匹配。

1. 参数精确匹配

    Map mockMap = Mockito.mock(Map.class);
    // 1.精确匹配
    Mockito.when(mockMap.get(11)).thenReturn(111); // 基础类型
    Mockito.when(mockMap.get(Lists.newArrayList("袁紫霞"))).thenReturn("白玉京"); // 引用类型
    TestCase.assertEquals(111, mockMap.get(11));
    TestCase.assertEquals("白玉京", mockMap.get(Lists.newArrayList("袁紫霞")));

2. 参数模糊匹配

    //多个参数，其中一个定义了匹配规则(Matcher)，其他的全要Matcher
    Mockito.when(mockMap.get(Mockito.endsWith("天"))).thenReturn("龙傲天"); // 字符串。eg:以天结尾的
    Mockito.when(mockMap.get(Mockito.anyLong())).thenReturn(999L); // 基础类型. eg:任何long类型
    Mockito.when(mockMap.get(Mockito.any(User.class))).thenReturn(new User()); // 引用类型
    TestCase.assertEquals("龙傲天", mockMap.get("星期天"));
    TestCase.assertEquals(999L, mockMap.get(1L));
    TestCase.assertEquals(new User(), mockMap.get(new User()));

3. 参数自定义匹配

Mockito.argThat（ArgumentMatcher< T>）：

    // 3.自定义匹配。 eg：定义只匹配PageParam的pageNo属性
    PageParam pageParam = PageParam.create(1, 20);
    ArgumentMatcher<PageParam> argPage = (page) -> page.getPageNo() == pageParam.getPageNo();
    Mockito.when(mockMap.get(Mockito.argThat(argPage))).thenReturn(Lists.newArrayList("袁紫霞", "白玉京"));
    TestCase.assertEquals(Lists.newArrayList("袁紫霞", "白玉京"), mockMap.get(PageParam.create()));