Mockito单元测试自定义mock返回

妖狐艹你老母 2022-02-18 19:39 547阅读 0赞

在上篇博客[https://blog.csdn.net/cml\_blog/article/details/82820753][https_blog.csdn.net_cml_blog_article_details_82820753]介绍了SpringBoot中进行单元测试各种小技巧后，发现还有个遗漏的case，那就是在使用的mock的时候，可能需要根据不同的参数返回不同的数据。比如传入aa返回true，传入bb返回false等情况，这里再补充一篇进行说明。

首先还是在AService中新增doSmthing2

@Component
    public class AService {
    
        //略....
    
        public String doSmthing2(String key) {
            return "doSmthing2:" + key;
        }
    }

在单元测试的时候，如果使用到了AService.doSmthing2方法，希望传入不同参数时返回不同的mock数据，那么就可以使用Mockito.argThat进行数据的自定义返回：

@RunWith(SpringRunner.class)
    public class MockSelectorTest {
        @Mock
        private AService aService;
    
        @Test
        public void testMock() {
            Mockito.when(aService.doSmthing2(Mockito.anyString())).thenReturn("mockValue");
            assert aService.doSmthing2("value").equals("mockValue");
        }
    
        @Test
        public void testSelector() {
            //默认返回
            Mockito.when(aService.doSmthing2(Mockito.any())).thenReturn("mockANYValue");
            Mockito.when(aService.doSmthing2(Mockito.argThat(arg -> "aa".equals(arg)))).thenReturn("mockAAAValue");
            Mockito.when(aService.doSmthing2(Mockito.argThat(arg -> "bb".equals(arg)))).thenReturn("mockBBBValue");
    
            assert "mockAAAValue".equals(aService.doSmthing2("aa"));
            assert "mockBBBValue".equals(aService.doSmthing2("bb"));
            assert "mockANYValue".equals(aService.doSmthing2("cc"));
            assert "mockANYValue".equals(aService.doSmthing2(null));
        }
    
        @Test
        public void testSelectorMockitoHamcrest() {
            //默认返回
            Mockito.when(aService.doSmthing2(Mockito.any())).thenReturn("mockANYValue");
            Mockito.when(aService.doSmthing2(MockitoHamcrest.argThat(new BaseMatcher<String>() {
                @Override
                public boolean matches(Object item) {
                    return "aa".equals(item);
                }
    
                @Override
                public void describeTo(Description description) {
    
                }
            }))).thenReturn("mockAAAValue");
            Mockito.when(aService.doSmthing2(MockitoHamcrest.argThat(new BaseMatcher<String>() {
                @Override
                public boolean matches(Object item) {
                    return "bb".equals(item);
                }
    
                @Override
                public void describeTo(Description description) {
    
                }
            }))).thenReturn("mockBBBValue");
    
            assert "mockAAAValue".equals(aService.doSmthing2("aa"));
            assert "mockBBBValue".equals(aService.doSmthing2("bb"));
            assert "mockANYValue".equals(aService.doSmthing2("cc"));
            assert "mockANYValue".equals(aService.doSmthing2(null));
        }
    }

以上代码已经提交到工程：LearnTest

SpringBootLearning是对springboot学习与研究项目，是根据实际项目的形式对进行配置与处理，欢迎star与fork。  
\[oschina 地址\]  
[http://git.oschina.net/cmlbeliever/SpringBootLearning][http_git.oschina.net_cmlbeliever_SpringBootLearning]  
\[github 地址\]  
[https://github.com/cmlbeliever/SpringBootLearning][https_github.com_cmlbeliever_SpringBootLearning]

[https_blog.csdn.net_cml_blog_article_details_82820753]: https://blog.csdn.net/cml_blog/article/details/82820753
[http_git.oschina.net_cmlbeliever_SpringBootLearning]: http://git.oschina.net/cmlbeliever/SpringBootLearning
[https_github.com_cmlbeliever_SpringBootLearning]: https://github.com/cmlbeliever/SpringBootLearning