最近想着搞搞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();
}
}