Spring Boot入门(7)-单元测试

我们在开发完 项目时，上线之前一般都会进行测试。下面介绍 SpringBoot 给 Service 和Controller 测试。

一、对 Service 进行测试

1、UserService.java

1. package com.liuyanzhao.blog.service;
3. import com.liuyanzhao.blog.domain.User;
4. import com.liuyanzhao.blog.repository.UserRepository;
5. import org.springframework.beans.factory.annotation.Autowired;
6. import org.springframework.stereotype.Service;
8. /**
9.  * @Author 言曌
10.  * @DATE 2017/10/29 下午9:02
11.  * @URL http://liuyanzhao.com
12.  */
14. @Service
15. public class UserService {
17.     @Autowired
18.     private UserRepository userRepository;
20.     /**
21.      * 根据id查询一个用户的信息
22.      * @param id
23.      * @return
24.      */
25.     public User findOne(Integer id) {
26.         return userRepository.findOne(id);
27.     }
28. }

 

2、UserServiceTest.java

1. package com.liuyanzhao.blog.service;
3. import com.liuyanzhao.blog.domain.User;
4. import org.junit.Assert;
5. import org.junit.Test;
6. import org.junit.runner.RunWith;
7. import org.springframework.beans.factory.annotation.Autowired;
8. import org.springframework.boot.test.context.SpringBootTest;
9. import org.springframework.test.context.junit4.SpringRunner;
12. /**
13.  * @Author 言曌
14.  * @DATE 2017/11/1 下午4:47
15.  * @URL http://liuyanzhao.com
16.  */
18. @RunWith(SpringRunner.class)
19. @SpringBootTest
20. public class UserServiceTest {
22.     @Autowired
23.     private UserService userService;
25.     @Test
26.     public void findOne() throws Exception {
27.         //获得id为14的用户信息
28.         User user = userService.findOne(14);
29.         //断言两者相等
30.         Assert.assertEquals(new Integer(7),user.getAge());
31.     }
33. }

注意第18-19行的两个注解，必须加。

获得id=14的用户信息，如果该用户的age=7,则测试通过(绿)，否则测试不通过(红)

 

 

二、对 Controller 进行测试

1、UserController.java

1. package com.liuyanzhao.blog.controller;
3. import com.liuyanzhao.blog.domain.User;
4. import com.liuyanzhao.blog.repository.UserRepository;
5. import com.liuyanzhao.blog.service.UserService;
6. import org.springframework.beans.factory.annotation.Autowired;
7. import org.springframework.web.bind.annotation.*;
9. import java.util.List;
11. /**
12.  * @Author 言曌
13.  * @DATE 2017/10/29 上午10:33
14.  * @URL http://liuyanzhao.com
15.  */
17. @RestController
18. public class UserController {
20.     @Autowired
21.     private UserRepository userRepository;
24.     /**
25.      * 查询所有用户列表
26.      * @return
27.      */
28.     @GetMapping(value = "/users")
29.     public List<User> listUser() {
30.         return userRepository.findAll();
31.     }
34. }

 

2、UserControllerTest.java

1. package com.liuyanzhao.blog.controller;
3. import org.junit.Test;
4. import org.junit.runner.RunWith;
5. import org.springframework.beans.factory.annotation.Autowired;
6. import org.springframework.boot.test.autoconfigure.web.servlet.AutoConfigureMockMvc;
7. import org.springframework.boot.test.context.SpringBootTest;
8. import org.springframework.test.context.junit4.SpringRunner;
9. import org.springframework.test.web.servlet.MockMvc;
10. import org.springframework.test.web.servlet.request.MockMvcRequestBuilders;
11. import org.springframework.test.web.servlet.result.MockMvcResultMatchers;
13. /**
14.  * @Author 言曌
15.  * @DATE 2017/11/1 下午5:00
16.  * @URL http://liuyanzhao.com
17.  */
19. @RunWith(SpringRunner.class)
20. @SpringBootTest
21. @AutoConfigureMockMvc
22. public class UserControllerTest {
24.     @Autowired
25.     private MockMvc mvc;
27.     @Test
28.     public void listUser() throws Exception {
29.         mvc.perform(MockMvcRequestBuilders.get("/users"))
30.                 .andExpect(MockMvcResultMatchers.status().isOk());//期望通过
31.                 //.andExpect(MockMvcResultMatchers.content().string("abc"));//期望结果是一个值
32.     }
34. }

注意第19-21行的三个注解，必须加。

如果测试通过绿，否则红。