pickleib.mobile.driver.AppiumDriverFactory Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of Pickleib Show documentation
Pickleib library for supporting web application automation with Selenium
The newest version!
package pickleib.mobile.driver;

import com.google.gson.JsonObject;
import context.ContextStore;
import io.appium.java_client.AppiumDriver;
import io.appium.java_client.remote.options.BaseOptions;
import org.json.simple.JSONObject;
import org.openqa.selenium.remote.DesiredCapabilities;
import pickleib.driver.DriverFactory;
import utils.Printer;
import java.net.URL;
import static pickleib.mobile.driver.ServiceFactory.service;
import static utils.StringUtilities.Color.*;
import static utils.StringUtilities.*;

public class AppiumDriverFactory implements DriverFactory {

    static Printer log = new Printer(AppiumDriverFactory.class);
    static String deviceName;

    public static AppiumDriver getDriver(String deviceName, JSONObject capabilitiesJSON, boolean remote){
        AppiumDriverFactory.deviceName = deviceName;
        DesiredCapabilities capabilities = getConfig(capabilitiesJSON);
        String urlString;
        if (remote) {
            String userName = ContextStore.get("remote-mobile-username");
            String accessKey = ContextStore.get("remote-mobile-access-key");
            String server = ContextStore.get("remote-mobile-server");
            urlString = String.format("https://%s:%s@%s/wd/hub", userName , accessKey, server);
        }
        else {
            String address = ContextStore.get("address", "0.0.0.0");
            String port = ContextStore.get("port", "4723");
            urlString = "http://" + address + ":" + port + "/wd/hub";
            capabilities.setCapability("app", contextCheck("UPLOAD-" + capabilitiesJSON.get("app")));
            if (service != null) urlString = service.getUrl().toString();
        }
        return getDriver(capabilities, urlString);
    }

    public static AppiumDriver getDriver(DesiredCapabilities capabilities, String urlString){
        try {
            log.important(deviceName + markup(GRAY, " was selected"));
            return new AppiumDriver(new URL(urlString), capabilities);
        }
        catch (Exception gamma) {
            gamma.printStackTrace();
            if(gamma.toString().contains("Could not start a new session. Possible causes are invalid address of the remote server or browser start-up failure")){
                log.info("Please make sure " + markup(PURPLE, "Appium ") + "is on & verify the port that its running on at 'resources/test.properties'.");
                throw new RuntimeException(markup(YELLOW, gamma.getMessage()));
            }
            else throw new RuntimeException(markup(YELLOW, "Something went wrong while selecting a driver") + "\n" + markup(RED, gamma.getMessage()));
        }
    }

    public static DesiredCapabilities getConfig(JSONObject capabilities) {
        log.info("Setting capabilities...");
        DesiredCapabilities desiredCapabilities = new DesiredCapabilities();
        for (Object key : capabilities.keySet()) desiredCapabilities.setCapability((String) key, capabilities.get(key));
        return desiredCapabilities;
    }
}