c，編寫好控制類的代碼，我們可以測試這個控制類了，下面的代碼是使用TestNg測試controller十分正確
package test.com.smart.web;
import com.smart.domain.User;
import com.smart.web.LoginController;
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.testng.AbstractTestNGSpringContextTests;
import org.springframework.test.util.ReflectionTestUtils;
import org.springframework.web.client.RestTemplate;
import org.springframework.web.servlet.HandlerMapping;
import org.springframework.web.servlet.ModelAndView;
import org.springframework.web.servlet.mvc.annotation.AnnotationMethodHandlerAdapter;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;
import static org.testng.Assert.assertEquals;
import static org.testng.Assert.assertNotNull;
import static org.testng.Assert.assertNull;
@ContextConfiguration(locations = {"classpath:applicationContext.xml"，"file:d:/actionSpring/chapter/chapter1/src/main/webapp/WEB-INF/viewspace-servlet.xml"})
public class LoginControllerTest extends AbstractTestNGSpringContextTests {
@Autowired
private AnnotationMethodHandlerAdapter handlerAdapter;
@Autowired
private LoginController controller;
//聲明Request與Response模擬對象
private MockHttpServletRequest request;
private MockHttpServletResponse response;
//執(zhí)行測試前先初始模擬對象
@BeforeMethod
public void before() {
request = new MockHttpServletRequest();
request.setCharacterEncoding("UTF-8");
response = new MockHttpServletResponse();
request.setAttribute(HandlerMapping.INTROSPECT_TYPE_LEVEL_MAPPING， true); //Spring3.1 存在的BUG
}
// 測試LoginController#loginCheck()方法
@Test
public void loginCheck() throws Exception {
//測試登陸成功的情況
request.setRequestURI("/admin/loginCheck.html");
request.addParameter("userName"， "admin"); // 設置請求URL及參數(shù)
request.addParameter("password"， "123456");
//向控制發(fā)起請求 ” /loginCheck.html”
ModelAndView mav = handlerAdapter.handle(request， response， controller);
User user = (User) request.getSession().getAttribute("user");
assertNotNull(mav);
assertEquals(mav.getViewName()， "main");
assertNotNull(user);
request.getSession().removeAttribute("user");
//測試登陸失敗的情況
request.setRequestURI("/admin/loginCheck.html");
request.addParameter("userName"， "test");
request.addParameter("password"， "123456");
mav = handlerAdapter.handle(request， response， controller);
user = (User) request.getSession().getAttribute("user");
assertNotNull(mav);
assertEquals(mav.getViewName()， "login");
assertNull(user);
}
}
注意：
@ContextConfiguration(locations = {"classpath:applicationContext.xml"，"file:d:/actionSpring/chapter/chapter1/src/main/webapp/WEB-INF/viewspace-servlet.xml"}) 這個注解可以整合多個spring配置文件中"file:d:/actionSpring/chapter/chapter1/src/main/webapp/WEB-INF/viewspace-servlet.xml"表示的是文件系統(tǒng)的形式給出配置文件
