android screenshot without appium

Author: Severi Haverila

src/main/java/AppiumCommons.java

@@ -1,4 +1,9 @@
+import java.io.File;
+import java.io.IOException;
+
+import org.apache.commons.io.FileUtils;
 import org.openqa.selenium.Dimension;
+import org.openqa.selenium.OutputType;
 import org.openqa.selenium.Point;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
@@ -17,6 +22,17 @@ public static void hideKeyboard(AppiumDriver<MobileElement> driver) {
             logger.debug(e.toString());
         }
     }
+    
+    public static void takeAppiumScreenshot(AppiumDriver<MobileElement> driver, String fullFileName) {
+		File scrFile = driver.getScreenshotAs(OutputType.FILE);
+		try {
+			File testScreenshot = new File(fullFileName);
+			FileUtils.copyFile(scrFile, testScreenshot);
+			logger.info("Screenshot stored to {}", testScreenshot.getAbsolutePath());
+		} catch (IOException e) {
+			e.printStackTrace();
+		}
+	}
 
     public static void swipeUp(AppiumDriver<MobileElement> driver) {
         swipeUp(driver, 0.15f, 0.15f);

src/main/java/TestdroidImageRecognition.java

@@ -235,28 +235,36 @@ public String takeScreenshot(String screenshotName) throws IOException, Interrup
     	String screenshotFile = screenshotsFolder + screenshotName + ".png";
 		String fullFileName = System.getProperty("user.dir") + "/" + screenshotFile;
 
-    	if (platformName.equalsIgnoreCase("iOS") && idevicescreenshotExists) {
+    	if (platformName.equalsIgnoreCase("iOS")) {
     		takeIDeviceScreenshot(fullFileName);
     	} else {
-    		takeAppiumScreenshot(fullFileName);
+    		takeAndroidScreenshot(fullFileName);
     	}
     	long end_time = System.nanoTime();
     	int difference = (int) ((end_time - start_time) / 1e6 / 1000);
     	logger.info("==> Taking a screenshot took " + difference + " secs.");
     	return screenshotFile;
 	}
 
-	private void takeAppiumScreenshot(String fullFileName) {
-		File scrFile = driver.getScreenshotAs(OutputType.FILE);
-		try {
-			File testScreenshot = new File(fullFileName);
-			FileUtils.copyFile(scrFile, testScreenshot);
-			logger.info("Screenshot stored to {}", testScreenshot.getAbsolutePath());
-		} catch (IOException e) {
-			e.printStackTrace();
+	private void takeAndroidScreenshot(String fullFileName) throws IOException, InterruptedException {
+		log("Taking android screenshot...");
+		log(fullFileName);
+		String[] cmd = new String[]{"screenshot2", "-d", fullFileName};
+		Process p = Runtime.getRuntime().exec(cmd);
+		BufferedReader in = new BufferedReader(new InputStreamReader(p.getInputStream()));
+		String line;
+		while ((line = in.readLine()) != null)
+			log(line);
+
+		int exitVal = p.waitFor();
+		if (exitVal != 0) {
+			log("screenshot2 process exited with value: " + exitVal);
 		}
 	}
 
+
+	
+
 	private static void takeIDeviceScreenshot(String fullFileName) throws IOException, InterruptedException {
 		String[] cmd = new String[]{"idevicescreenshot", "-u", udid, fullFileName};
 		Process p = Runtime.getRuntime().exec(cmd);

src/test/java/AndroidSample.java

@@ -55,7 +55,7 @@ public void mainPageTest() throws Exception {
 
         AppiumCommons.hideKeyboard(driver);
 
-        findImageOnScreen("bitbar_logo");
+        findImageOnScreen("bitbar_logo.png");
         log("Success.");
     }
 }