1 /**
  2  * 
  3  */
  4 package com.imooc.web.controller;
  5 
  6 import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.delete;
  7 import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.fileUpload;
  8 import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
 
  9 import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.post;
 10 import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.put;
 11 import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.jsonPath;
 12 import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
 
 13 
 14 import java.time.LocalDateTime;
 15 import java.time.ZoneId;
 16 import java.util.Date;
 17 
 18 import org.junit.Before;
 19 import org.junit.Test;
 20 import org.junit.runner.RunWith;
 21 import org.springframework.beans.factory.annotation.Autowired;
 22 import org.springframework.boot.test.context.SpringBootTest;
 
 23 import org.springframework.http.MediaType;
 24 import org.springframework.mock.web.MockMultipartFile;
 25 import org.springframework.test.context.junit4.SpringRunner;
 26 import org.springframework.test.web.servlet.MockMvc;
 27 import org.springframework.test.web.servlet.setup.MockMvcBuilders;
 28 import org.springframework.web.context.WebApplicationContext;
 29 
 30 /**
 31  * @author zhailiang
 32  *
 33  */
 34 @RunWith(SpringRunner.class)
 35 @SpringBootTest
 36 public class UserControllerTest {
 37 
 38     @Autowired
 39     private WebApplicationContext wac;
 40 
 41     private MockMvc mockMvc;
 42 
 43     @Before
 44     public void setup() {
 45         mockMvc = MockMvcBuilders.webAppContextSetup(wac).build();
 46     }
 47     
 48     @Test
 49     public void whenUploadSuccess() throws Exception {
 50         String result = mockMvc.perform(fileUpload("/file")
 51                 .file(new MockMultipartFile("file", "test.txt", "multipart/form-data", "hello upload".getBytes("UTF-8"))))
 52                 .andExpect(status().isOk())
 53                 .andReturn().getResponse().getContentAsString();
 54         System.out.println(result);
 55     }
 56     
 57 
 58     @Test
 59     public void whenQuerySuccess() throws Exception {
 60         String result = mockMvc.perform(
 61                 get("/user").param("username", "jojo").param("age", "18").param("ageTo", "60").param("xxx", "yyy")
 62                         // .param("size", "15")
 63                         // .param("page", "3")
 64                         // .param("sort", "age,desc")
 65                         .contentType(MediaType.APPLICATION_JSON_UTF8))
 66                 .andExpect(status().isOk()).andExpect(jsonPath("$.length()").value(3))
 67                 .andReturn().getResponse().getContentAsString();
 68         
 69         System.out.println(result);
 70     }
 71 
 72     @Test
 73     public void whenGetInfoSuccess() throws Exception {
 74         String result = mockMvc.perform(get("/user/1")
 75                 .contentType(MediaType.APPLICATION_JSON_UTF8))
 76                 .andExpect(status().isOk())
 77                 .andExpect(jsonPath("$.username").value("tom"))
 78                 .andReturn().getResponse().getContentAsString();
 79         
 80         System.out.println(result);
 81     }
 82     
 83     @Test
 84     public void whenGetInfoFail() throws Exception {
 85         mockMvc.perform(get("/user/a")
 86                 .contentType(MediaType.APPLICATION_JSON_UTF8))
 87                 .andExpect(status().is4xxClientError());
 88     }
 89     
 90     @Test
 91     public void whenCreateSuccess() throws Exception {
 92         
 93         Date date = new Date();
 94         System.out.println(date.getTime());
 95         String content = "{\"username\":\"tom\",\"password\":null,\"birthday\":"+date.getTime()+"}";
 96         String reuslt = mockMvc.perform(post("/user").contentType(MediaType.APPLICATION_JSON_UTF8)
 97                 .content(content))
 98                 .andExpect(status().isOk())
 99                 .andExpect(jsonPath("$.id").value("1"))
100                 .andReturn().getResponse().getContentAsString();
101         
102         System.out.println(reuslt);
103     }
104     
105     @Test
106     public void whenCreateFail() throws Exception {
107         
108         Date date = new Date();
109         System.out.println(date.getTime());
110         String content = "{\"username\":\"tom\",\"password\":null,\"birthday\":"+date.getTime()+"}";
111         String reuslt = mockMvc.perform(post("/user").contentType(MediaType.APPLICATION_JSON_UTF8)
112                 .content(content))
113 //                .andExpect(status().isOk())
114 //                .andExpect(jsonPath("$.id").value("1"))
115                 .andReturn().getResponse().getContentAsString();
116         
117         System.out.println(reuslt);
118     }
119     
120     @Test
121     public void whenUpdateSuccess() throws Exception {
122         
123         Date date = new Date(LocalDateTime.now().plusYears(1).atZone(ZoneId.systemDefault()).toInstant().toEpochMilli());
124         System.out.println(date.getTime());
125         String content = "{\"id\":\"1\", \"username\":\"tom\",\"password\":null,\"birthday\":"+date.getTime()+"}";
126         String reuslt = mockMvc.perform(put("/user/1").contentType(MediaType.APPLICATION_JSON_UTF8)
127                 .content(content))
128                 .andExpect(status().isOk())
129                 .andExpect(jsonPath("$.id").value("1"))
130                 .andReturn().getResponse().getContentAsString();
131         
132         System.out.println(reuslt);
133     }
134     
135     @Test
136     public void whenDeleteSuccess() throws Exception {
137         mockMvc.perform(delete("/user/1")
138                 .contentType(MediaType.APPLICATION_JSON_UTF8))
139                 .andExpect(status().isOk());
140     }
141 
142 }