com.infotel.seleniumrobot.grid.servlets.server.NodeTaskServlet Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of seleniumRobot-grid Show documentation
Show all versions of seleniumRobot-grid Show documentation
Selenium grid extension for mobile testing
The newest version!
/**
* Copyright 2017 www.infotel.com
*
* 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 com.infotel.seleniumrobot.grid.servlets.server;
import java.awt.AWTException;
import java.awt.MouseInfo;
import java.awt.Point;
import java.awt.Robot;
import java.io.File;
import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.time.Instant;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Base64;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.stream.Collectors;
import javax.servlet.ServletException;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.apache.commons.io.FileUtils;
import org.apache.commons.io.IOUtils;
import org.apache.commons.lang.StringUtils;
import org.apache.log4j.Logger;
import org.json.JSONObject;
import org.openqa.grid.internal.utils.configuration.GridNodeConfiguration;
import com.infotel.seleniumrobot.grid.config.LaunchConfig;
import com.infotel.seleniumrobot.grid.servlets.client.HubTaskServletClient;
import com.infotel.seleniumrobot.grid.tasks.CommandTask;
import com.infotel.seleniumrobot.grid.tasks.EndTask;
import com.infotel.seleniumrobot.grid.tasks.KillTask;
import com.infotel.seleniumrobot.grid.tasks.NodeRestartTask;
import com.infotel.seleniumrobot.grid.utils.Utils;
import com.seleniumtests.browserfactory.BrowserInfo;
import com.seleniumtests.browserfactory.mobile.AppiumLauncher;
import com.seleniumtests.browserfactory.mobile.LocalAppiumLauncher;
import com.seleniumtests.driver.CustomEventFiringWebDriver;
import com.seleniumtests.driver.DriverMode;
import com.seleniumtests.util.osutility.OSCommand;
import com.seleniumtests.util.osutility.OSUtility;
import com.seleniumtests.util.osutility.OSUtilityFactory;
import com.seleniumtests.util.osutility.ProcessInfo;
import com.seleniumtests.util.video.VideoRecorder;
import com.sun.jna.platform.win32.Advapi32;
import com.sun.jna.platform.win32.Advapi32Util;
import com.sun.jna.platform.win32.WinReg;
/**
* Servlet for getting all mobile devices information
* This helps the hub to update capabilities with information on the connected device
* @author behe
*
*/
public class NodeTaskServlet extends GenericServlet {
/**
*
*/
private static final long serialVersionUID = 216473127866019518L;
private static final Logger logger = Logger.getLogger(NodeTaskServlet.class);
public static final String VIDEOS_FOLDER = "videos";
private static Map videoRecorders = Collections.synchronizedMap(new HashMap<>());
private static Map recordedFiles = Collections.synchronizedMap(new HashMap<>());
private static Map appiumLaunchers = Collections.synchronizedMap(new HashMap<>());
private NodeRestartTask restartTask = new NodeRestartTask();
private EndTask endTask;
private KillTask killTask;
private HubTaskServletClient hubTaskServletClient;
private GridNodeConfiguration gridNodeConfiguration;
private Object lock = new Object();
public NodeTaskServlet() {
this(new KillTask(),
new EndTask(),
LaunchConfig.getCurrentNodeConfig(),
new HubTaskServletClient(LaunchConfig.getCurrentNodeConfig().getHubHost(), LaunchConfig.getCurrentNodeConfig().getHubPort()));
}
// for test purpose
public NodeTaskServlet(KillTask killTask, EndTask endTask, GridNodeConfiguration gridNodeConfiguration, HubTaskServletClient hubTaskServletClient) {
super(null);
this.killTask = killTask;
this.endTask = endTask;
this.gridNodeConfiguration = gridNodeConfiguration;
this.hubTaskServletClient = hubTaskServletClient;
}
/**
* POST `/extra/NodeTaskServlet?action=` supports several actions
*
* - `action=restart`: restart node computer
* - `action=kill&process=`: kill a process by name without extension
* - `action=killPid&pid=`: kill a process by pid
* - `action=leftClic&x=&y=`: perform a left click at point x,y
* - `action=rightClic&x=&y=`: perform a right click at point x,y
* - `action=sendKeys&keycodes=,` where kcX is a key code. Sends keys to desktop. Used to send non alphanumeric keys
* - `action=writeText&text=`: write text to desktop.
* - `action=displayRunningStep&stepName=&session=`: write text to desktop.
* - `action=uploadFile&name=&content=` use browser to upload a file when a upload file window is displayed. The base64 content is copied to a temps file which will then be read by browser.
* - `action=setProperty&key=&value=` set java property for the node
*/
@Override
protected void doPost(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
String onlyMainScreenStr = "false";
switch (req.getParameter("action")) {
case "restart":
restartNode();
break;
case "stop":
sendOk(resp, "OK");
stopNode();
break;
// call POST /extra/NodeTaskServlet/kill with process=
case "kill":
synchronized (lock) {
killTask(req.getParameter("process"), resp);
}
break;
case "killPid":
synchronized (lock) {
killPid(Long.parseLong(req.getParameter("pid")), resp);
}
break;
// call POST /extra/NodeTaskServlet/leftClic with x=,y=,onlyMainScreen=
case "leftClic":
case "leftClick":
onlyMainScreenStr = req.getParameter("onlyMainScreen");
leftClick(onlyMainScreenStr == null ? false: Boolean.parseBoolean(onlyMainScreenStr), Integer.parseInt(req.getParameter("x")), Integer.parseInt(req.getParameter("y")), resp);
break;
// call POST /extra/NodeTaskServlet/doubleClic with x=,y=,onlyMainScreen=
case "doubleClick":
onlyMainScreenStr = req.getParameter("onlyMainScreen");
doubleClick(onlyMainScreenStr == null ? false: Boolean.parseBoolean(onlyMainScreenStr), Integer.parseInt(req.getParameter("x")), Integer.parseInt(req.getParameter("y")), resp);
break;
// call POST /extra/NodeTaskServlet/rightClic with x=,y=,onlyMainScreen=
case "rightClic":
case "rightClick":
onlyMainScreenStr = req.getParameter("onlyMainScreen");
rightClick(onlyMainScreenStr == null ? false: Boolean.parseBoolean(onlyMainScreenStr), Integer.parseInt(req.getParameter("x")), Integer.parseInt(req.getParameter("y")), resp);
break;
// call POST /extra/NodeTaskServlet/sendKeys with keycodes=, ... where kc is a key code
case "sendKeys":
List keyCodes = new ArrayList<>();
for (String kc: req.getParameter("keycodes").split(",")) {
keyCodes.add(Integer.parseInt(kc));
}
sendKeys(keyCodes, resp);
break;
// call POST /extra/NodeTaskServlet/writeText with text=
case "writeText":
writeText(req.getParameter("text"), resp);
break;
// call POST /extra/NodeTaskServlet/displayRunningStep with stepName=
case "displayRunningStep":
displayRunningStep(req.getParameter("stepName"), req.getParameter("session"), resp);
break;
// call POST /extra/NodeTaskServlet/uploadFile with name=,content=
case "uploadFile":
String content = "";
if ("application/octet-stream".equals(req.getContentType())) {
content = Base64.getEncoder().encodeToString(IOUtils.toByteArray(req.getInputStream()));
} else {
content = req.getParameter("content"); // old fashion, for compatibility with core <= 4.21.0;
}
uploadFile(req.getParameter("name"), content, resp);
break;
case "setProperty":
System.setProperty(req.getParameter("key"), req.getParameter("value"));
break;
case "clean":
cleanNode();
break;
// call POST /extra/NodeTaskServlet/command with name=,arg0=,arg1=
case "command":
List args = new ArrayList();
for (int i = 0; i < 10; i++) {
String value = req.getParameter(String.format("arg%d", i));
if (value == null) {
break;
} else {
args.add(value);
}
}
executeCommand(req.getParameter("name"),
args,
req.getParameter("session"),
req.getParameter("timeout") == null ? null: Integer.parseInt(req.getParameter("timeout")),
resp);
break;
default:
sendError(HttpServletResponse.SC_NOT_FOUND, resp, String.format("POST Action %s not supported by servlet", req.getParameter("action")));
break;
}
}
/**
* - `action=version`: returns the version of the node
* - `action=screenshot`: returns a base64 string of the node screen (PNG format)
* - `action=startVideoCapture&session=`: start video capture on the node. SessionId is used to store the video file
* - `action=stopVideoCapture&session=`: stop video capture previously created (use the provided sessionId)
* - `action=startAppium&session=`: start appium server
* - `action=stopAppium&session=`: stop the appium server previously started with corresponding sessionId
* - `action=driverPids&browserName=&browserVersion=&existingPids=`: Returns list of PIDS for this driver exclusively. This allows the hub to know which browser has been recently started. If existingPids is not empty, these pids won't be returned by the command. Browser name and version refers to installed browsers, declared in grid node
* - `action=browserAndDriverPids&browserName=&browserVersion=&parentPids=`: Returns list of PIDs for this driver and for all subprocess created (driver, browser and other processes). This allows to kill any process created by a driver. parentPids are the processs for which we should search child processes.
* - `action=keepAlive`: move mouse from 1 pixel so that windows session does not lock
* - `action=processList&name=myprocessName`: returns the list of PIDs whose process name is the requested one
*/
@Override
protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
switch (req.getParameter("action")) {
case "version":
sendVersion(resp);
break;
case "screenshot":
takeScreenshot(resp);
break;
case "startVideoCapture":
startVideoCapture(req.getParameter("session"), resp);
break;
case "stopVideoCapture":
stopVideoCapture(req.getParameter("session"), resp);
break;
case "startAppium":
startAppium(req.getParameter("session"), resp);
break;
case "stopAppium":
stopAppium(req.getParameter("session"), resp);
break;
case "mouseCoordinates":
mouseCoordinates(resp);
break;
case "driverPids":
String existingPidsStr = req.getParameter("existingPids");
List existingPids = existingPidsStr.isEmpty() ? new ArrayList<>(): Arrays.asList(existingPidsStr.split(","))
.stream()
.map(Long::parseLong)
.collect(Collectors.toList());
getBrowserPids(req.getParameter("browserName"), req.getParameter("browserVersion"), existingPids, resp);
break;
case "browserAndDriverPids":
String parentPidsStr = req.getParameter("parentPids");
List parentPids = parentPidsStr.isEmpty() ? new ArrayList<>(): Arrays.asList(parentPidsStr.split(","))
.stream()
.map(Long::parseLong)
.collect(Collectors.toList());
getAllBrowserSubprocessPids(req.getParameter("browserName"), req.getParameter("browserVersion"), parentPids, resp);
break;
case "processList":
String processName = req.getParameter("name");
getProcessList(processName, resp);
break;
case "keepAlive":
keepAlive();
break;
default:
sendError(HttpServletResponse.SC_NOT_FOUND, resp, String.format("GET Action %s not supported by servlet", req.getParameter("action")));
break;
}
}
private void executeCommand(String commandName, List args, String sessionKey, Integer timeout, HttpServletResponse resp) throws IOException {
ExecutorService executorService = null;
if (sessionKey != null) {
logger.info("disable timeout during command execution");
hubTaskServletClient.disableTimeout(sessionKey);
executorService = Executors.newSingleThreadExecutor();
executorService.submit(() -> {
while (true) {
hubTaskServletClient.keepDriverAlive(sessionKey);
try {
System.out.println("waiting");
Thread.sleep(gridNodeConfiguration.timeout * 500);
} catch (InterruptedException e) {
break;
}
}
});
}
logger.info("executing command " + commandName);
try {
CommandTask commandTask = CommandTask.getInstance();
commandTask.setCommand(commandName, args, timeout);
commandTask.execute();
sendOk(resp, commandTask.getResult());
} catch (Exception e) {
logger.warn("Could not exeecute command: " + e.getMessage(), e);
sendError(HttpServletResponse.SC_INTERNAL_SERVER_ERROR, resp, e.getMessage());
} finally {
if (sessionKey != null) {
logger.info("enable timeout after command execution");
executorService.shutdownNow();
hubTaskServletClient.enableTimeout(sessionKey);
logger.info("timeout enabled");
}
}
}
private void killTask(String taskName, HttpServletResponse resp) throws IOException {
logger.info("killing process " + taskName);
try {
assert taskName != null;
killTask.setTaskName(taskName);
killTask.execute();
sendOk(resp, "process killed");
} catch (Exception e) {
logger.warn("Could not kill process: " + e.getMessage(), e);
sendError(HttpServletResponse.SC_INTERNAL_SERVER_ERROR, resp, e.getMessage());
}
}
private void killPid(Long pid, HttpServletResponse resp) throws IOException {
logger.info("killing process " + pid);
try {
assert pid != null;
killTask.setTaskPid(pid);
killTask.execute();
sendOk(resp, "process killed");
} catch (Exception e) {
logger.warn("Could not kill process: " + e.getMessage(), e);
sendError(HttpServletResponse.SC_INTERNAL_SERVER_ERROR, resp, e.getMessage());
}
}
private void stopNode() {
logger.info("stopping");
try {
endTask.execute();
} catch (Exception e) {
logger.warn("Could not stop node: " + e.getMessage(), e);
}
}
private void restartNode() {
logger.info("restarting");
try {
restartTask.execute();
} catch (Exception e) {
logger.warn("Could not restart node: " + e.getMessage(), e);
}
}
/**
* Kill drivers and browsers
* clean temp folder
*/
private void cleanNode() {
// delete video files older than 8 hours
try {
Files.walk(Paths.get(Utils.getRootdir(), VIDEOS_FOLDER))
.filter(Files::isRegularFile)
.filter(p -> p.toFile().lastModified() < Instant.now().minusSeconds(8 * 3600).toEpochMilli())
.forEach(t -> {
try {
String name = t.getFileName().toString();
recordedFiles.remove(name);
Files.delete(t);
} catch (IOException e) {}
});
} catch (IOException e) {
}
// do not clear drivers and browser when devMode is true
if (LaunchConfig.getCurrentLaunchConfig().getDevMode()) {
return;
}
try {
OSUtilityFactory.getInstance().killAllWebBrowserProcess(true);
OSUtilityFactory.getInstance().killAllWebDriverProcess();
} catch (Exception e) {
logger.error("Error while kill browser / drivers", e);
}
File temp;
try {
temp = File.createTempFile("temp-file-name", ".tmp");
File tempFolder = temp.getParentFile().getAbsoluteFile();
FileUtils.cleanDirectory(tempFolder);
} catch (IOException e) {
}
// kill popup raised on windows when a driver crashes on Windows
if (OSUtility.isWindows()) {
try {
OSUtilityFactory.getInstance().killProcessByName("Werfault", true);
} catch (Exception e) {
}
}
// reset proxy setting to "automatic" if requested on windows
if (LaunchConfig.getCurrentLaunchConfig().getProxyConfig() != null) {
OSUtilityFactory.getInstance().setSystemProxy(LaunchConfig.getCurrentLaunchConfig().getProxyConfig());
}
}
/**
* Left clic on desktop
* @param x
* @param y
* @throws IOException
*/
private void leftClick(boolean onlyMainScreen, int x, int y, HttpServletResponse resp) throws IOException {
try {
logger.info(String.format("left clic at %d,%d", x, y));
CustomEventFiringWebDriver.leftClicOnDesktopAt(onlyMainScreen, x, y, DriverMode.LOCAL, null);
sendOk(resp, "left clic ok");
} catch (Exception e) {
sendError(HttpServletResponse.SC_INTERNAL_SERVER_ERROR, resp, e.getMessage());
}
}
private void mouseCoordinates(HttpServletResponse resp) throws IOException {
try {
logger.info("mouse coordinates");
Point coords = CustomEventFiringWebDriver.getMouseCoordinates(DriverMode.LOCAL, null);
sendOk(resp, String.format("%d,%d", coords.x, coords.y));
} catch (Exception e) {
sendError(HttpServletResponse.SC_INTERNAL_SERVER_ERROR, resp, e.getMessage());
}
}
/**
* double clic on desktop
* @param x
* @param y
* @throws IOException
*/
private void doubleClick(boolean onlyMainScreen, int x, int y, HttpServletResponse resp) throws IOException {
try {
logger.info(String.format("left clic at %d,%d", x, y));
CustomEventFiringWebDriver.doubleClickOnDesktopAt(onlyMainScreen, x, y, DriverMode.LOCAL, null);
sendOk(resp, "double clic ok");
} catch (Exception e) {
sendError(HttpServletResponse.SC_INTERNAL_SERVER_ERROR, resp, e.getMessage());
}
}
/**
* right clic on desktop
* @param x
* @param y
* @throws IOException
*/
private void rightClick(boolean onlyMainScreen, int x, int y, HttpServletResponse resp) throws IOException {
try {
logger.info(String.format("right clic at %d,%d", x, y));
CustomEventFiringWebDriver.rightClicOnDesktopAt(onlyMainScreen, x, y, DriverMode.LOCAL, null);
sendOk(resp, "right clic ok");
} catch (Exception e) {
sendError(HttpServletResponse.SC_INTERNAL_SERVER_ERROR, resp, e.getMessage());
}
}
/**
* screenshot of the desktop
* @throws IOException
*/
private void takeScreenshot(HttpServletResponse resp) throws IOException {
logger.info("taking screenshot");
try {
ServletOutputStream outputStream = resp.getOutputStream();
outputStream.print(CustomEventFiringWebDriver.captureDesktopToBase64String(DriverMode.LOCAL, null));
outputStream.flush();
} catch (IOException e) {
logger.error("Error sending reply", e);
sendError(HttpServletResponse.SC_INTERNAL_SERVER_ERROR, resp, e.getMessage());
}
}
/**
* Send key events
* @param keys
* @throws IOException
*/
private void sendKeys(List keys, HttpServletResponse resp) throws IOException {
try {
logger.info("sending keys: " + keys);
CustomEventFiringWebDriver.sendKeysToDesktop(keys, DriverMode.LOCAL, null);
sendOk(resp, "send keys");
} catch (Exception e) {
sendError(HttpServletResponse.SC_INTERNAL_SERVER_ERROR, resp, e.getMessage());
}
}
/**
* write text using keyboard
* @param text
* @throws IOException
*/
private void writeText(String text, HttpServletResponse resp) throws IOException {
try {
logger.info("writing text: " + text);
CustomEventFiringWebDriver.writeToDesktop(text, DriverMode.LOCAL, null);
sendOk(resp, "write text ok");
} catch (Exception e) {
sendError(HttpServletResponse.SC_INTERNAL_SERVER_ERROR, resp, e.getMessage());
}
}
/**
* write text using keyboard
* @param text
* @throws IOException
*/
private void displayRunningStep(String stepName, String sessionId, HttpServletResponse resp) throws IOException {
try {
logger.info("display step: " + stepName);
CustomEventFiringWebDriver.displayStepOnScreen(stepName, DriverMode.LOCAL, null, videoRecorders.get(sessionId));
sendOk(resp, "display step ok");
} catch (Exception e) {
sendError(HttpServletResponse.SC_INTERNAL_SERVER_ERROR, resp, e.getMessage());
}
}
/**
* upload file to browser
*
* @param fileName name of the file to upload
* @param fileContent content as a base64 string
* @throws IOException
*/
private void uploadFile(String fileName, String fileContent, HttpServletResponse resp) throws IOException {
try {
logger.info("uploading file: " + fileName);
CustomEventFiringWebDriver.uploadFile(fileName, fileContent, DriverMode.LOCAL, null);
sendOk(resp, "upload file ok");
} catch (Exception e) {
sendError(HttpServletResponse.SC_INTERNAL_SERVER_ERROR, resp, e.getMessage());
}
}
private void sendVersion(HttpServletResponse resp) {
Map version = new HashMap<>();
version.put("version", Utils.getCurrentversion());
try (
ServletOutputStream outputStream = resp.getOutputStream()) {
outputStream.print(new JSONObject(version).toString());
} catch (IOException e) {
logger.error("Error sending reply", e);
}
}
private void startVideoCapture(String sessionId, HttpServletResponse resp) throws IOException {
try {
logger.info("start video capture for session: " + sessionId);
String videoName = sessionId + ".avi";
VideoRecorder recorder = CustomEventFiringWebDriver.startVideoCapture(DriverMode.LOCAL, null, Paths.get(Utils.getRootdir(), VIDEOS_FOLDER).toFile(), videoName);
videoRecorders.put(sessionId, recorder);
sendOk(resp, "start video ok");
} catch (Exception e) {
sendError(HttpServletResponse.SC_INTERNAL_SERVER_ERROR, resp, e.getMessage());
}
}
/**
* Stop capture, send file and delete it afterwards
* @param sessionId
* @param resp
* @throws IOException
*/
private void stopVideoCapture(String sessionId, HttpServletResponse resp) throws IOException {
logger.info("stop video capture for session: " + sessionId);
try {
VideoRecorder recorder = videoRecorders.remove(sessionId);
File knownFile = recordedFiles.get(sessionId);
File videoFile;
if (recorder == null) {
if (knownFile == null) {
return;
} else {
videoFile = knownFile;
}
} else {
videoFile = CustomEventFiringWebDriver.stopVideoCapture(DriverMode.LOCAL, null, recorder);
}
if (videoFile != null) {
recordedFiles.put(sessionId, videoFile);
resp.setContentLengthLong(videoFile.length());
ServletOutputStream outputStream = resp.getOutputStream();
outputStream.write(FileUtils.readFileToByteArray(videoFile));
outputStream.flush();
}
logger.info("video capture stopped");
} catch (Exception e) {
sendError(HttpServletResponse.SC_INTERNAL_SERVER_ERROR, resp, e.getMessage());
}
}
private void sendOk(HttpServletResponse resp, String message) throws IOException {
resp.setStatus(HttpServletResponse.SC_OK);
resp.setCharacterEncoding(StandardCharsets.UTF_8.toString());
try (ServletOutputStream outputStream = resp.getOutputStream()) {
outputStream.print(message);
outputStream.flush();
} catch (Exception e) {
e.printStackTrace();
} finally {
}
}
private void sendError(int code, HttpServletResponse resp, String msg) throws IOException {
resp.setStatus(code);
try (ServletOutputStream outputStream = resp.getOutputStream()) {
outputStream.print(msg);
outputStream.flush();
} catch (Exception e) {
e.printStackTrace();
}
}
public static Map getVideoRecorders() {
return videoRecorders;
}
private void startAppium(String sessionId, HttpServletResponse resp) throws IOException {
logger.info("starting appium");
try {
String logDir = Paths.get(Utils.getRootdir(), "logs", "appium").toString();
// start appium before creating instance
AppiumLauncher appiumLauncher = new LocalAppiumLauncher(logDir);
appiumLauncher.startAppium();
appiumLaunchers.put(sessionId, appiumLauncher);
logger.info("appium is running on " + ((LocalAppiumLauncher)appiumLauncher).getAppiumServerUrl());
ServletOutputStream outputStream = resp.getOutputStream();
outputStream.print(((LocalAppiumLauncher)appiumLauncher).getAppiumServerUrl());
outputStream.flush();
} catch (Exception e) {
logger.error("Error sending appium URL", e);
sendError(HttpServletResponse.SC_INTERNAL_SERVER_ERROR, resp, e.getMessage());
}
}
private void stopAppium(String sessionId, HttpServletResponse resp) throws IOException {
try {
AppiumLauncher appiumLauncher = appiumLaunchers.remove(sessionId);
if (appiumLauncher != null) {
appiumLauncher.stopAppium();
}
sendOk(resp, "stop appium ok");
} catch (Exception e) {
sendError(HttpServletResponse.SC_INTERNAL_SERVER_ERROR, resp, e.getMessage());
}
}
/**
* Returns list of PID for the given browser
* @param browserType
* @param browserVersion
* @param resp
* @throws IOException
*/
private void getBrowserPids(String browserName, String browserVersion, List existingPids, HttpServletResponse resp) throws IOException {
logger.info("get driver pids for browser " + browserName);
BrowserInfo browserInfo = getBrowserInfo(browserName, browserVersion);
List pidsToReturn = new ArrayList<>();
// get pid pre-existing the creation of this driver. This helps filtering drivers launched by other tests or users
if (browserInfo != null) {
pidsToReturn.addAll(browserInfo.getDriverAndBrowserPid(existingPids));
}
try (
ServletOutputStream outputStream = resp.getOutputStream()) {
outputStream.print(StringUtils.join(pidsToReturn, ","));
outputStream.flush();
} catch (IOException e) {
logger.error("Error sending browser pids", e);
sendError(HttpServletResponse.SC_INTERNAL_SERVER_ERROR, resp, e.getMessage());
}
}
/**
* Returns list of PIDs corresponding to driver and browser (+ processes that could have been created by browser)
* @param browserType
* @param browserVersion
* @param resp
* @throws IOException
*/
private void getAllBrowserSubprocessPids(String browserName, String browserVersion, List parentPids, HttpServletResponse resp) throws IOException {
logger.info("get browser/driver pids for browser " + browserName);
BrowserInfo browserInfo = getBrowserInfo(browserName, browserVersion);
List subProcessPids = new ArrayList<>();
if (browserInfo != null) {
subProcessPids.addAll(browserInfo.getAllBrowserSubprocessPids(parentPids));
}
try (
ServletOutputStream outputStream = resp.getOutputStream()) {
outputStream.print(StringUtils.join(subProcessPids, ","));
outputStream.flush();
} catch (IOException e) {
logger.error("Error sending browser/driver pids", e);
sendError(HttpServletResponse.SC_INTERNAL_SERVER_ERROR, resp, e.getMessage());
}
}
/**
* Returns BrowserInfo corresponding to this browser name and version
* @param browserName
* @param browserVersion
* @return
*/
private BrowserInfo getBrowserInfo(String browserName, String browserVersion) {
List browserInfos = OSUtility.getInstalledBrowsersWithVersion().get(
com.seleniumtests.driver.BrowserType.getBrowserTypeFromSeleniumBrowserType(browserName));
if (browserInfos == null) {
return null;
}
// select the right browserInfo depending on browser version
BrowserInfo browserInfo = null;
for (BrowserInfo bi: browserInfos) {
browserInfo = bi; // get at least one of the browserInfo
if (bi.getVersion().equals(browserVersion)) {
break;
}
}
return browserInfo;
}
/**
* Send to requester, the list of PIDs whose name is the requested process name
* @param processName
* @throws IOException
*/
private void getProcessList(String processName, HttpServletResponse resp) throws IOException {
try {
List processList = OSUtilityFactory.getInstance().getRunningProcesses(processName);
List pidsToReturn = processList.stream()
.map(ProcessInfo::getPid)
.collect(Collectors.toList());
ServletOutputStream outputStream = resp.getOutputStream();
outputStream.print(StringUtils.join(pidsToReturn, ","));
outputStream.flush();
} catch (Exception e) {
logger.error("Error sending browser pids", e);
sendError(HttpServletResponse.SC_INTERNAL_SERVER_ERROR, resp, e.getMessage());
}
}
private void keepAlive() {
// do not clear drivers and browser when devMode is true
if (!LaunchConfig.getCurrentLaunchConfig().getDevMode()) {
Point mouseLocation = MouseInfo.getPointerInfo().getLocation();
if (mouseLocation != null) {
double choice = Math.random();
try {
if (choice > 0.5) {
new Robot().mouseMove(mouseLocation.x - 1, mouseLocation.y);
} else {
new Robot().mouseMove(mouseLocation.x + 1, mouseLocation.y);
}
} catch (AWTException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
}
}
/***** for tests */
public static Map getAppiumLaunchers() {
return appiumLaunchers;
}
public static void resetAppiumLaunchers() {
appiumLaunchers = Collections.synchronizedMap(new HashMap<>());
}
public static void resetVideoRecorders() {
videoRecorders = Collections.synchronizedMap(new HashMap<>());
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy