Spring Boot Mock对Restful API测试 CRUD

package vip.fkandy;

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.mock.web.MockMultipartFile;
import org.springframework.test.context.junit4.SpringRunner;
import org.springframework.test.web.servlet.MockMvc;
import org.springframework.test.web.servlet.setup.MockMvcBuilders;
import org.springframework.web.context.WebApplicationContext;

import java.time.LocalDateTime;
import java.time.ZoneId;
import java.util.Date;

import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.*;
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.jsonPath;
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;

/**
* Spring Boot测试用例
*/
@RunWith(SpringRunner.class)
@SpringBootTest
public class UserControllerTest {
//伪造MVC环境，不会真正启动tomcat,所以会很快
@Autowired
private WebApplicationContext context;

private MockMvc mockMvc;

/**
* 每一个测试用例执行之前执行
*/
@Before
public void setUp() {
mockMvc = MockMvcBuilders.webAppContextSetup(context).build();
}

/**
* 查询列表成功测试用例
*
* @throws Exception
*/
@Test
public void whenQuerySuccess() throws Exception {
String result = mockMvc.perform(
get("/user")
.param("username", "jojo")
.param("age", "18")
.param("ageTo", "60")
.param("size", "15")
.param("page", "3")
.param("sort", "age,desc")
.contentType(MediaType.APPLICATION_JSON_UTF8))
.andExpect(status().isOk())
.andExpect(jsonPath("$.length()")
.value(3))
.andReturn().getResponse().getContentAsString();

System.out.println("Result:" + result);
}

/**
* 获取用户详细信息测试用例
*
* @throws Exception
*/
@Test
public void whenGetInfoSuccess() throws Exception {
String result = mockMvc.perform(
get("/user/1")
.contentType(MediaType.APPLICATION_JSON_UTF8))
.andExpect(status().isOk())
.andExpect(jsonPath("$.username")
.value("zhangsan"))
.andReturn().getResponse().getContentAsString();

System.out.println(result);
}

/**
* 获取用户信息失败测试用例
*
* @throws Exception
*/
@Test
public void whenGetInfoFail() throws Exception {
mockMvc.perform(get("/user/a")
.contentType(MediaType.APPLICATION_JSON_UTF8))
.andExpect(status().is4xxClientError());
}

/**
* 创建用户成功测试用例
*
* @throws Exception
*/
@Test
public void whenCreateSuccess() throws Exception {
Date date = new Date();
String content = "{\"username\":\"tom\",\"password\":\"123456\",\"birthday\":" + date.getTime() + "}";
String reuslt = mockMvc.perform(post("/user").contentType(MediaType.APPLICATION_JSON_UTF8)
.content(content))
.andExpect(status().isOk())
.andExpect(jsonPath("$.id").value("1"))
.andReturn().getResponse().getContentAsString();
System.out.println(reuslt);
}

/**
* 创建失败测试用例
*
* @throws Exception
*/
@Test
public void whenCreateFail() throws Exception {
Date date = new Date();
System.out.println(date.getTime());
String content = "{\"username\":\"tom\",\"password\":null,\"birthday\":" + date.getTime() + "}";
String reuslt = mockMvc.perform(post("/user").contentType(MediaType.APPLICATION_JSON_UTF8)
.content(content))
.andExpect(status().isOk())
.andExpect(jsonPath("$.id").value("1"))

4000
.andReturn().getResponse().getContentAsString();

System.out.println("---" + reuslt);
}

/**
* 修改用户信息成功测试用例
*
* @throws Exception
*/
@Test
public void whenUpdateSuccess() throws Exception {
Date date = new Date(LocalDateTime.now().plusYears(1).atZone(ZoneId.systemDefault()).toInstant().toEpochMilli());
System.out.println(date.getTime());
String content = "{\"id\":\"1\", \"username\":\"tom\",\"password\":null,\"birthday\":" + date.getTime() + "}";
String reuslt = mockMvc.perform(put("/user/1").contentType(MediaType.APPLICATION_JSON_UTF8)
.content(content))
.andExpect(status().isOk())
.andExpect(jsonPath("$.id").value("1"))
.andReturn().getResponse().getContentAsString();

System.out.println(reuslt);
}

/**
* 删除用户成功测试用例
*
* @throws Exception
*/
@Test
public void whenDeleteSuccess() throws Exception {
mockMvc.perform(delete("/user/1")
.contentType(MediaType.APPLICATION_JSON_UTF8))
.andExpect(status().isOk()); //因为是restful api 所以只需要查看状态码就行了
}

/**
* 上传成功测试用例
*
* @throws Exception
*/
@Test
public void whenUploadSuccess() throws Exception {
String result = mockMvc.perform(fileUpload("/file")
.file(new MockMultipartFile("file", "test.txt", "multipart/form-data", "hello upload".getBytes("UTF-8"))))
.andExpect(status().isOk())
.andReturn().getResponse().getContentAsString();
System.out.println(result);
}
}