最近做自动化测试，使用RestTemplate调用接口，碰到了各种场景下，什么都不带的，带cookie的，带header的，带参数的，需要获取返回状态码的，先在这稍微总结下，不定期更新
一、获取接口返回状态码
使用getForEntity调用接口，返回结果调用getStatusCode()方法取得HttpStatus对象，然后就可以调用里面的各种方法来满足你的需求了
//判断接口返回是否为200
public static Boolean ping(){
String url = domain + "/it/ping";
try{
ResponseEntity<String> responseEntity = template.getForEntity(url,String.class);
HttpStatus status = responseEntity.getStatusCode();//获取返回状态
return status.is2xxSuccessful();//判断状态码是否为2开头的
}catch(Exception e){
return false; //502 ,500是不能正常返回结果的，需要catch住，返回一个false
}
}
二、什么都不带，将参数拼接在请求url后面
将参数拼接在请求url后面，postForObject请求参数为null
public static void login(String userCode,String md5Password,int companyId,WebDriver driver){
RestTemplate template = new RestTemplate();
String url = "https://login.dooioo.net/api/autotest/userLogin";//请求地址
String param ="?userCode=" + userCode + "&md5Password=" + md5Password + "&companyId=" + companyId;
try{
String str = template.postForObject(url+param,null, String.class);//所得结果为调整成String类型
}catch(Exception e){
System.out.println("登录失败");
e.printStackTrace();
}
}
三、带cookie，header，参数
带cookie实际也是将参数塞入header中:
1、定义header对象： HttpHeaders headers = new HttpHeaders()
2、将要的cookie塞入header：headers.put(HttpHeaders.COOKIE,cookieList)（注意cookie需要是一个list，内容为 name=value 例如：loginticket=sldjfas112sadfsd）
3、也可以在Header中塞入其他值：headers.setContentType(MediaType.APPLICATION_FORM_URLENCODED)
/*
获取浏览器的cookie，将其塞入header中
*/
public static HttpHeaders getHeader(WebDriver driver){
HttpHeaders headers = new HttpHeaders();
Set<Cookie> cookies = driver.manage().getCookies();//获取浏览器cookies
List<String> cookieList = new ArrayList<String>();
for(Cookie cookie:cookies){ //将浏览器cookies放入list中
//System.out.println("当前cookies为:" + cookie.getDomain() + " " + cookie.getName() + ":" + cookie.getValue());
cookieList.add(cookie.getName() + "=" + cookie.getValue());
}
//System.out.println("cookie为：" + cookieList.toString());
headers.put(HttpHeaders.COOKIE,cookieList); //将cookie放入header
headers.setContentType(MediaType.APPLICATION_FORM_URLENCODED); //post表单 ，如果是个json则设置为MediaType.APPLICATION_JSON
return headers;
}
放入参数：
1、使用MultiValueMap用来放参数，（使用HashMap不行,具体原因可见http://www.cnblogs.com/shoren/p/RestTemplate-problem.html ），
2、根据header和参数实例化HttpEntity ,然后调用postForEntity方法调用接口
HttpEntity<MultiValueMap<String,String>> request = new HttpEntity<MultiValueMap<String,String>>(param,headers)
完整代码如下（使用方法可参考：https://segmentfault.com/a/1190000007778403）：
public static void gitUpLoad(WebDriver driver,String uploadUrl,String fileName,String content,String branch,String requestUrl) throws Exception{
String authenticity_token = getToken(driver,uploadUrl);//获取token
RestTemplate template = new RestTemplate();
HttpHeaders headers = getHeader(driver);//获取header
MultiValueMap<String,String> param = new LinkedMultiValueMap<String, String>();//参数放入一个map中，restTemplate不能用hashMap
//将请求参数放入map中
param.add("authenticity_token",authenticity_token);
param.add("file_name",fileName);
param.add("encoding","text");
param.add("commit_message","addFile");
param.add("target_branch",branch);
param.add("original_branch",branch);
param.add("content",content);
param.add("utf8","✓");
//System.out.println("参数内容为：" + param.toString());
HttpEntity<MultiValueMap<String,String>> request = new HttpEntity<MultiValueMap<String,String>>(param,headers);//将参数和header组成一个请求
ResponseEntity<String> response = template.postForEntity(requestUrl, request, String.class);
}
四、使用exchange指定调用方式
使用exchange方法可以指定调用方式
需要注意的一点是对于返回结果为204 no content,这种没有返回值的请求，RestTemplate会抛错，有需要的话可以使用httpClient的fluent
public void deleteQueue(String vhost,String queue){
HttpHeaders headers = new HttpHeaders();//header参数
headers.add("authorization",Auth);
headers.setContentType(MediaType.APPLICATION_JSON);
JSONObject content = new JSONObject();//放入body中的json参数
content.put("mode","delete");
content.put("name",queue);
content.put("vhost","/" + vhost);
String newVhost = "%2F" + vhost;
String newUrl = url + "/api/queues/" + newVhost + "/" + queue;
HttpEntity<JSONObject> request = new HttpEntity<>(content,headers); //组装
ResponseEntity<String> response = template.exchange(newUrl,HttpMethod.DELETE,request,String.class);
}
