- add ScreenShotUtil

src/test/java/com/ea/springbasic/test/SpringTestNGTest.java

@@ -1,6 +1,7 @@
 package com.ea.springbasic.test;
 
 import com.ea.springbasic.pages.MainPage2;
+import com.ea.springbasic.util.ScreenShotUtil;
 import org.openqa.selenium.WebDriver;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Value;
@@ -23,6 +24,9 @@ public class SpringTestNGTest extends AbstractTestNGSpringContextTests {
     @Autowired
     private MainPage2 mainPage2;
 
+    @Autowired
+    private ScreenShotUtil screenShotUtil;
+
     @BeforeMethod(alwaysRun = true)
     protected void setupWebDriver() {
         webDriver.navigate().to(appUrl);
@@ -38,6 +42,7 @@ protected void springTestContextPrepareTestInstance() throws Exception {
     @Test
     public void test() {
         mainPage2.performLogin();
+        screenShotUtil.takeScreenShot();
     }
 
     @AfterMethod

src/test/java/com/ea/springbasic/util/ScreenShotUtil.java

@@ -0,0 +1,35 @@
+package com.ea.springbasic.util;
+
+import org.apache.commons.io.FileUtils;
+import org.openqa.selenium.OutputType;
+import org.openqa.selenium.TakesScreenshot;
+import org.openqa.selenium.WebDriver;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.stereotype.Component;
+
+import java.io.File;
+import java.io.IOException;
+import java.nio.file.Path;
+import java.util.UUID;
+
+@Component
+public class ScreenShotUtil {
+
+    @Autowired
+    protected WebDriver webDriver;
+
+    @Value("${screenshot.path}")
+    private Path screenshotPath;
+
+    public void takeScreenShot() {
+        File screenshot = ((TakesScreenshot) this.webDriver).getScreenshotAs(OutputType.FILE);
+        try {
+            FileUtils.copyFile(screenshot, new File(
+                    screenshotPath.toString()  + "\\" + UUID.randomUUID() + ".png"));
+        } catch (IOException e) {
+            throw new RuntimeException(e);
+        }
+        System.out.println("Take screenshot");
+    }
+}

src/test/resources/application.properties

@@ -1,4 +1,5 @@
 app.url=http://eaapp.somee.com
 env=prod
 browser=chrome
+screenshot.path=target/screenshots
 #driver.options=