How to Take a Screenshot in Selenium WebDriver Using Java?

package io.learn;

import org.openqa.selenium.WebDriver;
import org.openqa.selenium.chrome.ChromeDriver;
import org.testng.annotations.AfterMethod;
import org.testng.annotations.BeforeMethod;

public class BaseTest {

    protected WebDriver driver;

    // Set up the ChromeDriver
    @BeforeMethod
    public void setup() {
        // Set the path to your chromedriver executable
        System.setProperty("webdriver.chrome.driver", "C:\\Users\\change the path for Chrome-Driver\\chromedriver.exe");

        // Initialize the ChromeDriver
        driver = new ChromeDriver();
    }

    // Close the browser after each test
    @AfterMethod
    public void teardown() {
        if (driver != null) {
            driver.quit();
        }
    }
}

package io.learn.screenshots;

import io.learn.BaseTest;
import org.openqa.selenium.By;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.TakesScreenshot;
import org.openqa.selenium.WebElement;
import org.testng.annotations.Test;

import java.io.File;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;

import static java.nio.file.StandardCopyOption.REPLACE_EXISTING;
import static org.assertj.core.api.Assertions.assertThat;

public class BasicScreenshotTest extends BaseTest {

    // Test for capturing a full page screenshot
    @Test
    void testCaptureScreenshot() throws IOException {
        // Open the page
        driver.get("https://bonigarcia.dev/selenium-webdriver-java/");

        // Capture screenshot using WebDriver's TakesScreenshot interface
        TakesScreenshot ts = (TakesScreenshot) driver;
        File screenshot = ts.getScreenshotAs(OutputType.FILE);  // Capture the screenshot as a file
        System.out.println("Screenshot captured at: " + screenshot.getAbsolutePath());

        // Define the destination path for saving the screenshot
        Path destination = Paths.get("C:\\path to save the screenshot\\bonigarcia-screenshot.png");

        // Move the screenshot to the desired location
        Files.move(screenshot.toPath(), destination, REPLACE_EXISTING);  // Move the screenshot file to the destination
        System.out.println("Screenshot saved at: " + destination);

        // Assert that the screenshot file exists at the destination
        assertThat(destination).exists();
    }

    // Test for capturing screenshot of a specific WebElement
    @Test
    void testCaptureWebElementScreenshot() throws IOException {
        // Open the page with a specific element to capture
        driver.get("https://bonigarcia.dev/selenium-webdriver-java/");

        // Find a specific web element (in this case, a header on the page)
        WebElement header = driver.findElement(By.tagName("h1"));

        // Capture screenshot of the specific web element
        File screenshot = header.getScreenshotAs(OutputType.FILE);  // Capture the screenshot of the header element
        System.out.println("WebElement screenshot captured at: " + screenshot.getAbsolutePath());

        // Define the destination path for saving the screenshot
        Path destination = Paths.get("C:\\path to save the screenshot\\bonigarcia-web-element-screenshot.png");

        // Move the screenshot to the desired location
        Files.move(screenshot.toPath(), destination, REPLACE_EXISTING);  // Move the screenshot file to the destination
        System.out.println("WebElement screenshot saved at: " + destination);

        // Assert that the screenshot file exists at the destination
        assertThat(destination).exists();
    }
}