Java tutorial
/* * Copyright 2012 ios-driver committers. * * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except * in compliance with the License. You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software distributed under the License * is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express * or implied. See the License for the specific language governing permissions and limitations under * the License. */ package org.uiautomation.ios.server.servlet; import java.io.IOException; import java.util.logging.Logger; import javax.servlet.ServletException; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; import org.json.JSONArray; import org.json.JSONException; import org.json.JSONObject; import org.openqa.selenium.WebDriverException; import org.openqa.selenium.remote.BeanToJsonConverter; import org.openqa.selenium.remote.ErrorCodes; import org.openqa.selenium.remote.Response; import org.uiautomation.ios.communication.WebDriverLikeCommand; import org.uiautomation.ios.communication.WebDriverLikeRequest; import org.uiautomation.ios.server.CommandMapping; import org.uiautomation.ios.server.command.Handler; public class IOSServlet extends DriverBasedServlet { private static final Logger log = Logger.getLogger(IOSServlet.class.getName()); private static final long serialVersionUID = -1190162363756488569L; private final ErrorCodes errorCodes = new ErrorCodes(); @Override protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException { try { process(request, response); } catch (Exception e) { // TODO Auto-generated catch block e.printStackTrace(); } } @Override protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException { try { process(request, response); } catch (Exception e) { // TODO Auto-generated catch block e.printStackTrace(); } } @Override protected void doDelete(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException { try { process(request, response); } catch (Exception e) { // TODO Auto-generated catch block e.printStackTrace(); } } private void process(HttpServletRequest request, HttpServletResponse response) throws Exception { WebDriverLikeRequest req = new WebDriverLikeRequest(request); response.setContentType("application/json;charset=UTF-8"); response.setCharacterEncoding("UTF-8"); response.setStatus(200); Response resp = getResponse(req); // TODO implement the json protocol properly. if (req.getGenericCommand() == WebDriverLikeCommand.NEW_SESSION && resp.getStatus() == 0) { response.setStatus(301); String session = resp.getSessionId(); String scheme = request.getScheme(); // http String serverName = request.getServerName(); // hostname.com int serverPort = request.getServerPort(); // 80 String contextPath = request.getContextPath(); // /mywebapp // Reconstruct original requesting URL String url = scheme + "://" + serverName + ":" + serverPort + contextPath; response.setHeader("location", url + "/session/" + session); } // String s = toString(resp); BeanToJsonConverter convertor = new BeanToJsonConverter(); String s = convertor.convert(resp); // status is also used for debugging, it's worth formatting it nice. if (req.getGenericCommand() == WebDriverLikeCommand.STATUS) { JSONObject o = new JSONObject(s); response.getWriter().print(o.toString(2)); } else { response.getWriter().print(s); } response.getWriter().close(); } private String toString(Response r) throws Exception { JSONObject o = new JSONObject(); o.put("sessionId", r.getSessionId()); o.put("status", r.getStatus()); o.put("value", r.getValue().toString()); return o.toString(); } private Response getResponse(WebDriverLikeRequest request) { long start = System.currentTimeMillis(); String command = ""; WebDriverLikeCommand wdlc = null; try { wdlc = request.getGenericCommand(); Handler h = CommandMapping.get(wdlc).createHandler(getDriver(), request); command = wdlc.method() + "\t " + wdlc.path(); return h.handleAndRunDecorators(); } catch (WebDriverException we) { Response response = new Response(); if (wdlc.isSessionLess()) { response.setSessionId(""); } else { response.setSessionId(request.getSession()); } response.setStatus(errorCodes.toStatusCode(we)); try { JSONObject o = serializeException(we); response.setValue(o); } catch (JSONException e) { // TODO Auto-generated catch block e.printStackTrace(); } return response; } catch (Exception e) { throw new WebDriverException("bug." + e.getMessage(), e); } finally { String message = ((System.currentTimeMillis() - start) + "ms.\t" + command); log.info(message); } } private JSONObject serializeException(Throwable e) throws JSONException { JSONObject res = new JSONObject(); res.put("message", e.getMessage()); res.put("class", e.getClass().getCanonicalName()); res.put("screen", JSONObject.NULL); res.put("stackTrace", serializeStackTrace(e.getStackTrace())); if (e.getCause() != null) { res.put("cause", serializeException(e.getCause())); } return res; } private JSONArray serializeStackTrace(StackTraceElement[] els) throws JSONException { JSONArray stacktace = new JSONArray(); for (StackTraceElement el : els) { JSONObject stckEl = new JSONObject(); stckEl.put("fileName", el.getFileName()); stckEl.put("className", el.getClassName()); stckEl.put("methodName", el.getMethodName()); stckEl.put("lineNumber", el.getLineNumber()); stacktace.put(stckEl); } return stacktace; } }