SpringMVC单元测试之MockMVC，模拟登入用户

╰+哭是因爲堅強的太久メ 2021-09-12 08:16 466阅读 0赞

[http://blog.csdn.net/lee272616/article/details/52760447][http_blog.csdn.net_lee272616_article_details_52760447]

今天介绍一下springMVC的单元[测试][Link 1]，可以参考[spring][][官方文档][Link 2]进行

前提准备，springmvc的demo工程，这里就不做叙述了

junit 需要4.12以上

## pom.xml ##

**\[html\]**  [view plain][]   [copy][view plain]

1.   **<****dependency****>**  
2.  **<****groupId****>**org.springframework**</****groupId****>**  
3.  **<****artifactId****>**spring-core**</****artifactId****>**  
4.   **</****dependency****>**  
5.   **<****dependency****>**  
6.     **<****groupId****>**org.springframework**</****groupId****>**  
7.     **<****artifactId****>**spring-beans**</****artifactId****>**  
8.   **</****dependency****>**  
9.   **<****dependency****>**  
10. **<****groupId****>**org.springframework**</****groupId****>**  
11. **<****artifactId****>**spring-context**</****artifactId****>**  
12.  **</****dependency****>**  
13.  **<****dependency****>**  
14.    **<****groupId****>**org.springframework**</****groupId****>**  
15.    **<****artifactId****>**spring-context-support**</****artifactId****>**  
16.  **</****dependency****>**  
17.  **<****dependency****>**  
18.    **<****groupId****>**org.springframework**</****groupId****>**  
19.    **<****artifactId****>**spring-web**</****artifactId****>**  
20.  **</****dependency****>**  
21.  **<****dependency****>**  
22.    **<****groupId****>**org.springframework**</****groupId****>**  
23.    **<****artifactId****>**spring-webmvc**</****artifactId****>**  
24.  **</****dependency****>**  
25.  **<****dependency****>**  
26.    **<****groupId****>**org.springframework**</****groupId****>**  
27.    **<****artifactId****>**spring-orm**</****artifactId****>**  
28.  **</****dependency****>**  
29.  **<****dependency****>**  
30.    **<****groupId****>**org.springframework**</****groupId****>**  
31.    **<****artifactId****>**spring-tx**</****artifactId****>**  
32.  **</****dependency****>**  
33.  **<****dependency****>**  
34.    **<****groupId****>**org.springframework**</****groupId****>**  
35.    **<****artifactId****>**spring-test**</****artifactId****>**  
36.  **</****dependency****>**  
37.  **<****dependency****>**  
38.    **<****groupId****>**junit**</****groupId****>**  
39.    **<****artifactId****>**junit**</****artifactId****>**  
40.  **</****dependency****>**

## controller层 ##

**\[java\]**  [view plain][]   [copy][view plain]

1.  **package** controller;  
2.    
3.  **import** javax.servlet.http.HttpSession;  
4.    
5.  **import** org.springframework.beans.factory.annotation.Autowired;  
6.  **import** org.springframework.web.bind.annotation.PathVariable;  
7.  **import** org.springframework.web.bind.annotation.RequestMapping;  
8.  **import** org.springframework.web.bind.annotation.RequestMethod;  
9.  **import** org.springframework.web.bind.annotation.RequestParam;  
10. **import** org.springframework.web.bind.annotation.RestController;  
11.   
12. **import** service.UserService;  
13. **import** domain.User;  
14.   
15. /\*\* 
16.  \* UserController. 
17.  \* @author Leon Lee 
18.  \*/  
19. @RestController  
20. @RequestMapping(value = "user")  
21. **public** **class** UserController \{  
22.   
23.     /\*\* 
24.      \* UserService interface. 
25.      \*/  
26.     @Autowired  
27.     **private** UserService userService;  
28.   
29.     /\*\* 
30.      \* Get user MSG. 
31.      \* @param userId 
32.      \* @return user Msg 
33.      \*/  
34.     @RequestMapping(value = "userMsg/\{userId\}", method = RequestMethod.GET)  
35.     **public** User getUserMsg(@PathVariable(value = "userId") String userId) \{  
36.         **return** userService.getUserMsg(userId);  
37.     \}  
38.       
39.     /\*\* 
40.      \* Update user MSG. 
41.      \* @param userId 
42.      \* @param userName 
43.      \* @return updated user MSG 
44.      \*/  
45.     @RequestMapping(value = "userMsg/\{userId\}", method = RequestMethod.PUT)  
46.     **public** User putUserMsg(@PathVariable(value = "userId") String userId, @RequestParam String userName,HttpSession session)\{  
47.         **if**(**null** == (String)session.getAttribute("loginUser"))  
48.             **return** **new** User();  
49.         System.out.println((String)session.getAttribute("loginUser"));  
50.         **return** userService.putUserMsg(userId, userName);  
51.     \}  
52.       
53.     /\*\* 
54.      \* Delete user. 
55.      \* @param userId 
56.      \* @return deleted user MSG 
57.      \*/  
58.     @RequestMapping(value = "userMsg/\{userId\}", method = RequestMethod.DELETE)  
59.     **public** User delUserMsg(@PathVariable(value = "userId") String userId)\{  
60.         **return** userService.delUserMsg(userId);  
61.     \}  
62.       
63.     /\*\* 
64.      \* Add user. 
65.      \* @param userName 
66.      \* @return added user MSG 
67.      \*/  
68.     @RequestMapping(value = "userMsg", method = RequestMethod.POST)  
69.     **public** User postUserMsg(@RequestParam String userName)\{  
70.         **return** userService.postUserMsg(userName);  
71.     \}  
72.       
73.     /\*\* 
74.      \* login User. Note that do not send password as url. 
75.      \* @param userId 
76.      \* @param password 
77.      \* @return 
78.      \*/  
79.     @RequestMapping(value = "userMsg/\{userId\}/\{password\}", method = RequestMethod.GET)  
80.     **public** **boolean** loginUser(@PathVariable String userId, @PathVariable String password, HttpSession session)\{  
81.         **if**("loginUser".equals(userId)&&"loginUser".equals(password))\{  
82.             session.setAttribute("loginUser", userId);  
83.             **return** **true**;  
84.         \}  
85.         **return** **false**;  
86.     \}  
87. \}

## 单元测试类 ##

这里的静态导入比较重要，有时候没办法自动导入的

就是下面的 import static xxx.\*

另一点，

**\[java\]**  [view plain][]   [copy][view plain]

1.  @ContextConfiguration(locations = \{ "classpath:applicationContext.xml","classpath:applicationContext.mvc.xml"\})

代表的是加载的配置文件，可以根据需要进行添加

**\[java\]**  [view plain][]   [copy][view plain]

1.  **package** controller.test;  
2.    
3.  **import** **static** org.springframework.test.web.servlet.setup.MockMvcBuilders.\*;  
4.  **import** **static** org.springframework.test.web.servlet.request.MockMvcRequestBuilders.\*;  
5.  **import** **static** org.springframework.test.web.servlet.result.MockMvcResultHandlers.\*;  
6.  **import** **static** org.springframework.test.web.servlet.result.MockMvcResultMatchers.\*;  
7.    
8.  **import** javax.servlet.http.HttpSession;  
9.    
10. **import** org.junit.Before;  
11. **import** org.junit.Test;  
12. **import** org.junit.runner.RunWith;  
13. **import** org.springframework.beans.factory.annotation.Autowired;  
14. **import** org.springframework.http.MediaType;  
15. **import** org.springframework.mock.web.MockHttpSession;  
16. **import** org.springframework.test.annotation.Rollback;  
17. **import** org.springframework.test.context.ContextConfiguration;  
18. **import** org.springframework.test.context.junit4.SpringJUnit4ClassRunner;  
19. **import** org.springframework.test.context.transaction.TransactionConfiguration;  
20. **import** org.springframework.test.context.web.WebAppConfiguration;  
21. **import** org.springframework.test.web.servlet.MockMvc;  
22. **import** org.springframework.test.web.servlet.MvcResult;  
23. **import** org.springframework.transaction.annotation.Transactional;  
24. **import** org.springframework.web.context.WebApplicationContext;  
25.   
26. /\*\* 
27.  \* spring mvc Test. 
28.  \* @author Leon Lee 
29.  \* @since spring-4.1.7 
30.  \*/  
31. // spring 4.3 change to SpringRunner.class  
32. @RunWith(SpringJUnit4ClassRunner.**class**)  
33. @WebAppConfiguration  
34. @ContextConfiguration(locations = \{ "classpath:applicationContext.xml","classpath:applicationContext.mvc.xml"\})  
35. // do rollback  
36. @TransactionConfiguration(defaultRollback = **true**)  
37. @Transactional  
38. **public** **class** TestTemplate \{  
39.     @Autowired  
40.     **private** WebApplicationContext wac;  
41.       
42.     **private** MockMvc mockMvc;  
43.     **private** MockHttpSession session;  
44.   
45.     @Before  
46.     **public** **void** setup() \{  
47.         // init applicationContext  
48.         **this**.mockMvc = webAppContextSetup(**this**.wac).build();  
49.         **this**.session = **new** MockHttpSession();  
50.     \}  
51.   
52.     @Test  
53.     **public** **void** getUserMsg() **throws** Exception \{  
54.         // get using get  
55.         **this**.mockMvc  
56.                 .perform((get("/user/userMsg/003"))  
57.                         .accept(MediaType.parseMediaType("application/json;charset=UTF-8")))  
58.                 .andExpect(status().isOk())  
59.                 .andExpect(content().contentType("application/json;charset=UTF-8"))  
60.                 .andDo(print()); // print  
61.     \}  
62.   
63.     @Test  
64.     // don't rollback  
65.     @Rollback(**false**)  
66.     **public** **void** putUserMsg() **throws** Exception \{  
67.         // update using put  
68.         **this**.mockMvc  
69.                 .perform((put("/user/userMsg/003"))  
70.                         .contentType(MediaType.APPLICATION\_FORM\_URLENCODED)  
71.                         .param("userName","新名字03号")  
72.                         .session((MockHttpSession)getLoginSession())  
73.                         .accept(MediaType.parseMediaType("application/json;charset=UTF-8"))  
74.                         )  
75.                 .andExpect(status().isOk())  
76.                 .andExpect(content().contentType("application/json;charset=UTF-8"))  
77.                 .andDo(print()); // print  
78.     \}  
79.   
80.     @Test  
81.     **public** **void** delUser() **throws** Exception \{  
82.         // delete using delete  
83.         **this**.mockMvc  
84.                 .perform((delete("/user/userMsg/004"))  
85.                         .accept(MediaType.parseMediaType("application/json;charset=UTF-8"))  
86.                         )  
87.                 .andExpect(status().isOk())  
88.                 .andExpect(content().contentType("application/json;charset=UTF-8"))  
89.                 .andDo(print()); //print  
90.     \}  
91.       
92.     @Test  
93.     // don't rollback  
94.     @Rollback(**false**)  
95.     **public** **void** postUser() **throws** Exception\{  
96.         // add using post  
97.         **this**.mockMvc  
98.                 .perform((post("/user/userMsg"))  
99.                         .param("userName", "最新的用户")  
100.                         .accept(MediaType.parseMediaType("application/json;charset=UTF-8"))  
101.                         )  
102.                 .andExpect(status().isOk())  
103.                 .andExpect(content().contentType("application/json;charset=UTF-8"))  
104.                 .andDo(print()); //print  
105.     \}  
106.       
107.     /\*\* 
108.      \* 获取登入信息session 
109.      \* @return 
110.      \* @throws Exception 
111.      \*/  
112.     **private** HttpSession getLoginSession() **throws** Exception\{  
113.         // mock request get login session  
114.         // url = /xxx/xxx/\{username\}/\{password\}  
115.         MvcResult result = **this**.mockMvc  
116.                                 .perform((get("/user/userMsg/loginUser/loginUser")))  
117.                                 .andExpect(status().isOk())  
118.                                 .andReturn();  
119.         **return** result.getRequest().getSession();  
120.     \}  
121. \}  
122.

[http_blog.csdn.net_lee272616_article_details_52760447]: http://blog.csdn.net/lee272616/article/details/52760447
[Link 1]: http://lib.csdn.net/base/softwaretest
[spring]: http://lib.csdn.net/base/javaee
[Link 2]: http://docs.spring.io/spring/docs/current/spring-framework-reference/htmlsingle/#spring-mvc-test-framework
[view plain]: http://blog.csdn.net/lee272616/article/details/52760447#