单元测试SpringMVC带文件上传的controller

@Test
public void postUpdateTeamPhotoEditWithStringParameter() throws Exception {
MockMultipartHttpServletRequest request = new MockMultipartHttpServletRequest() ;
String strTEAMPHOTOCLASSID = String.valueOf( TEAMPHOTOCLASSID ) ;
String strSPECIFICATIONID = String.valueOf( SPECIFICATIONID ) ;
final FileInputStream fis = new FileInputStream("c://ff3.jpg");
MockMultipartFile multipartFile = new MockMultipartFile("photopath","ff3.jpg","image/jpeg",fis);
request.addFile(multipartFile);
request.setMethod("POST");
request.setContentType("multipart/form-data");
request.addHeader("Content-type", "multipart/form-data");
request.setRequestURI("/teamphoto/edit/"+TEAMPHOTOID+"/");
request.addParameter( "leader", "lzj update" ) ;
request.addParameter( "STeamphotoclass", strTEAMPHOTOCLASSID ) ;
request.addParameter( "SSpecification", strSPECIFICATIONID ) ;
int countTeamphoto = teamPhotoDao.getTeamphotoCount() ;
int countTeamphotoclass = teamPhotoDao.getSTeamphotoclassCount() ;
int countSpecification = teamPhotoDao.getSSpecificationCount() ;
new AnnotationMethodHandlerAdapter()
.handle( request, new MockHttpServletResponse(), teamPhotoController ) ;
assertEquals(
teamPhotoDao.getSTeamphotoById( TEAMPHOTOID ).getLeader(), "lzj update"
) ;
assertEquals( teamPhotoDao.getTeamphotoCount(), countTeamphoto ) ;
assertEquals( teamPhotoDao.getSTeamphotoclassCount(), countTeamphotoclass ) ;
assertEquals( teamPhotoDao.getSSpecificationCount(), countSpecification ) ;
}

猜你喜欢

转载自liangzhijian.iteye.com/blog/1030399