springBoot中使用Junit和Mockito进行单元测试

概述

比如现在有一个场景。需要测试一个方法，但是该方法里面有一个分支方法，这个方法需要调用外部接口方法，现在接口还未实现，我们实际测试的时候想跳过这个方法继续往后执行。那么我们就可以通过Mockito对外部服务进行模拟、打桩，从而达到对测试方法的完整测试。

过程

需要注意的语句：
Mockito.when(A).thenReturn(B);

需要注意的注解：
@InjectMocks
说明：该注解跟@Autowired类似，注入一个服务。
@Mock
说明：注入一个服务，该服务会是“假对象”，并不会真正执行。

ApplicationTest.java

package com.shushan;

import com.shushan.dao.UserHandleDao;
import com.shushan.entity.Student;
import com.shushan.service.impl.UserHandleServiceImpl;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.Mockito;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.junit4.SpringRunner;

import java.util.ArrayList;
import java.util.List;

@RunWith(SpringRunner.class)
@SpringBootTest
public class ApplicationTest {

    @InjectMocks
    private UserHandleServiceImpl userHandleServiceImpl;

    @Mock
    private UserHandleDao userHandleDao;

   /**
     * 定义需要“假执行”的方法，并预先设定好该方法需要返回的值。当底层执行到该方法时，
     * 不会真正执行方法并返回自己预先设定好的值。
     * 注意点：实际调用方法传入的参数必须跟预先设定“假执行”的方法参数一致，否则不生效。
     * 如下：预先定义“假执行”userHandleDao.findUserListByCode("1001")的方法参数是1001，
     * 那么userHandleServiceImpl.findUserListByCode("1001")传入的但是也必须为1001
     */
    @Test
    public void myTest1() {
        List list = new ArrayList();
        Student stu = new Student();
        stu.setId(111);
        stu.setName("测试姓名");
        list.add(stu);
        Mockito.when(userHandleDao.findUserListByCode("1001")).thenReturn(list);
        List relist = userHandleServiceImpl.findUserListByCode("1001");
        System.out.println(relist);
    }

    
    /**
     * 另一种情况：service层跟dao层方法参数实在做不到一致。
     *  如下：service层userHandleServiceImpl.findUserByParam("1004", "李四");
     *        dao层userHandleDao.findUserAll(Student);
     *   这种情况我们使用Mockito.isA(Student.class); 定义只有方法参数
     *   是类对象Student就行
     */
    @Test
    public void myTest2() {
        List list = new ArrayList();
        Student stu2 = new Student();
        stu2.setName("李四");
        list.add(stu2);
        Mockito.when(userHandleDao.findUserAll(Mockito.isA(Student.class))).thenReturn(list);
        List relist = userHandleServiceImpl.findUserByParam("1004", "李四");
    }

}

UserHandleService.java

package com.shushan.service;

import com.shushan.entity.Student;

import java.util.List;

public interface UserHandleService {

    public List findUserListByCode(String code);

    public List findUserByParam(String code, String name);

}

UserHandleServiceImpl.java

package com.shushan.service.impl;

import com.shushan.dao.UserHandleDao;
import com.shushan.entity.Student;
import com.shushan.service.UserHandleService;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.util.List;

@Service
public class UserHandleServiceImpl implements UserHandleService {

    @Resource
    private UserHandleDao userHandleDao;

    /**
     * 通过学号查询学生信息
     * @param code
     * @return
     */
    @Override
    public List findUserListByCode(String code) {
        return userHandleDao.findUserListByCode(code);
    }

   /**
     * 通过参数查询学生信息
     * @param code
     * @param name
     * @return
     */
    @Override
    public List findUserByParam(String code, String name) {
        Student student = new Student();
        student.setName("hahh");
        student.setCode(code);
        return userHandleDao.findUserAll(student);
    }

}

UserHandleDao.java

package com.shushan.dao;

import com.shushan.entity.Student;
import org.apache.ibatis.annotations.Mapper;

import java.util.List;

@Mapper
public interface UserHandleDao {

    public List findUserListByCode(String code);

    public List findUserAll(Student student);
}