最近想著搞搞UI自動化測試用來替代部分的回歸測試工作，但要進入系統就要登錄，而登錄又需要驗證碼，這個就比較蛋疼，正好人資中心那有提供登錄接口，解決思路就是：
1、調用人資的登錄接口登錄從接口返回結果中拿到驗證登錄用的參數
2、把參數塞入cookies中，跳轉到自己需要的頁面
一、調用登錄接口獲取返回參數
使用RestTemplate調用接口（只是聽說比較好用，不太了解里面的原理，自己只引入了jar包，其他的沒做配置）
使用POST方式，RestTemplate提供的方法有以下三個：
public <T> T postForObject(String url, Object request, Class<T> responseType, Object... uriVariables)
public <T> T postForObject(String url, Object request, Class<T> responseType, Map<String, ?> uriVariables)
public <T> T postForObject(URI url, Object request, Class<T> responseType)```
第二個參數request為請求的參數，我使用的是第一個方法，實際使用過程中原想著放個參數對象進去結果不行報400錯誤，如下
RestTemplate template = new RestTemplate();
LoginData param = new LoginData();//請求參數
param.setCompanyId(1);
param.setMd5Password("84d91117b84279f06e03cdecb0b5fa31");
param.setUserCode("200523");
String url = "https://login.dooioo.net/api/autotest/userLogin";
JSONObject result = template.postForObject(url, param, JSONObject.class);
然后想著放個Map，結果還是不行,如下，仍舊報400
RestTemplate template = new RestTemplate();
Map<String,Object> map = new HashMap<String,Object>();//創建一個map用來放請求參數
map.put("userCode", "200523");//把參數放進去
map.put("md5Password", "84d91117b84279f06e03cdecb0b5fa31");
map.put("companyId", 1);
String url = "https://login.dooioo.net/api/autotest/userLogin";//請求地址
JSONObject result = template.postForObject(url, null, JSONObject.class,map);//這種寫法仍舊報400
//JSONObject result = template.postForObject(url, map, JSONObject.class);//這種寫法仍舊報400
應該是我不了解里面的實際的Rest的原理，期望哪位大神能給我解釋下~~
以上兩種都失敗了，就干脆直接講請求的參數信息拼在url后面了，這種順利成功，如下
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類型
JSONObject result = JSONObject.parseObject(str);//轉成json，直接講接口結果弄成json的話，所得的creatAt格式不對為科學計數法的格式1。49940827876612，最終導致生成的loginCsrf不對，然后登陸失敗
}catch(Exception e){
System.out.println("登錄失敗");
e.printStackTrace();
}```
有一個小坑的地方，開始我直接將返回結果轉成json
JSONObject str = template.postForObject(url+param,null, JSONObject.class);//所得結果為調整成String類型
結果在后面的塞入cookie后跳轉到我的目標頁面時，一直提示我登錄信息過期，開始一直沒找到問題，后來把接口返回結果和cookie都打印出來才發現，接口返回結果的時間戳打印出來竟然是科學計數法的格式（這種：1.49940827876612），因為我需要用到這個參數來生成我的登錄參數，結果當然生成的一直是錯的，生成的是無效的，這個RestTemplate有點坑啊，這又是一個我不明白的地方,所以最終結果轉成String類型的，然后再轉成json，所以才有了上面這種，有點丑。
二、將返回參數塞入cookie
webDriver對cookie的操作：
//塞入cookie 先生成一個cookie示例，構造方法為Cookie(String name, String value, String domain, String path, Date expiry) 對應的反別名稱，值，域，路徑，過期時間
Cookie companyId_cookie = new Cookie("companyId",result.getString("companyId"),".dooioo.net","/",null);//result.getString("companyId")為獲取返回結果的companyId參數
//獲取cookie,有兩種方式
//第一種
System.out.println(driver.manage().getCookieNamed("name"));
//第二種通過selenium提供的Cookie類獲取
Set<Cookie> cookies = driver.manage().getCookies(); //獲取瀏覽器當前的cookies
for(Cookie cookie : cookies){ //打印出來
System.out.println("當前cookies為:" + cookie.getDomain() + " " + cookie.getName() + ":" + cookie.getValue());
}
//刪除cookie
//deleteAllCookies 刪除所有cookie
//deleteCookie 刪除指定的cookie，參數一個cookie對象
//deleteCookieNamed 根據cookie名稱刪除
//刪除名稱為value的cookie
driver.manage().deleteCookieNamed("value");
//刪除剛才新增的cookie java
driver.manage().deleteCookie(cookie);
//deleteAllCookies 刪除所有cookie
driver.manage().deleteAllCookies();
以上是對cookie的增，刪，取
最終的代碼如下:
需要提下的是塞入cookie前需要先訪問登錄頁面，不然cookie塞不進去，因為不知道塞哪
driver.get("https://login.dooioo.net");//不能直接塞cookie需要跳轉到d頁面
Cookie companyId_cookie = new Cookie("companyId",String.valueOf(result.getIntValue("companyId")),".dooioo.net","/",null);
Cookie login_ticket_cookie = new Cookie("login_ticket",result.getString("loginTicket"),".dooioo.net","/",null);
Cookie login_Csrf_cookie = new Cookie("login_csrf",loginCsrf,".dooioo.net","/",null);
driver.manage().addCookie(companyId_cookie);
driver.manage().addCookie(login_ticket_cookie);
driver.manage().addCookie(login_Csrf_cookie);
三、將以上兩部合并封裝成一個方法
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類型
JSONObject result = JSONObject.parseObject(str);//轉成json，直接講接口結果弄成json的話，所得的creatAt格式不對為科學計數法的格式1。49940827876612，最終導致生成的loginCsrf不對，然后登陸失敗
String loginCsrf = Hashing.md5().hashString(result.getString("createAt") + "" + String.valueOf(result.getString("userCode"))
+ "" + String.valueOf(result.getIntValue("companyId")), StandardCharsets.UTF_8).toString();
System.out.println("接口返回結果為："+ result);
driver.get("https://login.dooioo.net");//不能直接塞cookie需要跳轉到d頁面
Cookie companyId_cookie = new Cookie("companyId",String.valueOf(result.getIntValue("companyId")),".dooioo.net","/",null);
Cookie login_ticket_cookie = new Cookie("login_ticket",result.getString("loginTicket"),".dooioo.net","/",null);
Cookie login_Csrf_cookie = new Cookie("login_csrf",loginCsrf,".dooioo.net","/",null);
driver.manage().addCookie(companyId_cookie);
driver.manage().addCookie(login_ticket_cookie);
driver.manage().addCookie(login_Csrf_cookie);
}catch(Exception e){
System.out.println("登錄失敗");
e.printStackTrace();
}
}
