ScreenShotExample.java

import org.apache.commons.io.FileUtils;
import org.openqa.selenium.By;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.chrome.ChromeDriver;
import org.openqa.selenium.firefox.FirefoxDriver;

import java.io.File;
import java.io.IOException;
public class ScreenShotExamples {

    final static String PROJECT_PATH = System.getProperty("user.dir");

    public static void main(String[] args) throws IOException {
        WebDriver driver = null;

        System.setProperty("webdriver.chrome.driver", PROJECT_PATH+ "/src/main/resources/chromedriver");
        driver = new ChromeDriver();
        driver.manage().window().fullscreen();
        driver.get("https://www.google.com");

        // take screenshot of an element
        WebElement logo= driver.findElement(By.id("hplogo"));
        File file=logo.getScreenshotAs(OutputType.FILE);
        File destFile =new File("logo.png");
        FileUtils.copyFile(file,destFile);


        System.setProperty("webdriver.gecko.driver", PROJECT_PATH+ "/src/main/resources/geckodriver");
        FirefoxDriver firefoxDriver = new FirefoxDriver();
        firefoxDriver.get("https://www.youtube.com");

        // take full page screenshot - available for firefox driver
        File src = firefoxDriver.getFullPageScreenshotAs(OutputType.FILE);
        File destFile1 =new File("youtubeLandingPage.png");
        FileUtils.copyFile(src,destFile1);

    }
}