Controller:
@RestController
public class UserController {
@Autowired
private UserRepository userRepository;
@GetMapping("users")
public List<User> userList() {
return userRepository.findAll();
}
@PostMapping("users")
public User addUser(@RequestParam("name") String name, @RequestParam("phone") String phone) {
User u = new User();
u.setName(name);
u.setPhone(phone);
this.userRepository.save(u);
return u;
}
}
单元测试:
@RunWith(SpringRunner.class)
@SpringBootTest
@AutoConfigureMockMvc
public class UserControllerTest {
@Autowired
private MockMvc mvc;
@Test
public void userList() throws Exception {
User u1= new User();
u1.setId(5L);
u1.setName("jim");
u1.setPhone("13011111111");
mvc.perform(MockMvcRequestBuilders.post("/users")//url
.param("name","jim")//参数1
.param("phone","13011111111"))//参数2
.andExpect(MockMvcResultMatchers.status().isOk())//是否返回200码
.andExpect(MockMvcResultMatchers.content().json(JSON.toJSONString(u1)));//返回数据是否和预期一样
}
}