spring boot using TestRestTemplate integration testing RESTful interface

Simply record how access api security protected with TestRestTemplate, for later reference.

 

@ Slf4j
@RunWith(SpringRunner.class)
@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
public class AccountControllerTests {
@Autowired
Private Test Rest Template rest template;
private HttpEntity httpEntity;

/**
* log in
* @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);
// Add the cookie to maintain state
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);
}

/**
* Sign out
* @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);
}

/**
* getting information
* @throws Exception
*/
private void getUserInfo() throws Exception {
Detail detail = new Detail();
detail.setNickname ( "Crazy Mickey Mouse");
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( "[email protected]");
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();
}