MockMVC测试Controller中常见的请求方式

叁歲伎倆 2022-04-18 01:35 484阅读 0赞

##### TestController.java #####

@RestController
    public class TestController {
    
        private final String PATH = "D:\\Develop\\JavaEE\\laboratory";
    
        @GetMapping(value = "/mockTestA")
        public String mockTestA(@RequestParam("a") String a,
                                @RequestParam("b") String b) {
            return a + b;
        }
    
        @PostMapping(value = "/mockTestB")
        public String mockTestB(@RequestParam(value = "a", defaultValue = "1") String a,
                                @RequestParam(value = "b", defaultValue = "2") String b) {
            return a + b;
        }
    
        @PostMapping(value = "/mockTestC")
        public User mockTestC(@RequestBody User user) {
            return user;
        }
    
        @PostMapping(value = "/file")
        public String file(MultipartFile file) throws IOException {
            File localFile = new File(PATH, "test.txt");
    
            file.transferTo(localFile);
    
            return file.getName();
        }
        
    }

##### User.java #####

@Setter
    @Getter
    @AllArgsConstructor
    @NoArgsConstructor
    public class User {
    
        private String id;
    
        private String username;
    }

##### TestControllerTest #####

@RunWith(SpringRunner.class)
    @SpringBootTest
    public class TestControllerTest {
    
        @Autowired
        private WebApplicationContext wac;
    
        private MockMvc mockMvc;
    
        @Before
        public void setUp() {
            mockMvc = MockMvcBuilders.webAppContextSetup(wac).build();
        }
    
        @Test
        public void mockTestA() throws Exception {
            String result = mockMvc.perform(get("/mockTestA")
                    .param("a", "1")        // 添加参数
                    .param("b", "2")
                    .contentType(MediaType.APPLICATION_JSON_UTF8))      // 设置数据格式
                    .andDo(print())     // 打印输出发出请求的详细信息
                    .andExpect(status().isOk())     // 对返回值进行断言
                    .andReturn().getResponse().getContentAsString();        // 获取方法的返回值
    
            Assert.assertEquals("12", result);
        }
    
        @Test
        public void mockTestB() throws Exception {
            String result = mockMvc.perform(post("/mockTestB")
                    .param("a", "10")
                    .param("b", "11")
                    .contentType(MediaType.APPLICATION_JSON_UTF8))
                    .andDo(print())
                    .andExpect(status().isOk())
                    .andReturn().getResponse().getContentAsString();
    
            Assert.assertEquals("1011", result);
        }
    
        @Test
        public void mockTestC() throws Exception {
            Map<String, String> map = new HashMap<>();
            map.put("id", "123");
            map.put("username", "tom");
            String content = JSONObject.toJSONString(map);
    
            String result = mockMvc.perform(post("/mockTestC")
                    .contentType(MediaType.APPLICATION_JSON_UTF8)
                    .content(content))
                    .andDo(print())
                    .andExpect(status().isOk())
                    .andExpect(jsonPath("$.id").value("123"))
                    .andReturn().getResponse().getContentAsString();
    
            Assert.assertNotNull(result);
        }
    
        @Test
        public void file() throws Exception {
            String result = mockMvc.perform(fileUpload("/file")
                    .file(new MockMultipartFile("file", "test.txt",
                            "multipart/form-data",
                            "hello upload".getBytes("UTF-8"))))
                    .andDo(print())
                    .andExpect(status().isOk())
                    .andReturn().getResponse().getContentAsString();
            Assert.assertEquals("file", result);
        }
    }