Selenium 4 : How to capture screenshot

Question

I need an After method to capture a screenshot compatible with Selenium 4 Java

Answer 1

@After
public void tearDown(Scenario scenario) {
    if (scenario.isFailed()) {
        final byte[] screenshot = ((TakesScreenshot) Driver.get()).getScreenshotAs(OutputType.BYTES);
        scenario.attach(screenshot, "image/png", "screenshot");
    }


    Driver.closeDriver();

}