added code to take screenshot after navigating to registration page

Author: mfaisalkhatri

src/test/java/io/github/mfaisalkhatri/pages/RegistrationPage.java

@@ -1,10 +1,6 @@
 package io.github.mfaisalkhatri.pages;
 
-import java.text.DateFormat;
 import java.time.Duration;
-import java.time.format.DateTimeFormatter;
-import java.util.Date;
-import java.util.Locale;
 
 import io.github.mfaisalkhatri.data.RegistrationData;
 import org.openqa.selenium.By;

src/test/java/io/github/mfaisalkhatri/test/JsonDataProviderTest.java

@@ -2,13 +2,21 @@
 
 import static org.testng.Assert.assertEquals;
 
+import java.io.File;
+import java.io.IOException;
+import java.nio.file.Files;
+import java.nio.file.Paths;
+import java.text.SimpleDateFormat;
+import java.util.Date;
 import java.util.Iterator;
 
 import io.github.mfaisalkhatri.data.RegistrationData;
 import io.github.mfaisalkhatri.data.RegistrationDataBuilder;
 import io.github.mfaisalkhatri.pages.HomePage;
 import io.github.mfaisalkhatri.pages.LoginPage;
 import io.github.mfaisalkhatri.pages.RegistrationPage;
+import org.openqa.selenium.OutputType;
+import org.openqa.selenium.TakesScreenshot;
 import org.testng.annotations.DataProvider;
 import org.testng.annotations.Test;
 
@@ -29,6 +37,17 @@ public void testRegistrationPasswordAlert (final RegistrationData registrationDa
         LoginPage loginPage = homePage.navigateToLoginPage ();
         RegistrationPage registrationPage = loginPage.navigateToRegistrationPage ();
 
+        String timestamp = new SimpleDateFormat ("yyyyMMdd_HHmmss_SSS").format (new Date ());
+
+        File screenshot = ((TakesScreenshot) driver).getScreenshotAs (OutputType.FILE);
+        String filename = timestamp + ".png";
+        try {
+            Files.createDirectories (Paths.get ("screenshots"));
+            Files.copy (screenshot.toPath (), Paths.get ("screenshots", filename));
+        } catch (IOException e) {
+            throw new RuntimeException (e);
+        }
+
         assertEquals (registrationPage.pageHeader (), "Customer registration");
 
         registrationPage.fillRegistrationForm (registrationData);

src/test/java/io/github/mfaisalkhatri/utils/ScreenshotListener.java

@@ -20,15 +20,14 @@ public void onTestFailure (final ITestResult result) {
         WebDriver driver = ((WebDriverProvider) testClass).getDriver ();
         String timestamp = new SimpleDateFormat ("yyyyMMdd_HHmmss_SSS").format (new Date ());
 
-
         if (driver != null) {
             String testName = result.getMethod ()
                 .getMethodName ();
             File screenshot = ((TakesScreenshot) driver).getScreenshotAs (OutputType.FILE);
             String filename = timestamp + ".png";
             try {
                 Files.createDirectories (Paths.get ("screenshots"));
-                Files.copy (screenshot.toPath (), Paths.get ("screenshots", testName+"_"+filename));
+                Files.copy (screenshot.toPath (), Paths.get ("screenshots", testName + "_" + filename));
             } catch (IOException e) {
                 throw new RuntimeException (e);
             }