Scroll.java

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.TimeUnit;

import org.openqa.selenium.By;
import org.openqa.selenium.Dimension;
import org.openqa.selenium.NoSuchElementException;
import org.openqa.selenium.TimeoutException;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.remote.RemoteWebDriver;
import org.openqa.selenium.support.ui.FluentWait;

import com.google.common.base.Function;

import io.appium.java_client.*;
import io.appium.java_client.android.*;
import io.appium.java_client.ios.*;

public class Scroll {

	// IOS Implementation

	// Generic scroll using send keys
	// Pass in values to be selected as a String array to the list parameter
	// Method will loop through looking for scroll wheels based on the number of
	// values you supply
	// For instance Month, Day, Year for a birthday would have this loop 3 times
	// dynamically selecting each scroll wheel
	public static void scrollKeys(IOSDriver driver, String[] list) {
		System.out.println("Starting the process");
		for (int i = 0; i < list.length; i++) {
			MobileElement we = (MobileElement) driver.findElementByXPath("//UIAPickerWheel["+(i+1)+"]");
			we.sendKeys(list[i]);
		}
		System.out.println("Ending Process");
	}

	// Scrolling using a slower however necessary process in some cases where
	// the value can't be set for ios scroll wheels
	// Pass in values to be selected as a String array to the list parameter
	// Method will loop through looking for scroll wheels based on the number of
	// values you supply
	// For instance Month, Day, Year for a birthday would have this loop 3 times
	// dynamically selecting each scroll wheel
	// Some additional leg work needs to be done here to determine if you should
	// be scrolling up or down
	// In the below implementation i'm checking the current month against the
	// month i supplied (by name) and
	// based on this i'm telling the scroll code to go up or down searching for
	// the value
	// additionally i'm checking the same for the day and year
	public static void scrollChecker(IOSDriver driver, String[] list) {
		System.out.println("Starting the process");
		for (int i = 0; i < list.length; i++) {

			MobileElement me = (MobileElement) driver.findElement(By.xpath("//UIAPickerWheel["+(i+1)+"]"));
			int mget = getMonthInt(me.getText().split(",")[0]);

			if (i == 0) {
				if (mget > getMonthInt(list[i])) {
					scrollAndSearch(driver, list[i], me, true);
				} else {
					scrollAndSearch(driver, list[i], me, false);
				}
			} else {
				if (Integer.parseInt(me.getText().split(",")[0]) > Integer.parseInt(list[i])) {
					scrollAndSearch(driver, list[i], me, true);
				} else {
					scrollAndSearch(driver, list[i], me, false);
				}
			}
		}
		System.out.println("Ending Process");
	}

	// Used to get the dynamic location of an object
	// Code here shouldn't be modified
	private static void scrollAndSearch(IOSDriver driver, String value, MobileElement me, Boolean direction) {
		String x = getLocationX(me);
		String y = getLocationY(me);
		while (!driver.findElementByXPath(getXpathFromElement(me)).getText().contains(value)) {
			swipe(driver, x, y, direction);
		}
	}

	// Performs the swipe and search operation
	// Code here shouldn't be modified
	private static void swipe(IOSDriver driver, String start, String end, Boolean up) {
		String direction;
		if (up) {
			direction = start + "," + (Integer.parseInt(end) + 70);
		} else {
			direction = start + "," + (Integer.parseInt(end) - 70);
		}

		Map<String, Object> params1 = new HashMap<>();
		params1.put("location", start + "," + end);
		params1.put("operation", "down");
		Object result1 = driver.executeScript("mobile:touch:tap", params1);

		Map<String, Object> params2 = new HashMap<>();
		List<String> coordinates2 = new ArrayList<>();

		coordinates2.add(direction);
		params2.put("location", coordinates2);
		params2.put("auxiliary", "notap");
		params2.put("duration", "3");
		Object result2 = driver.executeScript("mobile:touch:drag", params2);

		Map<String, Object> params3 = new HashMap<>();
		params3.put("location", direction);
		params3.put("operation", "up");
		Object result3 = driver.executeScript("mobile:touch:tap", params3);
	}

	// Android Implementation

	// Generic scroll using send keys
	// Pass in values to be selected as a String array to the list parameter
	// Method will loop through looking for scroll wheels based on the number of
	// values you supply
	// For instance Month, Day, Year for a birthday would have this loop 3 times
	// dynamically selecting each scroll wheel
	// Code here shouldn't be modified
	public static void scrollKeys(AndroidDriver driver, String[] list) {
		System.out.println("Starting the process");
		for (int i = 0; i < list.length; i++) {

			By meX = By.xpath("//android.widget.NumberPicker["+(i+1)+"]/android.widget.EditText[1]");
			fluentWait(driver, meX);
			MobileElement me = (MobileElement) driver.findElement(meX);

			TouchAction touchAction6 = new TouchAction(driver);
			touchAction6.longPress(me, 1000).release();
			driver.performTouchAction(touchAction6);

			driver.getKeyboard().pressKey(convertAndroidMonthName(list[i]) + "");

		}
		System.out.println("Ending Process");
	}

	// Used to get the integer for a month based on the string of the month
	private static int getMonthInt(String month) {
		int monthInt = 0;
		switch (month) {
		case "Jan":
			monthInt = 1;
			break;
		case "January":
			monthInt = 1;
			break;
		case "February":
			monthInt = 2;
			break;
		case "Feb":
			monthInt = 2;
			break;
		case "March":
			monthInt = 3;
			break;
		case "Mar":
			monthInt = 3;
			break;
		case "April":
			monthInt = 4;
			break;
		case "Apr":
			monthInt = 4;
			break;
		case "May":
			monthInt = 5;
			break;
		case "June":
			monthInt = 6;
			break;
		case "Jun":
			monthInt = 6;
			break;
		case "July":
			monthInt = 7;
			break;
		case "Jul":
			monthInt = 7;
			break;
		case "August":
			monthInt = 8;
			break;
		case "Aug":
			monthInt = 8;
			break;
		case "September":
			monthInt = 9;
			break;
		case "Sep":
			monthInt = 9;
			break;
		case "October":
			monthInt = 10;
			break;
		case "Oct":
			monthInt = 10;
			break;
		case "November":
			monthInt = 11;
			break;
		case "Nov":
			monthInt = 11;
			break;
		case "December":
			monthInt = 12;
			break;
		case "Dec":
			monthInt = 12;
			break;
		}
		return monthInt;
	}

	// converts the full string of the month to androids short form name
	private static String convertAndroidMonthName(String month) {
		String monthName = "";
		switch (month) {
		case "January":
			monthName = "Jan";
			break;
		case "February":
			monthName = "Feb";
			break;
		case "March":
			monthName = "Mar";
			break;
		case "April":
			monthName = "Apr";
			break;
		case "May":
			monthName = "May";
			break;
		case "June":
			monthName = "Jun";
			break;
		case "July":
			monthName = "Jul";
			break;
		case "August":
			monthName = "Aug";
			break;
		case "September":
			monthName = "Sep";
			break;
		case "October":
			monthName = "Oct";
			break;
		case "November":
			monthName = "Nov";
			break;
		case "December":
			monthName = "Dec";
			break;
		default:
			monthName = month;
			break;
		}
		return monthName;
	}

	// Parses webelement to retrieve the xpath used for identification
	private static String getXpathFromElement(MobileElement me) {
		return (me.toString().split("-> xpath: ")[1]).substring(0, (me.toString().split("-> xpath: ")[1]).length() - 1);
	}

	// Gets the objects X location in pixels
	private static String getLocationX(MobileElement me) {
		int x = me.getLocation().x;
		int width = (Integer.parseInt(me.getAttribute("width")) / 2) + x;
		return width + "";
	}

	// Gets the objects X location in pixels
	private static String getLocationY(MobileElement me) {
		int y = me.getLocation().y;
		int height = (Integer.parseInt(me.getAttribute("height")) / 2) + y;
		return height + "";
	}

	// performs a wait command on a web element
	private static MobileElement fluentWait(AppiumDriver driver, By xpath) {
		MobileElement waitElement = null;

		FluentWait<RemoteWebDriver> fwait = new FluentWait<RemoteWebDriver>(driver).withTimeout(15, TimeUnit.SECONDS)
				.pollingEvery(500, TimeUnit.MILLISECONDS).ignoring(NoSuchElementException.class)
				.ignoring(TimeoutException.class);

		try {
			waitElement = (MobileElement) fwait.until(new Function<RemoteWebDriver, WebElement>() {
				public WebElement apply(RemoteWebDriver driver) {
					return driver.findElement(xpath);
				}
			});
		} catch (Exception e) {
		}
		return waitElement;
	}
}