SpringBoot Junit测试Controller
原文链接:https://blog.csdn.net/xiaolyuh123/article/details/73281522
import java.util.List; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.data.domain.Page; import org.springframework.data.domain.PageRequest; import org.springframework.data.domain.Sort; import org.springframework.data.domain.Sort.Direction; import org.springframework.web.bind.annotation.RequestBody; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.RequestParam; import org.springframework.web.bind.annotation.RestController; import com.xiaolyuh.entity.Person; import com.xiaolyuh.repository.PersonRepository; @RestController public class DataController { // 1 Spring Data JPA已自动为你注册bean,所以可自动注入 @Autowired PersonRepository personRepository; /** * 保存 save支持批量保存:<S extends T> Iterable<S> save(Iterable<S> entities); * * 删除: 支持使用id删除对象、批量删除以及删除全部: void delete(ID id); void delete(T entity); * void delete(Iterable<? extends T> entities); void deleteAll(); * */ @RequestMapping("/save") public Person save(@RequestBody Person person) { Person p = personRepository.save(person); return p; } /** * 测试findByAddress */ @RequestMapping("/q1") public List<Person> q1(String address) { List<Person> people = personRepository.findByAddress(address); return people; } /** * 测试findByNameAndAddress */ @RequestMapping("/q2") public Person q2(String name, String address) { Person people = personRepository.findByNameAndAddress(name, address); return people; } /** * 测试withNameAndAddressQuery */ @RequestMapping("/q3") public Person q3(String name, String address) { Person p = personRepository.withNameAndAddressQuery(name, address); return p; } /** * 测试withNameAndAddressNamedQuery */ @RequestMapping("/q4") public Person q4(String name, String address) { Person p = personRepository.withNameAndAddressNamedQuery(name, address); return p; } /** * 测试排序 */ @RequestMapping("/sort") public List<Person> sort() { List<Person> people = personRepository.findAll(new Sort(Direction.ASC, "age")); return people; } /** * 测试分页 */ @RequestMapping("/page") public Page<Person> page(@RequestParam("pageNo") int pageNo, @RequestParam("pageSize") int pageSize) { Page<Person> pagePeople = personRepository.findAll(new PageRequest(pageNo, pageSize)); return pagePeople; } }
import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.post; import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.content; import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status; import java.util.HashMap; import java.util.Map; import org.junit.Before; import org.junit.Test; import org.junit.runner.RunWith; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.boot.test.context.SpringBootTest; import org.springframework.http.MediaType; import org.springframework.test.context.junit4.SpringRunner; import org.springframework.test.web.servlet.MockMvc; import org.springframework.test.web.servlet.MvcResult; import org.springframework.test.web.servlet.setup.MockMvcBuilders; import org.springframework.web.context.WebApplicationContext; import net.minidev.json.JSONObject; @RunWith(SpringRunner.class) @SpringBootTest public class SpringBootStudentDataJpaApplicationTests { @Test public void contextLoads() { } private MockMvc mockMvc; // 模拟MVC对象,通过MockMvcBuilders.webAppContextSetup(this.wac).build()初始化。 @Autowired private WebApplicationContext wac; // 注入WebApplicationContext // @Autowired // private MockHttpSession session;// 注入模拟的http session // // @Autowired // private MockHttpServletRequest request;// 注入模拟的http request\ @Before // 在测试开始前初始化工作 public void setup() { this.mockMvc = MockMvcBuilders.webAppContextSetup(this.wac).build(); } @Test public void testQ1() throws Exception { Map<String, Object> map = new HashMap<>(); map.put("address", "合肥"); MvcResult result = mockMvc.perform(post("/q1?address=合肥").content(JSONObject.toJSONString(map))) .andExpect(status().isOk())// 模拟向testRest发送get请求 .andExpect(content().contentType(MediaType.APPLICATION_JSON_UTF8))// 预期返回值的媒体类型text/plain;charset=UTF-8 .andReturn();// 返回执行请求的结果 System.out.println(result.getResponse().getContentAsString()); } @Test public void testSave() throws Exception { Map<String, Object> map = new HashMap<>(); map.put("address", "合肥"); map.put("name", "测试"); map.put("age", 50); MvcResult result = mockMvc.perform(post("/save").contentType(MediaType.APPLICATION_JSON).content(JSONObject.toJSONString(map))) .andExpect(status().isOk())// 模拟向testRest发送get请求 .andExpect(content().contentType(MediaType.APPLICATION_JSON_UTF8))// 预期返回值的媒体类型text/plain;charset=UTF-8 .andReturn();// 返回执行请求的结果 System.out.println(result.getResponse().getContentAsString()); } @Test public void testPage() throws Exception { MvcResult result = mockMvc.perform(post("/page").param("pageNo", "1").param("pageSize", "2")) .andExpect(status().isOk())// 模拟向testRest发送get请求 .andExpect(content().contentType(MediaType.APPLICATION_JSON_UTF8))// 预期返回值的媒体类型text/plain;charset=UTF-8 .andReturn();// 返回执行请求的结果 System.out.println(result.getResponse().getContentAsString()); } }