Listeners — Screenshots, Retries, and Custom Reporting

import org.testng.ITestListener;
import org.testng.ITestResult;
import org.testng.ITestContext;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.TakesScreenshot;
import org.openqa.selenium.WebDriver;
import java.io.File;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.text.SimpleDateFormat;
import java.util.Date;

public class TestListener implements ITestListener {

    @Override
    public void onStart(ITestContext context) {
        System.out.println("\n=== Suite Started: "
            + context.getName() + " ===");
    }

    @Override
    public void onFinish(ITestContext context) {
        System.out.println("\n=== Suite Finished: "
            + context.getName() + " ===");
        System.out.println("Passed:  " + context.getPassedTests().size());
        System.out.println("Failed:  " + context.getFailedTests().size());
        System.out.println("Skipped: " + context.getSkippedTests().size());
    }

    @Override
    public void onTestStart(ITestResult result) {
        System.out.println("\nStarting: "
            + result.getMethod().getMethodName());
    }

    @Override
    public void onTestSuccess(ITestResult result) {
        System.out.println("PASSED: "
            + result.getMethod().getMethodName());
    }

    @Override
    public void onTestFailure(ITestResult result) {
        System.out.println("FAILED: "
            + result.getMethod().getMethodName());
        System.out.println("Reason: "
            + result.getThrowable().getMessage());

        // Take a screenshot on failure
        takeScreenshot(result);
    }

    @Override
    public void onTestSkipped(ITestResult result) {
        System.out.println("SKIPPED: "
            + result.getMethod().getMethodName());
    }

    private void takeScreenshot(ITestResult result) {
        try {
            Object testInstance = result.getInstance();
            WebDriver driver = ((BaseTest) testInstance).getDriver();

            File screenshot = ((TakesScreenshot) driver)
                .getScreenshotAs(OutputType.FILE);

            String timestamp = new SimpleDateFormat("yyyyMMdd_HHmmss")
                .format(new Date());
            String fileName = result.getMethod().getMethodName()
                + "_" + timestamp + ".png";

            Files.createDirectories(Paths.get("screenshots"));
            Files.copy(screenshot.toPath(),
                Paths.get("screenshots/" + fileName));

            System.out.println("Screenshot saved: screenshots/"
                + fileName);
        } catch (Exception e) {
            System.out.println("Could not take screenshot: "
                + e.getMessage());
        }
    }
}

import org.testng.IRetryAnalyzer;
import org.testng.ITestResult;

public class RetryAnalyzer implements IRetryAnalyzer {
    private int retryCount = 0;
    private static final int MAX_RETRIES = 2;

    @Override
    public boolean retry(ITestResult result) {
        if (retryCount < MAX_RETRIES) {
            retryCount++;
            System.out.println("Retrying "
                + result.getMethod().getMethodName()
                + " — attempt " + (retryCount + 1) + " of "
                + (MAX_RETRIES + 1));
            return true;   // true = retry the test
        }
        return false;      // false = give up, mark as failed
    }
}

// Apply to a specific test:
@Test(retryAnalyzer = RetryAnalyzer.class)
public void testFlakyDashboard() {
    // If this fails, it retries up to 2 more times
}

import org.testng.IAnnotationTransformer;
import org.testng.annotations.ITestAnnotation;
import java.lang.reflect.Constructor;
import java.lang.reflect.Method;

public class RetryTransformer implements IAnnotationTransformer {

    @Override
    public void transform(ITestAnnotation annotation,
                          Class testClass,
                          Constructor testConstructor,
                          Method testMethod) {
        annotation.setRetryAnalyzer(RetryAnalyzer.class);
    }
}

<suite name="Suite with Retry">
    <listeners>
        <listener class-name="com.autopractice.listeners.TestListener" />
        <listener class-name="com.autopractice.listeners.RetryTransformer" />
    </listeners>

    <test name="All Tests">
        <classes>
            <class name="com.autopractice.tests.BankingLoginTest" />
        </classes>
    </test>
</suite>