From d2fe95728c43ff1e7167c47e1692aa5f9798b60e Mon Sep 17 00:00:00 2001
From: Jonathan Cook <jonathan.cook@erars.plus.com>
Date: Tue, 30 Jun 2020 18:06:48 +0200
Subject: [PATCH] BAEL-4024 - Taking Screenshots with Selenium WebDriver
 (#9620)

* BAEL-4198 - Fix Selenium Live Tests

* BAEL-4198

* BAEL-4024 - Taking Screenshots with Selenium WebDriver

Co-authored-by: Jonathan Cook <jcook@sciops.esa.int>
---
 testing-modules/selenium-junit-testng/pom.xml |  6 ++
 .../TakeScreenShotSeleniumLiveTest.java       | 85 +++++++++++++++++++
 2 files changed, 91 insertions(+)
 create mode 100644 testing-modules/selenium-junit-testng/src/test/java/com/baeldung/selenium/screenshot/TakeScreenShotSeleniumLiveTest.java

diff --git a/testing-modules/selenium-junit-testng/pom.xml b/testing-modules/selenium-junit-testng/pom.xml
index 210c2051da..8d661997f8 100644
--- a/testing-modules/selenium-junit-testng/pom.xml
+++ b/testing-modules/selenium-junit-testng/pom.xml
@@ -41,6 +41,11 @@
             <artifactId>hamcrest-all</artifactId>
             <version>${hamcrest-all.version}</version>
         </dependency>
+        <dependency>
+            <groupId>ru.yandex.qatools.ashot</groupId>
+            <artifactId>ashot</artifactId>
+            <version>${ashot.version}</version>
+        </dependency>
     </dependencies>
 
     <build>
@@ -59,6 +64,7 @@
     <properties>
         <testng.version>6.10</testng.version>
         <selenium-java.version>3.4.0</selenium-java.version>
+        <ashot.version>1.5.4</ashot.version>
     </properties>
 
 </project>
diff --git a/testing-modules/selenium-junit-testng/src/test/java/com/baeldung/selenium/screenshot/TakeScreenShotSeleniumLiveTest.java b/testing-modules/selenium-junit-testng/src/test/java/com/baeldung/selenium/screenshot/TakeScreenShotSeleniumLiveTest.java
new file mode 100644
index 0000000000..cf705bb81f
--- /dev/null
+++ b/testing-modules/selenium-junit-testng/src/test/java/com/baeldung/selenium/screenshot/TakeScreenShotSeleniumLiveTest.java
@@ -0,0 +1,85 @@
+package com.baeldung.selenium.screenshot;
+
+import static org.junit.Assert.assertTrue;
+
+import java.io.File;
+import java.io.IOException;
+import java.util.concurrent.TimeUnit;
+
+import javax.imageio.ImageIO;
+
+import org.apache.commons.io.FileUtils;
+import org.junit.AfterClass;
+import org.junit.BeforeClass;
+import org.junit.Test;
+import org.openqa.selenium.By;
+import org.openqa.selenium.Capabilities;
+import org.openqa.selenium.OutputType;
+import org.openqa.selenium.TakesScreenshot;
+import org.openqa.selenium.WebElement;
+import org.openqa.selenium.chrome.ChromeDriver;
+import org.openqa.selenium.remote.DesiredCapabilities;
+
+import ru.yandex.qatools.ashot.AShot;
+import ru.yandex.qatools.ashot.Screenshot;
+import ru.yandex.qatools.ashot.coordinates.WebDriverCoordsProvider;
+import ru.yandex.qatools.ashot.shooting.ShootingStrategies;
+
+public class TakeScreenShotSeleniumLiveTest {
+
+    private static ChromeDriver driver;
+
+    @BeforeClass
+    public static void setUp() {
+        System.setProperty("webdriver.chrome.driver", resolveResourcePath("chromedriver.mac"));
+
+        Capabilities capabilities = DesiredCapabilities.chrome();
+        driver = new ChromeDriver(capabilities);
+        driver.manage()
+            .timeouts()
+            .implicitlyWait(5, TimeUnit.SECONDS);
+
+        driver.get("http://www.google.com/");
+    }
+
+    @AfterClass
+    public static void tearDown() throws IOException {
+        driver.close();
+
+        System.clearProperty("webdriver.chrome.driver");
+    }
+
+    @Test
+    public void whenGoogleIsLoaded_thenCaptureScreenshot() throws IOException {
+        takeScreenshot(resolveTestResourcePath("google-home.png"));
+
+        assertTrue(new File(resolveTestResourcePath("google-home.png")).exists());
+    }
+
+    @Test
+    public void whenGoogleIsLoaded_thenCaptureLogo() throws IOException {
+        WebElement logo = driver.findElement(By.id("hplogo"));
+
+        Screenshot screenshot = new AShot().shootingStrategy(ShootingStrategies.viewportPasting(1000))
+            .coordsProvider(new WebDriverCoordsProvider())
+            .takeScreenshot(driver, logo);
+
+        ImageIO.write(screenshot.getImage(), "jpg", new File(resolveTestResourcePath("google-logo.png")));
+        assertTrue(new File(resolveTestResourcePath("google-logo.png")).exists());
+    }
+
+    public void takeScreenshot(String pathname) throws IOException {
+        File src = ((TakesScreenshot) driver).getScreenshotAs(OutputType.FILE);
+        FileUtils.copyFile(src, new File(pathname));
+    }
+
+    private static String resolveResourcePath(String filename) {
+        File file = new File("src/main/resources/" + filename);
+        return file.getAbsolutePath();
+    }
+
+    private static String resolveTestResourcePath(String filename) {
+        File file = new File("src/test/resources/" + filename);
+        return file.getAbsolutePath();
+    }
+}