Spring Boot Junit 测试Controller

Controller：

package com.xiaolyuh.controller;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;@RestControllerpublic class DataController {// 1 Spring Data JPA已自动为你注册bean，所以可自动注入@AutowiredPersonRepository 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;}}

测试类：

package com.xiaolyuh;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)@SpringBootTestpublic class SpringBootStudentDataJpaApplicationTests {@Testpublic 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());      }  }