spring mvc的简单单元测试,说白了就是测试spring mvc的controller.

先参考两篇帖子:

1.http://spring.io/blog/2011/06/21/spring-3-1-m2-testing-with-configuration-classes-and-profiles

2.http://spring.io/blog/2012/11/07/spring-framework-3-2-rc1-new-testing-features

实践:

UserController.java

package com.test.web;import java.io.File;import java.io.IOException;import java.util.List;import javax.annotation.Resource;import javax.servlet.http.HttpServletRequest;import org.springframework.stereotype.Controller;import org.springframework.util.FileCopyUtils;import org.springframework.web.bind.annotation.RequestBody;import org.springframework.web.bind.annotation.RequestMapping;import org.springframework.web.bind.annotation.RequestMethod;import org.springframework.web.bind.annotation.RequestParam;import org.springframework.web.bind.annotation.ResponseBody;import org.springframework.web.multipart.MultipartFile;import com.test.domain.User;import com.test.service.UserService;@Controller@RequestMapping("/user")public class UserController{	private static final String FILES_DIR="e:/";	@Resource	private UserService userService;	@RequestMapping("/save")	public String save(User user) {		userService.save(user);		return "result";	}	@RequestMapping("/getUser")	public @ResponseBody User getUser(HttpServletRequest request,@RequestBody User user) {		System.out.println("AAAAAAA::"+user);		request.setAttribute("data1",user);		return user;	}	@RequestMapping("/getUsers")	public @ResponseBody List
    
      getUsers(HttpServletRequest request,@RequestBody List
     
       users) {		System.out.println(users.getClass().getName());		for (User user : users) {			System.out.println("BBBBBBBB::"+user);		}		request.setAttribute("data2", users);		return users;	}	@RequestMapping(value = "/upload", method = RequestMethod.POST)	public String add(HttpServletRequest request,@RequestParam("file") MultipartFile file) {		//System.out.println("real path::"+request.getServletContext().getRealPath("/"));		String filename=file.getOriginalFilename();		File tmpFile=new File(FILES_DIR+filename);		if(filename!=null&&!file.isEmpty()){			try {				FileCopyUtils.copy(file.getBytes(), tmpFile);				System.out.println("上传成功");			} catch (IOException e) {				e.printStackTrace();			}		}		return "result";	}	}
     
    

package com.test.web;import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.post;import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.fileUpload;import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.view;import java.io.IOException;import java.util.ArrayList;import java.util.List;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.MockMultipartFile;import org.springframework.test.context.ContextConfiguration;import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;import org.springframework.test.context.web.WebAppConfiguration;import org.springframework.test.web.servlet.MockMvc;import org.springframework.test.web.servlet.setup.MockMvcBuilders;import org.springframework.web.context.WebApplicationContext;import com.test.config.AppConfig;import com.test.domain.User;import com.test.util.JsonUtil;@RunWith(SpringJUnit4ClassRunner.class)@WebAppConfiguration@ContextConfiguration(classes={ AppConfig.class})public class UserControllerTest {	@Autowired	private WebApplicationContext webApplicationContext;	private MockMvc mockMvc;	@Before	public void init(){		mockMvc = MockMvcBuilders.webAppContextSetup(webApplicationContext).build();	}	@Test	public void printBeans(){		String[] beans=webApplicationContext.getBeanDefinitionNames();		for (String bean : beans) {			System.out.println(bean);		}	}	/*测试以普通请求参数发送的请求*/	@Test	public void testSave() throws Exception {		mockMvc.perform(post("/user/save").param("id", "123").param("username", "you")).andExpect(status().isOk()).andExpect(view().name("result"));	}	/*测试将数据以JSON格式写入请求体发送的请求*/	@Test	public void testGet() throws Exception {		mockMvc.perform(post("/user/getUser").contentType(JsonUtil.APPLICATION_JSON_UTF8).content(JsonUtil.convertObjectToJsonBytes(new User(22,"werwr")))).andExpect(status().isOk());	}	/*测试将数据以JSON格式写入请求体发送的请求*/	@Test	public void testGetAll() throws IOException, Exception {		List
    
      list=new ArrayList
     
      ();		list.add(new User(23,"你爱我"));		list.add(new User(25,"我不爱你"));		mockMvc.perform(post("/user/getUsers").contentType(JsonUtil.APPLICATION_JSON_UTF8).content(JsonUtil.convertObjectToJsonBytes(list))).andExpect(status().isOk());	}	/*测试文件上传发送的请求*/	@Test	public void testUpload() throws Exception{		MockMultipartFile file = new MockMultipartFile("file", "orig.txt", null, "bar".getBytes());		mockMvc.perform(fileUpload("/user/upload").file(file)).andExpect(status().isOk());	}}
     
    

完整源代码:

顺便贴一下spring junit(非controller简单bean)

package com.test.service;import javax.annotation.Resource;import org.junit.Test;import org.junit.runner.RunWith;import org.springframework.context.annotation.ComponentScan;import org.springframework.context.annotation.Configuration;import org.springframework.test.context.ContextConfiguration;import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;import com.test.domain.User;@RunWith(SpringJUnit4ClassRunner.class)@ContextConfigurationpublic class UserServiceTest {	@Configuration	@ComponentScan(basePackages={"com.test.service"})	static class ContextConfiguration {}	@Resource	private UserService userService;		@Test	public void testSave(){		userService.save(new User(123,"xxx"));	}}