Junit+Mock

base project

1. <parent>
2.     <groupId>org.springframework.boot</groupId>
3.      <artifactId>spring-boot-starter-parent</artifactId>
4.      <version>2.6.11</version>
5.      <relativePath/>
6. </parent>
8. <dependencies>
9.    <!--添加mysql依赖-->
10.     <dependency>
11.         <groupId>org.mariadb.jdbc</groupId>
12.         <artifactId>mariadb-java-client</artifactId>
13.         <version>2.7.2</version>
14.     </dependency>
16.     <dependency>
17.         <groupId>com.baomidou</groupId>
18.         <artifactId>mybatis-plus-boot-starter</artifactId>
19.         <version>3.5.2</version>
20.     </dependency>
22.     <!--单元测试-->
23.     <dependency>
24.         <groupId>org.springframework.boot</groupId>
25.         <artifactId>spring-boot-starter-test</artifactId>
26.     </dependency>
27.     <!--junit4-->
28.     <dependency>
29.         <groupId>org.junit.vintage</groupId>
30.         <artifactId>junit-vintage-engine</artifactId>
31.     </dependency>
33.     <dependency>
34.         <groupId>org.springframework.boot</groupId>
35.         <artifactId>spring-boot-starter-web</artifactId>
36.     </dependency>
37.    
38.     <dependency>
39.         <groupId>org.projectlombok</groupId>
40.         <artifactId>lombok</artifactId>
41.     </dependency>
42. </dependencies>

复制代码

application.yaml

1. server:
2.   port: 8896
4. spring:
5.   datasource:
6.     username: root
7.     password: root
8.     url: jdbc:mariadb://local_IP_replace:3306/junit?serverTimezone=GMT%2B8&useSSL=false&allowPublicKeyRetrieval=true
9.     driver-class-name: org.mariadb.jdbc.Driver

复制代码

controller

1. @RestController
2. @RequestMapping("/ele")
3. public class EleController {
5.     private final IEleService eleService;
8.     @Autowired
9.     public EleController(IEleService eleService) {
10.         this.eleService = eleService;
11.     }
13.     @GetMapping(value = "/selectEleById")
14.     public Result<Object> selectEleById(@RequestParam(value = "orderId") int orderId) {
15.         Object object = eleService.selectEleById(orderId);
16.         return Result.success(object);
17.     }
19.     @GetMapping(value = "/selectEleByIdAndName")
20.     public Result<Object> selectEleByIdAndName(
21.             @RequestParam(value = "id") int id, @RequestParam(value = "name") String name) {
22.         Object object = eleService.selectEleByIdAndName(id,name);
23.         return Result.success(object);
24.     }
25. }

复制代码

service

1. public interface IEleService {
2.     Object selectEleById(int orderId);
3.     Object selectEleByIdAndName(int id, String name);
4. }

复制代码

1. @Service
2. public class EleService implements IEleService {
3.     @Autowired
4.     private IElementsMapper elementsMapper;
5.     @Override
6.     public Object selectEleById(int orderId) {
7.         ElementsDO elementsDO = elementsMapper.selectById(orderId);
8.         return elementsDO;
9.     }
11.     @Override
12.     public Object selectEleByIdAndName(int id, String name) {
13.         LambdaQueryWrapper<ElementsDO> queryWrapper = new LambdaQueryWrapper<>();
14.         queryWrapper.eq(ElementsDO::getId, id);
15.         queryWrapper.eq(ElementsDO::getElementName, name);
16.         ElementsDO elementsDO = elementsMapper.selectOne(queryWrapper);
17.         return elementsDO;
18.     }
19. }

复制代码

mapper

1. @Mapper
2. public interface IElementsMapper extends BaseMapper<ElementsDO> {
3. }

复制代码

entity

1. @Data
2. @Builder
3. @NoArgsConstructor
4. @AllArgsConstructor
5. @EqualsAndHashCode(callSuper = false)
6. @TableName("ELEMENTS")
7. public class ElementsDO implements Serializable {
8.     /**
9.      * id
10.      */
11.     @TableId(value = "ID", type = IdType.AUTO)
12.     private Integer id;
13.     /**
14.      * 要素代码
15.      */
16.     @TableField(value = "ELEMENT_CODE")
17.     private String elementCode;
18.     /**
19.      * 要素名称
20.      */
21.     @TableField(value = "ELEMENT_NAME")
22.     private String elementName;
23.     /**
24.      * 单位
25.      */
26.     @TableField(value = "UNIT")
27.     private String unit;
28. }

复制代码

controller返回封装对象

1. @JsonInclude(JsonInclude.Include.NON_NULL)
2. public class Result<T> {
3.     private int code;
4.     private String msg="success";
5.     private T data;
7.     /**
8.      *  成功时候的调用
9.      * */
11.     public static <T> Result<T> success() {
12.         Result<T> result = new Result<T>();
13.         result.code = 1;
14.         return result;
15.     }
17.     public static<T> Result<T> success(T data){
18.         return new Result<T>(data);
19.     }
21.     /**
22.      *  失败时候的调用
23.      * */
24.     public static  <T> Result<T> error(CodeMsg codeMsg){
25.         return new Result<T>(codeMsg);
26.     }
28.     private Result(T data) {
29.         this.data = data;
30.     }
32.     private Result(int code, String msg) {
33.         this.code = code;
34.         this.msg = msg;
35.     }
36.     private Result() {
37.     }
39.     private Result(CodeMsg codeMsg) {
40.         if(codeMsg != null) {
41.             this.code = codeMsg.getCode();
42.             this.msg = codeMsg.getMsg();
43.         }
44.     }
47.     /**
48.      * BindingResult统一处理
49.      */
50.     public static Result resolveBindResult(BindingResult bindingResult){
51.         StringBuilder stringBuilder = new StringBuilder();
52.         for (String s : bindingResult.getFieldErrors().stream().map(FieldError::getDefaultMessage).collect(Collectors.toList())) {
53.             stringBuilder.append(",").append(s);
54.         }
55.         return Result.error(new CodeMsg(502,stringBuilder.toString().substring(1)));
56.     }
58.     public int getCode() {
59.         return code;
60.     }
61.     public void setCode(int code) {
62.         this.code = code;
63.     }
64.     public String getMsg() {
65.         return msg;
66.     }
67.     public void setMsg(String msg) {
68.         this.msg = msg;
69.     }
70.     public T getData() {
71.         return data;
72.     }
73.     public void setData(T data) {
74.         this.data = data;
75.     }
77.     @Override
78.     public String toString() {
79.         return "Result [code=" + code + ", msg=" + msg + ", data=" + data + "]";
80.     }
82. }

复制代码

Mock初始化

方式1

1. import org.junit.jupiter.api.Test;
2. import org.junit.jupiter.api.extension.ExtendWith;
3. import org.mockito.Mock;
4. import org.mockito.Mockito;
5. import org.mockito.Spy;
6. import org.mockito.junit.jupiter.MockitoExtension;
8. @ExtendWith(MockitoExtension.class)
9. public class Init1 {
10.     //初始化模拟对象
11.     @Mock
12.     private EleService eleService;
13.     //初始化真实对象
14.     @Spy
15.     private EleService eleService2;
17.     @Test
18.     public void test1(){
19.         boolean mock = Mockito.mockingDetails(eleService).isMock();
20.         System.out.println(mock);
22.         boolean mock1 = Mockito.mockingDetails(eleService2).isSpy();
23.         System.out.println(mock1);
24.     }
25. }

复制代码

方式2

1. import org.junit.jupiter.api.BeforeEach;
2. @ExtendWith(MockitoExtension.class)
3. public class Init2 {
4.     private EleService eleService;
5.     //初始化
6.     @BeforeEach
7.     public void init() {
8.         eleService = Mockito.mock(EleService.class);
9.     }
11.     @Test
12.     public void test1(){
13.         //判断对象是否为muck对象
14.         boolean mock = Mockito.mockingDetails(eleService).isMock();
15.         System.out.println(mock);
16.     }
17. }

复制代码

方式3

1. import org.junit.jupiter.api.BeforeEach;
2. @ExtendWith(MockitoExtension.class)
3. public class Init3 {
4.     //初始化模拟对象
5.     @Mock
6.     private EleService eleService;
7.     //初始化真实对象
8.     @Spy
9.     private EleService eleService2;
11.     //初始化
12.     @BeforeEach
13.     public void init() {
14.         //初始化当前类对象下的相关mock
15.         MockitoAnnotations.openMocks(this);
16.     }
18.     @Test
19.     public void test1(){
20.         boolean mock = Mockito.mockingDetails(eleService).isMock();
21.         System.out.println(mock);
23.         boolean mock1 = Mockito.mockingDetails(eleService2).isSpy();
24.         System.out.println(mock1);
25.     }
26. }

复制代码

Mock默认返回值

int默认返回0，对象默认返回null，聚集默认返回空聚集（空的ArrayList、空的HashSet、空的HashMap）

1. @ExtendWith(MockitoExtension.class)
2. public class MockMethod1 {
3.     //初始化模拟对象
4.     @Mock
5.     private EleService eleService;
7.     @Test
8.     public void test1(){
9.         //Mock对象返回默认值,不会调用真正方法
10.         Object o1 = eleService.selectEleById(1);//null
11.         System.out.println(o1);
12.     }
13. }

复制代码

Mock参数设置

1. @ExtendWith(MockitoExtension.class)
2. public class MockMethod1 {
3.     //初始化模拟对象
4.     @Mock
5.     private EleService eleService;
7.     @Test
8.     public void test1(){
9.         /**
10.          * 插桩,只拦截参数为1
11.          */
12.         //指定参数为1时,selectEleById方法时返回new Object()
13.         Mockito.doReturn(new Object()).when(eleService).selectEleById(1);
14.         Object o2 = eleService.selectEleById(1);
15.         System.out.println(o2);//java.lang.Object@16b2bb0c
16.     }
18.     @Test
19.     public void test2(){
20.         /**
21.          * 插桩,拦截selectEleById()的任意参数值
22.          */
23.         Mockito.doReturn(new Object()).when(eleService).selectEleById(ArgumentMatchers.any(Integer.class));
24.         Object o2 = eleService.selectEleById(2);
25.         System.out.println(o2);//java.lang.Object@c7045b9
26.     }
27. }

复制代码

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。