Added webdriver listener

Author: edinc

src/test/java/selenium/listeners/WebDriverListener.java

@@ -0,0 +1,82 @@
+package selenium.listeners;
+
+import driver.Driver;
+import driver.DriverManager;
+import io.qameta.allure.Attachment;
+import org.apache.commons.io.FileUtils;
+import org.apache.commons.io.IOUtils;
+import org.openqa.selenium.OutputType;
+import org.openqa.selenium.TakesScreenshot;
+import org.openqa.selenium.WebDriver;
+import org.testng.IInvokedMethod;
+import org.testng.IInvokedMethodListener;
+import org.testng.ITestResult;
+
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.IOException;
+import java.io.InputStream;
+import java.net.MalformedURLException;
+import java.text.SimpleDateFormat;
+import java.util.Date;
+
+public class WebDriverListener implements IInvokedMethodListener {
+
+    @Override
+    public void beforeInvocation(IInvokedMethod method, ITestResult testResult) {
+
+        if(method.isTestMethod()) {
+
+            String browserName = ((Base)method.getTestMethod().getInstance()).getBrowser();
+            try {
+                WebDriver driver = DriverManager.createInstance(browserName, ((Base)method.getTestMethod().getInstance()).getAppUrl());
+                System.out.println("Initializing webdriver session --> Thread ID: " + Thread.currentThread().getId());
+                System.out.println("Running test --> " + method.getTestMethod().getMethodName());
+                Driver.setWebDriver(driver);
+            } catch (MalformedURLException e) {
+                e.printStackTrace();
+            }
+        }
+    }
+
+    @Override
+    public void afterInvocation(IInvokedMethod method, ITestResult testResult) {
+
+        if(method.isTestMethod()) {
+
+            WebDriver driver = Driver.getDriver();
+            if(driver != null) {
+
+                try {
+                    takeScreenshotOnFailure(testResult);
+                } catch (IOException e) {
+                    e.printStackTrace();
+                }
+                System.out.println("Closing webdriver session: " + Thread.currentThread().getId());
+                driver.quit();
+            }
+        }
+    }
+
+    public void takeScreenshotOnFailure(ITestResult testResult) throws IOException {
+
+        if(testResult.getStatus() == ITestResult.FAILURE) {
+
+            File screenShot = ((TakesScreenshot) Driver.getDriver()).getScreenshotAs(OutputType.FILE);
+            File destination = new File("target/failure-screenshots/" + testResult.getName() + "-"
+                    + new SimpleDateFormat("dd-MM-yyyy HH-mm-ss").format(new Date()) + ".png");
+            FileUtils.copyFile(screenShot, destination);
+
+            InputStream screenShotStream = new FileInputStream(destination);
+            byte[] screen = IOUtils.toByteArray(screenShotStream);
+
+            saveScreenshot(screen);
+        }
+    }
+
+    @Attachment(value = "Screenshot of the failure", type = "image/png")
+    public byte[] saveScreenshot(byte[] screenShot) {
+
+        return screenShot;
+    }
+}

testng.xml

@@ -1,6 +1,10 @@
 <!DOCTYPE suite SYSTEM "http://beust.com/testng/testng-1.0.dtd" >
 <suite name="Java Selenium Framework" parallel="methods" verbose="1" thread-count="5">
 
+    <listeners>
+        <listener class-name="selenium.listeners.WebDriverListener" />
+    </listeners>
+
     <test name="all-tests">
         <packages>
             <package name="selenium.tests.*"/>