spring boot使用TestRestTemplate集成测试 RESTful 接口

时间：2020-04-01 19:04:30 阅读：157 评论：0 收藏：0 [点我收藏+]

标签：etc nic password pen use exchange ntc info lin

只是单纯的记录一下如何用TestRestTemplate访问受security保护的api，供以后查阅。

 
@Slf4j
 
@RunWith(SpringRunner.class)
 
@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
 
public class AccountControllerTests {
 
@Autowired
 
private TestRestTemplate restTemplate;
 
private HttpEntity httpEntity;
 
/**
 
 * 登录
 
 * @throws Exception
 
 */
 
private void login() throws Exception {
 
String expectStr = "{\"code\":0,\"msg\":\"success\"}";
 
MultiValueMap<String, String> map = new LinkedMultiValueMap<>();
 
map.add("username", "183xxxxxxxx");
 
map.add("password", "123456");
 
ResponseEntity responseEntity = restTemplate.postForEntity("/api/account/sign_in", map, String.class);
 
//添加cookie以保持状态
 
HttpHeaders headers = new HttpHeaders();
 
String headerValue = responseEntity.getHeaders().get("Set-Cookie").toString().replace("[", "");
 
headerValue = headerValue.replace("]", "");
 
headers.set("Cookie", headerValue);
 
httpEntity = new HttpEntity(headers);
 
assertThat(responseEntity.getBody()).isEqualTo(expectStr);
 
}
 
/**
 
 * 登出
 
 * @throws Exception
 
 */
 
private void logout() throws Exception {
 
String expectStr = "{\"code\":0,\"msg\":\"success\"}";
 
String result = restTemplate.postForObject("/api/account/sign_out", null, String.class, httpEntity);
 
httpEntity = null;
 
assertThat(result).isEqualTo(expectStr);
 
}
 
/**
 
 * 获取信息
 
 * @throws Exception
 
 */
 
private void getUserInfo() throws Exception {
 
Detail detail = new Detail();
 
detail.setNickname("疯狂的米老鼠");
 
detail.setNicknamePinyin("fengkuangdemilaoshu");
 
detail.setSex(1);
 
SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
 
detail.setCreatedAt(sdf.parse("2017-11-03 16:43:27"));
 
detail.setUpdatedAt(sdf.parse("2017-11-03 16:43:27"));
 
Role role = new Role();
 
role.setName("ROLE_USER_NORMAL");
 
Set<Role> roles = new HashSet<>();
 
roles.add(role);
 
User user = new User();
 
user.setId(1L);
 
user.setPhone("183xxxxxxxx");
 
user.setEmail("xxxxxx@gmail.com");
 
user.setDetail(detail);
 
user.setRoles(roles);
 
ResultBean<User> resultBean = new ResultBean<>();
 
resultBean.setData(user);
 
ObjectMapper om = new ObjectMapper();
 
String expectStr = om.writeValueAsString(resultBean);
 
ResponseEntity<String> responseEntity = restTemplate.exchange("/api/user/get_user_info", HttpMethod.GET, httpEntity, String.class);
 
assertThat(responseEntity.getBody()).isEqualTo(expectStr);
 
}
 
@Test
 
public void testAccount() throws Exception {
 
login();
 
getUserInfo();
 
logout();
 
}

标签：etc nic password pen use exchange ntc info lin

原文地址：https://www.cnblogs.com/zgq123456/p/12614545.html

踩

(0)

评论一句话评论（0）

分享档案

更多>

2021年07月29日 (22)
2021年07月28日 (40)
2021年07月27日 (32)
2021年07月26日 (79)
2021年07月23日 (29)
2021年07月22日 (30)
2021年07月21日 (42)
2021年07月20日 (16)
2021年07月19日 (90)
2021年07月16日 (35)

周排行