TestNG Listeners — Screenshots on Failure, Retry Analyzer

package com.testerrank.listeners;

import com.testerrank.base.BaseTest;
import com.testerrank.utils.ScreenshotUtils;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.openqa.selenium.WebDriver;
import org.testng.ITestContext;
import org.testng.ITestListener;
import org.testng.ITestResult;

public class TestListener implements ITestListener {

    private static final Logger logger =
        LogManager.getLogger(TestListener.class);

    @Override
    public void onTestStart(ITestResult result) {
        logger.info("========================================");
        logger.info("STARTING: " + getTestName(result));
        logger.info("========================================");
    }

    @Override
    public void onTestSuccess(ITestResult result) {
        long duration = result.getEndMillis()
            - result.getStartMillis();
        logger.info("PASSED: " + getTestName(result)
            + " (" + duration + "ms)");
    }

    @Override
    public void onTestFailure(ITestResult result) {
        logger.error("FAILED: " + getTestName(result));
        logger.error("Reason: "
            + result.getThrowable().getMessage());

        // Capture screenshot on failure
        Object testInstance = result.getInstance();
        if (testInstance instanceof BaseTest) {
            WebDriver driver =
                ((BaseTest) testInstance).getDriver();
            if (driver != null) {
                String path = ScreenshotUtils.capture(
                    driver, result.getName());
                logger.info("Screenshot: " + path);
            }
        }
    }

    @Override
    public void onTestSkipped(ITestResult result) {
        logger.warn("SKIPPED: " + getTestName(result));
        if (result.getThrowable() != null) {
            logger.warn("Skip reason: "
                + result.getThrowable().getMessage());
        }
    }

    @Override
    public void onStart(ITestContext context) {
        logger.info("===== Suite Starting: "
            + context.getName() + " =====");
    }

    @Override
    public void onFinish(ITestContext context) {
        int passed = context.getPassedTests()
            .getAllResults().size();
        int failed = context.getFailedTests()
            .getAllResults().size();
        int skipped = context.getSkippedTests()
            .getAllResults().size();

        logger.info("===== Suite Finished: "
            + context.getName() + " =====");
        logger.info("Results: " + passed + " passed, "
            + failed + " failed, " + skipped + " skipped");
    }

    private String getTestName(ITestResult result) {
        return result.getTestClass().getRealClass().getSimpleName()
            + "." + result.getName();
    }
}

package com.testerrank.utils;

import org.openqa.selenium.OutputType;
import org.openqa.selenium.TakesScreenshot;
import org.openqa.selenium.WebDriver;
import java.io.File;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.time.LocalDateTime;
import java.time.format.DateTimeFormatter;

public class ScreenshotUtils {

    private static final String DIR =
        ConfigReader.get("screenshot.dir", "screenshots");

    public static String capture(WebDriver driver,
                                  String testName) {
        try {
            Path dir = Paths.get(DIR);
            if (!Files.exists(dir)) {
                Files.createDirectories(dir);
            }

            String timestamp = LocalDateTime.now().format(
                DateTimeFormatter.ofPattern(
                    "yyyy-MM-dd_HH-mm-ss"));
            String fileName = testName + "_"
                + timestamp + ".png";
            Path filePath = dir.resolve(fileName);

            File srcFile = ((TakesScreenshot) driver)
                .getScreenshotAs(OutputType.FILE);
            Files.copy(srcFile.toPath(), filePath);

            return filePath.toString();
        } catch (IOException e) {
            System.err.println(
                "Screenshot failed: " + e.getMessage());
            return "";
        }
    }

    // Useful for Allure report attachments
    public static byte[] captureAsBytes(WebDriver driver) {
        return ((TakesScreenshot) driver)
            .getScreenshotAs(OutputType.BYTES);
    }
}

package com.testerrank.listeners;

import com.testerrank.utils.ConfigReader;
import org.testng.IRetryAnalyzer;
import org.testng.ITestResult;

public class RetryAnalyzer implements IRetryAnalyzer {

    private int retryCount = 0;

    @Override
    public boolean retry(ITestResult result) {
        int maxRetry = ConfigReader.getInt("retry.count", 1);
        if (retryCount < maxRetry) {
            retryCount++;
            System.out.println("Retrying test: "
                + result.getName()
                + " (attempt " + retryCount
                + "/" + maxRetry + ")");
            return true;  // Yes, retry this test
        }
        return false; // No more retries
    }
}

// Usage in test class:
// @Test(retryAnalyzer = RetryAnalyzer.class)
// public void testSomething() { ... }