com.infotel.seleniumrobot.grid.config.GridNodeConfiguration Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of seleniumRobot-grid4 Show documentation
Selenium grid extension for mobile testing
The newest version!
package com.infotel.seleniumrobot.grid.config;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;

import com.seleniumtests.browserfactory.SeleniumRobotCapabilityType;
import io.appium.java_client.remote.options.SupportsAutomationNameOption;
import org.json.JSONObject;
import org.openqa.selenium.MutableCapabilities;
import org.openqa.selenium.grid.node.config.NodeOptions;
import org.openqa.selenium.grid.server.BaseServerOptions;

import com.google.gson.Gson;

import org.openqa.selenium.remote.CapabilityType;

public class GridNodeConfiguration extends GridConfiguration {
	
	
	public static final String WEBDRIVER_PATH = "webdriver-executable";
	public static final String VIDEOS_FOLDER = "videos";
	private Map> configuration = new HashMap<>();
	
	public List capabilities = new ArrayList<>();
	public String appiumUrl = "http://localhost:4723/wd/hub"; 
	public List appiumCapabilities = new ArrayList<>(); // for node relay feature
	private BaseServerOptions serverOptions;
	private NodeOptions nodeOptions;
	
	public GridNodeConfiguration() {
		configuration.put("node", new HashMap<>());
		configuration.get("node").put("detect-drivers", false);
	}
	
	private int sessionTimeout = 540;

	/**
	 * https://www.selenium.dev/documentation/grid/configuration/toml_options/
	 * @return
	 */
	public String toToml() {
		
		StringBuilder tomlOut = new StringBuilder();
		
		for (Entry> confEntry: configuration.entrySet()) {
			tomlOut.append(String.format("[%s]\n", confEntry.getKey()));
			for (Entry subconfEntry: confEntry.getValue().entrySet()) {
				if (subconfEntry.getValue() instanceof String) {
					tomlOut.append(String.format("%s = \"%s\"\n", subconfEntry.getKey(), subconfEntry.getValue().toString()));
				} else {
					tomlOut.append(String.format("%s = %s\n", subconfEntry.getKey(), subconfEntry.getValue().toString()));
				}
			}

			tomlOut.append("\n");
		}
		tomlOut.append("\n");
		
		for (MutableCapabilities caps: capabilities) {
			tomlOut.append("[[node.driver-configuration]]\n");
			tomlOut.append(String.format("display-name = \"%s %s\"\n", caps.getBrowserName(), caps.getBrowserVersion()));
			if (caps.getCapability(WEBDRIVER_PATH) != null) {
				tomlOut.append(String.format("webdriver-executable = \"%s\"\n", caps.getCapability(WEBDRIVER_PATH)));
			}
			tomlOut.append(String.format("max-sessions = %d\n", caps.getCapability(LaunchConfig.TOTAL_SESSIONS)));
			Map stereotypesMap = new HashMap<>(caps.asMap());
			stereotypesMap.remove(WEBDRIVER_PATH);
			stereotypesMap.remove(LaunchConfig.TOTAL_SESSIONS);
			
			tomlOut.append(String.format("stereotype = \"%s\"\n", new Gson().toJson(stereotypesMap).toString().replace("\"", "\\\"")));

			tomlOut.append("\n");
		}
		tomlOut.append("\n");

		tomlOut.append("[distributor]\n");
		tomlOut.append("slot-matcher = \"com.infotel.seleniumrobot.grid.distributor.SeleniumRobotSlotMatcher\"\n");
		
		if (!appiumCapabilities.isEmpty()) {
			tomlOut.append("[relay]\n");
			tomlOut.append(String.format("url = \"%s\"\n", appiumUrl));
			tomlOut.append("status-endpoint = \"/status\"\n");
			tomlOut.append("configs = [");
			List configs = new ArrayList<>();


			for (MutableCapabilities caps: appiumCapabilities) {

				caps.setCapability(LaunchConfig.TOTAL_SESSIONS, (String)null);
				caps.setCapability(SeleniumRobotCapabilityType.APPIUM_PREFIX + SupportsAutomationNameOption.AUTOMATION_NAME_OPTION, (String)null);

				Map capsMap = new HashMap<>(caps.asMap());
				if (capsMap.containsKey(CapabilityType.PLATFORM_NAME)) {
					capsMap.put(CapabilityType.PLATFORM_NAME, capsMap.get(CapabilityType.PLATFORM_NAME).toString());
				}
				configs.add("\"1\"");
				configs.add(String.format("\"%s\"", new JSONObject(capsMap).toString().replace("\"", "\\\"")));
			}
			tomlOut.append(String.join(",", configs));
			tomlOut.append("]\n");


		}
		
		return tomlOut.toString();
	}
	
	public void addNodeConfiguration(String key, Object value) {
		configuration.get("node").put(key, value);
	}

	public BaseServerOptions getServerOptions() {
		return serverOptions;
	}

	public void setServerOptions(BaseServerOptions serverOptions) {
		this.serverOptions = serverOptions;
	}

	public NodeOptions getNodeOptions() {
		return nodeOptions;
	}

	public void setNodeOptions(NodeOptions nodeOptions) {
		this.nodeOptions = nodeOptions;
	}

	public List getCapabilities() {
		return capabilities;
	}

}