SpringBoot 单元测试

单元测试（模块测试）是开发者编写的一小段代码，用于检验被测代码的一个很小的、很明确的功能是否正确。

单元测试是由程序员自己来完成，最终受益的也是程序员自己。可以这么说，程序员有责任编写功能代码，同时也就有责任为自己的代码编写单元测试。

执行单元测试，就是为了证明这段代码的行为和我们期望的一致。

对于程序员来说，如果养成了对自己写的代码进行单元测试的习惯，不但可以写出高质量的代码，而且还能提高编程水平。

对Controller层进行单元测试

@RunWith(SpringRunner.class)
@WebAppConfiguration
@SpringBootTest(classes = AuthServerApplication.class)
public class BaseControllerTest {
    protected Logger logger = LoggerFactory.getLogger(BaseControllerTest.class);

    @Before
    public void init() {
        logger.info("开始测试...");
    }

    @After
    public void after() {
        logger.info("测试结束...");
    }
}

public class LoginControllerTest extends BaseControllerTest {
    private MockMvc mockMvc;
    @Autowired
    private WebApplicationContext wac;
    @Before
    public void setup(){
        mockMvc = MockMvcBuilders.webAppContextSetup(wac).build();
    }


    @Test
    public void login() throws Exception {
        MvcResult mvcResult = mockMvc.perform(MockMvcRequestBuilders.get("/login/getTemporaryToken")
                .accept(MediaType.APPLICATION_JSON)
                .contentType(MediaType.APPLICATION_JSON))
                .andDo(MockMvcResultHandlers.print())
                .andReturn();
        int status=mvcResult.getResponse().getStatus();
        Assert.assertEquals(ResultCode.OK, status);
        if(ResultCode.OK != status){
            logger.error("/login/getTemporaryToken:" + mvcResult.getResponse().getContentAsString());
            return;
        }
        ResultData rd = JSON.parseObject(mvcResult.getResponse().getContentAsString(),ResultData.class);
        Assert.assertEquals(ResultCode.OK, (int)rd.getCode());
        TemporaryTokenDTO tokenDTO = JSON.parseObject(rd.getData().toString(),TemporaryTokenDTO.class);
        LoginDTO loginDTO = new LoginDTO();
        loginDTO.setAccount("hujinshui");
        loginDTO.setPassword("123456");
        MvcResult loginResult = mockMvc.perform(MockMvcRequestBuilders.post("/login/in")
                .accept(MediaType.APPLICATION_JSON)
                .contentType(MediaType.APPLICATION_JSON)
                .content(JSONObject.toJSONString(loginDTO))
                .header("temporaryToken",tokenDTO.getTemporaryToken()))
                .andDo(MockMvcResultHandlers.print())
                .andReturn();
        int loginStatus=loginResult.getResponse().getStatus();
        if(ResultCode.OK != loginStatus){
            logger.error("/login/in:" + loginResult.getResponse().getContentAsString());
        }
        Assert.assertEquals(ResultCode.OK, loginStatus);
    }
}

对service层进行测试

@RunWith(SpringRunner.class)
@SpringBootTest(classes = AuthServerApplication.class)
public class BaseServiceTest {
    protected Logger logger = LoggerFactory.getLogger(BaseServiceTest.class);

    @Before
    public void init() {
        logger.info("service测试开始...");
    }

    @After
    public void after() {
        logger.info("service测试结束...");
    }
}

public class LoginServiceTest extends BaseServiceTest {
    @Autowired
    private LoginService loginService;
    private MockHttpServletRequest request;
    private MockHttpServletResponse response;

    @Before
    public void setUp(){
        request = new MockHttpServletRequest();
        request.setCharacterEncoding("UTF-8");
        response = new MockHttpServletResponse();
    }

    @Test
    public void getTemporaryToken(){
        ResultData rd = loginService.getTemporaryToken();
        Assert.assertEquals(ResultCode.OK, (int)rd.getCode());
        logger.info("LoginService getTemporaryToken："+JSON.toJSONString(rd));
    }

    @Test
    public void login(){
        ResultData rd = loginService.login("hujinshui","123456", request);
        Assert.assertEquals(ResultCode.OK, (int)rd.getCode());
        logger.info("LoginService login："+JSON.toJSONString(rd));
    }
    
    @Test
    public void test(){
        getTemporaryToken();
        login();
    }

}

对dao层进行测试

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

    protected Logger logger = LoggerFactory.getLogger(BaseDaoTest.class);

    @Before
    public void init() {
        logger.info("dao测试开始...");
    }

    @After
    public void after() {
        logger.info("dao测试结束...");
    }
}

public class RoleDaoTest extends BaseDaoTest{
    @Autowired
    private RoleDao roleDao;

    @Test
    public void test() {
        Role role = roleDao.selectByIfDefault(1);
        Assert.assertEquals((int)role.getIfDefault(),1);
        logger.info("selectByIfDefault :" + JSON.toJSONString(role));

        Role role2 = roleDao.selectByPrimaryKey(role.getId());
        Assert.assertEquals(role.getId(),role2.getId());
        logger.info("selectByPrimaryKey :" + JSON.toJSONString(role2));
    }
}