spring controller层测试

spring controller层测试
2011年07月01日
　　 这是我最近写的测试类对应的方法的测试代码其下面，请大家提一些建议，看哪里写的不周到，谢谢，        
　　PositionController postController;  Controller对象 
　　第一个方法： 
　　public String allPositions(HttpServletRequest request, ModelMap model) { 
　　TableFacade tableFacade = TableFacadeFactory.createSpringTableFacade("allPos itionsTable", request); 
　　tableFacade.setMaxRows(10); 
　　Limit limit = tableFacade.getLimit(); 
　　FilterSet filterSet = limit.getFilterSet(); 
　　int totalRows = projectService.getTotalRowCountForPositions(filter Set); 
　　tableFacade.setTotalRows(totalRows); 
　　SortSet sortSet = limit.getSortSet(); 
　　int rowStart = limit.getRowSelect() == null? 0 : limit.getRowSelect().getRowStart(); 
　　int rowEnd = limit.getRowSelect() == null? 10 : limit.getRowSelect().getRowEnd(); 
　　List positions = projectService.getPositions(filterSet, sortSet, rowStart, rowEnd); 
　　model.addAttribute("positions", positions); 
　　model.addAttribute("limit", limit); 
　　return "/core/positions_table"; 
　　}  
　　它对应的测试程序如下：      
　　@Test 
　　public void testAllPositions(){ 
　　String view = postController.allPositions(request, model); 
　　Limit limit = (Limit)model.get("limit"); 
　　List positions = (List)model.get("positions"); 
　　Assert.assertNotNull(model.get("positions")); 
　　Assert.assertEquals(10, positions.size()); 
　　Assert.assertNotNull(model.get("limit")); 
　　Assert.assertEquals(0, limit.getRowSelect().getRowStart()); 
　　Assert.assertEquals(10, limit.getRowSelect().getRowEnd()); 
　　Assert.assertTrue(view.contains("/core/positions_t able")); 
　　} 
　　第二个方法： 
　　public String getAllPosition(HttpServletRequest request, 
　　HttpServletResponse response, 
　　ModelMap model) { 
　　List positions = this.positionService.getAllPositions(); 
　　//define the number of max rows per page for jmesa when the result is empty 
　　Integer maxRows = 10; 
　　if (CollectionUtils.isNotEmpty(positions)) { 
　　maxRows = positions.size(); 
　　} 
　　model.addAttribute("count", maxRows); 
　　request.getSession().setAttribute("positions",posi tions); 
　　return "/crt/position_table"; 
　　} 
　　它对应的测试程序如下：  
　　@Test 
　　public void testGetAllPosition(){ 
　　String view = postController.getAllPosition(request, response, model); 
　　List positions = (List)request.getSession().getAttribute("p ositions"); 
　　int maxRow = (Integer)model.get("count"); 
　　Assert.assertNotNull(positions); 
　　Assert.assertEquals(maxRow, positions.size()); 
　　Assert.assertTrue(view.contains("/crt/position_tab le")); 
　　} 
　　第三个方法： 
　　public void getBindPositionRolesInAllRoles(Integer positionId, HttpServletRequest request, HttpServletResponse response, ModelMap model) { 
　　Map> groupRoles = new HashMap>(); 
　　try { 
　　groupRoles = positionService.getBindPositionRoleInAllRole(posit ionId); 
　　} catch (Exception e) { 
　　MessageUtils.outputJSONResult("occurError", response); 
　　return; 
　　} 
　　MessageUtils.outputJSONResult(JSONObject.fromObjec t(groupRoles).toString(), response); 
　　} 
　　它对应的测试程序如下：  
　　@Test 
　　public void testGetBindPositionRolesInAllRoles(){ 
　　postController.getBindPositionRolesInAllRoles(1, request, response, model); 
　　try { 
　　Assert.assertEquals("utf-8", response.getCharacterEncoding()); 
　　Assert.assertNotNull(response.getContentAsString() ); 
　　//just the content whether is json format 
　　boolean flag = false; 
　　try{ 
　　JSONObject.fromObject(JSONObject.fromObject(respon se.getContentAsString())); 
　　flag = true; 
　　}catch (Exception e){ 
　　flag = false; 
　　} 
　　Assert.assertEquals(true, flag); 
　　} catch (UnsupportedEncodingException e) { 
　　e.printStackTrace(); 
　　} 
　　} 
　　工具类MessageUtils的方法： 
　　public static void outputJSONResult(String result, HttpServletResponse response) { 
　　try { 
　　response.setHeader("ContentType", "text/json"); 
　　response.setCharacterEncoding("utf-8"); 
　　PrintWriter pw = response.getWriter(); 
　　pw.write(result); 
　　pw.flush(); 
　　pw.close(); 
　　} catch (IOException e) { 
　　e.printStackTrace(); 
　　} 
　　}