TipsTricksTests.java

@Test
public void takeFullScreenshot_test() throws Exception {
    driver.navigate().to("http://automatetheplanet.com");
    takeFullScreenshot("testImage");
}

@Test
public void takeElementScreenshot_test() throws Exception {
    driver.navigate().to("http://automatetheplanet.com");
    var element = wait.until(ExpectedConditions.visibilityOfElementLocated(By.xpath("/html/body/div[1]/header/div/div[2]/div/div[2]/nav")));
    takeScreenshotOfElement(element, "testElementImage");
}