Mockito+junit5搞定单元测试

package com.surpass.service; import com.surpass.entity.TeacherEntity; import com.surpass.entity.UserEntity; import org.junit.jupiter.api.Assertions; import org.junit.jupiter.api.Test; import org.junit.jupiter.api.extension.ExtendWith; import org.mockito.Mock; import org.mockito.Mockito; import org.mockito.invocation.InvocationOnMock; import org.mockito.junit.jupiter.MockitoExtension; import org.mockito.stubbing.Answer; import java.util.List; import java.util.Set; import static org.mockito.ArgumentMatchers.any; import static org.mockito.Mockito.*; /** * 参数匹配：通过方法签名（参数）来指定哪些方法调用时需要被处理（插桩、verify验证） * 注意：在只用匹配器是要么都用要么都不用，禁止混搭！！举例 * 正确：getUserListByTeacher("name", "address")或者getUserListByTeacher(anyString(), anyString()) * 错误：getUserListByTeacher(anyString(), "address") */ @ExtendWith(MockitoExtension.class) public class ParamMatcherTest { @Mock private TeacherService teacherService; /** * 对于mock对象不会调用真实方法，直接返回mock对象的默认值 * 默认值（int）、null、空集合 */ @Test public void test1(){ TeacherEntity teacher = teacherService.getTeacherByName(""); System.out.println("teacher = " + teacher); Set<UserEntity> userListByTeacherName = teacherService.getUserListByTeacherName(""); System.out.println("userListByTeacherName = " + userListByTeacherName); } /** * 方法插桩时的参数匹配 * 参数匹配时类（ArgumentMatchers）是匹配参数的主要成员，例如： * any()则表示某一个方法传入任何类型都符合要求 * anyString()则表示字符串类型都符合要求 * ！！！注意：所有的匹配都不包括null值 */ @Test public void test2(){ TeacherEntity teacherEntity = new TeacherEntity(); teacherEntity.setName("张三"); teacherEntity.setAddress("北京"); //when(teacherService.getTeacherByName("")).thenReturn(teacherEntity); 此行插桩方式也可 doReturn(teacherEntity).when(teacherService).getTeacherByName(any()); TeacherEntity teacher = teacherService.getTeacherByName("123"); System.out.println("teacher = " + teacher); //验证校验teacherService.getTeacherByName()调用的次数，之前调了一次所以校验通过，如两次则抛异常 verify(teacherService, times(2)).getTeacherByName(any()); } @Mock private List<String> mockList; /** * 被插桩的方法在调用时不会执行实际的逻辑，直接返回指定的返回值 */ @Test public void test3(){ /** * 指定返回值 */ //方法插桩：当调用mockList.get(0)时返回指定返回值“zero” doReturn("zero").when(mockList).get(0); Assertions.assertEquals("zero", mockList.get(0)); when(mockList.get(1)).thenReturn("one"); Assertions.assertEquals("one", mockList.get(1)); /** * void返回值方法插桩 */ doNothing().when(mockList).clear(); mockList.clear(); verify(mockList, times(1)).clear(); /** * 抛异常 */ doThrow(RuntimeException.class).when(mockList).clear(); try { mockList.clear(); //断言证明插桩失败 Assertions.fail(); } catch (Exception e) { Assertions.assertTrue(e instanceof RuntimeException); } //或 when(mockList.get(anyInt())).thenThrow(RuntimeException.class); try { mockList.get(4); //断言证明插桩失败 Assertions.fail(); } catch (Exception e) { Assertions.assertTrue(e instanceof RuntimeException); } /** * 多次插桩 * 意为第一次调用返回1，第二次调用返回2，第三次以及以后调用返回3 */ when(mockList.size()).thenReturn(1).thenReturn(2).thenReturn(3); //或-----两者相同 when(mockList.size()).thenReturn(1, 2, 3); Assertions.assertEquals(1, mockList.size()); Assertions.assertEquals(2, mockList.size()); Assertions.assertEquals(3, mockList.size()); Assertions.assertEquals(3, mockList.size()); } /** * thenAnswer来实现对指定逻辑的插桩 */ @Test public void test4() { when(mockList.get(anyInt())).thenAnswer((Answer<String>) invocation -> { //getArgument表示获取插桩方法（此处为mockList.get(anyInt())）的第几个参数值 Integer argument = invocation.getArgument(0, Integer.class); return String.valueOf(argument * 100); }); //执行get方法 System.out.println(mockList.get(1)); } /** * 调用真实逻辑 */ @Test public void test5() { when(teacherService.getTeacher(any())).thenCallRealMethod(); System.out.println(teacherService.getTeacher("123")); } }