利用junit对springMVC的Controller进行测试

本文转自http://www.tuicool.com/articles/7rMziy

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

代码1：直接Controller调用方法

import static org.junit.Assert.*;

import java.sql.SQLException;

import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.mock.web.MockHttpServletRequest;
import org.springframework.mock.web.MockHttpServletResponse;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;

import com.zz.login.web.LoginController;

/**
 * @author zhzh
 * 2015-4-7
 */
@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration({"classpath*:/beans.xml","classpath*:/spring-mvc.xml"})
public class TestController {

	// 模拟request,response
    private MockHttpServletRequest request;
    private MockHttpServletResponse response;   

    // 注入loginController
    @Autowired
    private LoginController loginController ;  

    // 执行测试方法之前初始化模拟request,response
    @Before
    public void setUp(){
        request = new MockHttpServletRequest();
        request.setCharacterEncoding("UTF-8");
        response = new MockHttpServletResponse();
    }
    /**
     *
     * @Title：testLogin
     * @Description: 测试用户登录
     */
    @Test
    public void testLogin() {
        try {
        	request.setParameter("userName", "admin");
        	request.setParameter("password", "2");
            assertEquals("login",loginController.loginIn(request,response)) ;
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}

代码2：调用请求路径

import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.post;
import static org.springframework.test.web.servlet.result.MockMvcResultHandlers.print;
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
import static org.springframework.test.web.servlet.setup.MockMvcBuilders.webAppContextSetup;

import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
import org.springframework.test.context.transaction.TransactionConfiguration;
import org.springframework.test.context.web.WebAppConfiguration;
import org.springframework.test.web.servlet.MockMvc;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.web.context.WebApplicationContext;

/**
 * @author zhzh
 * 2015-4-7
 */
@RunWith(SpringJUnit4ClassRunner.class)
@WebAppConfiguration
@ContextConfiguration({"classpath*:/beans.xml","classpath*:/spring-mvc.xml"})
//当然 你可以声明一个事务管理 每个单元测试都进行事务回滚 无论成功与否
@TransactionConfiguration(defaultRollback = true)
@Transactional
public class TestController2 {
	@Autowired
    private WebApplicationContext wac;  

    private MockMvc mockMvc; 

    @Before
    public void setup() {
        this.mockMvc = webAppContextSetup(this.wac).build();
    } 

    @Test
    public void testLogin() throws Exception {
        mockMvc.perform((post("/loginTest").param("userName", "admin").param("password", "1"))).andExpect(status().isOk())
                .andDo(print());
    } 

    /*@Test
    //有些单元测试你不希望回滚
    @Rollback(false)
	public void testInsert() throws Exception {
	    mockMvc.perform((post("/insertTest"))).andExpect(status().isOk())
	            .andDo(print());
	} */
}

注意：import static 的spring类

代码3：

LoginController代码片段

/**
	 * 登录入口
	 * @param request
	 * @param response
	 * @return
	 */
	@RequestMapping(value = "/loginTest", method = RequestMethod.POST)
	public String loginTest(HttpServletRequest request,HttpServletResponse response){
		String account = request.getParameter("userName");
		String password = request.getParameter("password");
		if (account.equals("admin")&&password.equals("1")){
			return "index";
		}else{
			return "login";
		}
	}

参考： http://blog.csdn.net/a95473004/article/details/8926929

http://ysj5125094.iteye.com/blog/2082110