学习直接调用Webdriver JsonWireProtocol 的 RESTful API

package DriverJsonWirePro;

import java.io.File;
import java.io.FileOutputStream;
import java.util.Base64;
import java.util.Date;
import java.util.Base64.Decoder;

import org.apache.http.HttpResponse;
import org.apache.http.client.HttpClient;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.entity.StringEntity;
import org.apache.http.impl.client.HttpClientBuilder;
import org.apache.http.util.EntityUtils;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;

public class CallRestFul {
    static String hubUrl="http://localhost:4444/wd/hub";
    static String sessionApi="/session";
    static String deleteSessionApi="/session/:sessionId";
    static String getUrlApi="/session/:sessionId/url";
    static String findElementApi="/session/:sessionId/element";
    static String clickElementApi="/session/:sessionId/element/:id/click";
    static String sendKeysApi="/session/sessionId/element/:id/value";
    static String takeScreenApi="/session/:sessionId/screenshot";
    static HttpClient client = getHttpClient();
    static String sessionId;
    
    
    public static void main(String args[]) throws Exception{
        sessionId=getSessionId("firefox");
        try{
        navigateUrl("www.google.com");
        String elementId=findElementByCss(".mockClass.span #mockId");
        sendKeys(elementId,"JsonWireProtocol");
        elementId=findElementByCss(".mockClass #mockButon");
        clickElement(elementId);
        }catch(Throwable t){
            //TODO
        }finally {
            deleteSession(sessionId);
        }
    
        
    }
    
    public static String getSessionId(String browser) throws Exception{
        String para="{\"desiredCapabilities\":{\"browserName\":\""+browser+"\"}}";
        String url=hubUrl+sessionApi;
        HttpPost post=postRequest(url, JSON.parseObject(para));
        HttpResponse response = client.execute(post);
        if(response.getStatusLine().getStatusCode()!=200) return null;
            
        String respStr=EntityUtils.toString(response.getEntity());
        String sessionId=(String)JSON.parseObject(respStr).get("sessionId");
        CallRestFul.sessionId=sessionId;
        return sessionId;
    }
    
    public static HttpClient getHttpClient() {
        HttpClient httpClient= HttpClientBuilder.create().build();
        return httpClient;
    }
    
    
    public static boolean navigateUrl(String url) throws Exception {
        String apiUrl=hubUrl+getUrlApi.replace(":sessionId", sessionId);
        String para="{\"url\":\""+url+"\"}";
        HttpPost post=postRequest(apiUrl, JSON.parseObject(para));
        HttpResponse response=client.execute(post);
        if(response.getStatusLine().getStatusCode()!=200) return false;
        return true;
    }
    
    public static boolean deleteSession(String sessionId) throws Exception {
        String apiUrl=hubUrl+deleteSessionApi.replace(":sessionId", sessionId);
        HttpGet get=getRequest(apiUrl);
        HttpResponse response = client.execute(get);
        if(response.getStatusLine().getStatusCode()!=200) return false;
        return true;
    }
    
    public static String findElementByCss(String cssSelector) throws Exception{
        String para="{\"using\":\"css selector\","+"\"value\":\""+cssSelector+"\"}";
        String apiUrl=hubUrl+findElementApi.replace(":sessionId", sessionId);
        HttpPost post=postRequest(apiUrl, JSON.parseObject(para));
        HttpResponse response = client.execute(post);
        if(response.getStatusLine().getStatusCode()!=200) return null;
            
        String respStr=EntityUtils.toString(response.getEntity());
        String elementId=(String)JSON.parseObject(respStr).getJSONObject("value").get("ELEMENT");
        return elementId;
    }
    
    public static boolean clickElement(String elementId) throws Exception {
        String apiUrl=hubUrl+clickElementApi
                .replace(":sessionId", sessionId)
                .replace(":id", elementId);
        HttpPost post= postRequest(apiUrl,null);
        HttpResponse response = client.execute(post);
        if(response.getStatusLine().getStatusCode()!=200) return false;
        return true;
    }
    
    public static boolean sendKeys(String elementId, String textToSend) throws Exception{
        clickElement(elementId);
        String para="{\"value\":[\""+textToSend+"\"]}";
        String apiUrl=hubUrl+sendKeysApi
                .replace(":sessionId", sessionId)
                .replace(":id", elementId);
        HttpPost post=postRequest(apiUrl, JSON.parseObject(para));
        HttpResponse response = client.execute(post);
        if(response.getStatusLine().getStatusCode()!=200) return false;
        return true;
    }
    
    public static boolean takeScreenshot() throws Exception{
    
        String apiUrl=hubUrl+takeScreenApi.replace(":sessionId", sessionId);
        HttpPost post=postRequest(apiUrl, null);
        HttpResponse response = client.execute(post);
        if(response.getStatusLine().getStatusCode()!=200) return false;
        JSONObject jsonObject=JSON.parseObject(EntityUtils.toString(response.getEntity()));
        String srcreenInStr=(String)jsonObject.getString("value");
        saveToLocal(srcreenInStr);
        return true;
    }
    
    public static HttpPost postRequest(String url, JSONObject postJSON) {
        HttpPost post= new HttpPost(url);
        if(postJSON==null) 
            return post;
        
        StringEntity sEntity;
        try {
            sEntity = new StringEntity(JSON.toJSONString(postJSON),"utf-8");
            sEntity.setContentType("application/json");
            post.setEntity(sEntity);
        } catch (Throwable e){
            e.printStackTrace();
        }
        return post;
    }
    
    
    public static HttpGet getRequest(String url) {
        HttpGet get= new HttpGet(url);
        return get;
    }
    
    public static boolean saveToLocal(String base64String) throws Exception {
        Decoder decoder=Base64.getDecoder();
        byte[] bytes=decoder.decode(base64String);
        
        FileOutputStream fileOutputStream=new FileOutputStream(new File(".\\screenshot"+new Date().getTime()+".jpg"));
        fileOutputStream.write(bytes, 0, bytes.length);
        fileOutputStream.close();
        return true;
        
    }
}