利用junit对springMVC的Controller进行测试

深藏阁楼爱情的钟 2021-06-11 15:11 463阅读 0赞

[http://zhangzhaoaaa.iteye.com/blog/2200096?utm\_source=tuicool][http_zhangzhaoaaa.iteye.com_blog_2200096_utm_source_tuicool]

平时对junit测试service/DAO层已经很熟悉不过了，如果不了解，可以[猛戳这里][Link 1]，但是我们要测试controller层，不能总重启服务器吧，（重启tomcat好慢的飘过，别。。。别走啊），那么我们就用junit4模拟请求，测试controller层的方法。

代码1：直接Controller调用方法

Java代码   ![收藏代码][icon_star.png]

1.  **import** **static** org.junit.Assert.\*;  
2.    
3.  **import** java.sql.SQLException;  
4.    
5.  **import** org.junit.Before;  
6.  **import** org.junit.Test;  
7.  **import** org.junit.runner.RunWith;  
8.  **import** org.springframework.beans.factory.annotation.Autowired;  
9.  **import** org.springframework.mock.web.MockHttpServletRequest;  
10. **import** org.springframework.mock.web.MockHttpServletResponse;  
11. **import** org.springframework.test.context.ContextConfiguration;  
12. **import** org.springframework.test.context.junit4.SpringJUnit4ClassRunner;  
13.   
14. **import** com.zz.login.web.LoginController;  
15.   
16. /\*\* 
17.  \* @author zhzh 
18.  \* 2015-4-7 
19.  \*/  
20. @RunWith(SpringJUnit4ClassRunner.**class**)    
21. @ContextConfiguration(\{ "classpath\*:/beans.xml","classpath\*:/spring-mvc.xml"\})    
22. **public** **class** TestController \{  
23.   
24.     // 模拟request,response    
25.     **private** MockHttpServletRequest request;    
26.     **private** MockHttpServletResponse response;     
27.         
28.     // 注入loginController    
29.     @Autowired    
30.     **private** LoginController loginController ;    
31.         
32.     // 执行测试方法之前初始化模拟request,response    
33.     @Before      
34.     **public** **void** setUp()\{      
35.         request = **new** MockHttpServletRequest();        
36.         request.setCharacterEncoding("UTF-8");        
37.         response = **new** MockHttpServletResponse();        
38.     \}           
39.     /\*\*  
40.      \*   
41.      \* @Title：testLogin  
42.      \* @Description: 测试用户登录    
43.      \*/    
44.     @Test    
45.     **public** **void** testLogin() \{     
46.         **try** \{    
47.             request.setParameter("userName", "admin");  
48.             request.setParameter("password", "2");  
49.             assertEquals("login",loginController.loginIn(request,response)) ;    
50.         \} **catch** (Exception e) \{    
51.             e.printStackTrace();    
52.         \}    
53.     \}    
54. \}

代码2：调用请求路径

Java代码   ![收藏代码][icon_star.png]

1.  **import** **static** org.springframework.test.web.servlet.request.MockMvcRequestBuilders.post;  
2.  **import** **static** org.springframework.test.web.servlet.result.MockMvcResultHandlers.print;  
3.  **import** **static** org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;  
4.  **import** **static** org.springframework.test.web.servlet.setup.MockMvcBuilders.webAppContextSetup;  
5.    
6.  **import** org.junit.Before;  
7.  **import** org.junit.Test;  
8.  **import** org.junit.runner.RunWith;  
9.  **import** org.springframework.beans.factory.annotation.Autowired;  
10. **import** org.springframework.test.context.ContextConfiguration;  
11. **import** org.springframework.test.context.junit4.SpringJUnit4ClassRunner;  
12. **import** org.springframework.test.context.transaction.TransactionConfiguration;  
13. **import** org.springframework.test.context.web.WebAppConfiguration;  
14. **import** org.springframework.test.web.servlet.MockMvc;  
15. **import** org.springframework.transaction.annotation.Transactional;  
16. **import** org.springframework.web.context.WebApplicationContext;  
17.   
18. /\*\* 
19.  \* @author zhzh 
20.  \* 2015-4-7 
21.  \*/  
22. @RunWith(SpringJUnit4ClassRunner.**class**)    
23. @WebAppConfiguration    
24. @ContextConfiguration(\{ "classpath\*:/beans.xml","classpath\*:/spring-mvc.xml"\})   
25. //当然 你可以声明一个事务管理 每个单元测试都进行事务回滚 无论成功与否    
26. @TransactionConfiguration(defaultRollback = **true**)    
27. @Transactional   
28. **public** **class** TestController2 \{  
29.     @Autowired    
30.     **private** WebApplicationContext wac;    
31.     
32.     **private** MockMvc mockMvc;   
33.       
34.     @Before    
35.     **public** **void** setup() \{     
36.         **this**.mockMvc = webAppContextSetup(**this**.wac).build();    
37.     \}   
38.       
39.     @Test    
40.     **public** **void** testLogin() **throws** Exception \{    
41.         mockMvc.perform((post("/loginTest").param("userName", "admin").param("password", "1"))).andExpect(status().isOk())    
42.                 .andDo(print());    
43.     \}   
44.       
45.     /\*@Test   
46.     //有些单元测试你不希望回滚   
47.     @Rollback(false)   
48.     public void testInsert() throws Exception \{   
49.         mockMvc.perform((post("/insertTest"))).andExpect(status().isOk())   
50.                 .andDo(print());   
51.     \} \*/  
52. \}

**注意：import static 的spring类**

**     **代码3：

LoginController代码片段

Java代码   ![收藏代码][icon_star.png]

1.  /\*\* 
2.       \* 登录入口 
3.       \* @param request 
4.       \* @param response 
5.       \* @return 
6.       \*/  
7.      @RequestMapping(value = "/loginTest", method = RequestMethod.POST)  
8.      **public** String loginTest(HttpServletRequest request,HttpServletResponse response)\{  
9.          String account = request.getParameter("userName");  
10.         String password = request.getParameter("password");  
11.         **if** (account.equals("admin")&&password.equals("1"))\{  
12.             **return** "index";  
13.         \}**else**\{  
14.             **return** "login";  
15.         \}  
16.     \}

[http_zhangzhaoaaa.iteye.com_blog_2200096_utm_source_tuicool]: http://zhangzhaoaaa.iteye.com/blog/2200096?utm_source=tuicool
[Link 1]: http://zhangzhaoaaa.iteye.com/admin/blogs/2033355
[icon_star.png]: /images/20210527/dc9267e31ee2471bb5c0c1c4e907cfbf.png