How can I handle capturing a screenshot by using selenium when a test fails?

In the context of selenium, you can capture screenshots for debugging after the failure of the conducted test, by using the several steps which are given below:-

Capture screenshot during test failure

You can implement a mechanism for the task of capturing a screenshot when a test fails which was performed by you. Here is how you can do so:-

@Test

Public void yourTest() {
    Try {
        // Your test actions and assertions
        Assert.assertTrue(false, “Sample Assertion Failure”);
    } catch (AssertionError | Exception e) {
        captureScreenshot();
        // Handle the exception or rethrow it based on your needs
        Throw e;
    }
}
Private void captureScreenshot() {
    Try {
        TakesScreenshot ts = (TakesScreenshot) driver;
        File screenshotFile = ts.getScreenshotAs(OutputType.FILE);
        FileUtils.copyFile(screenshotFile, new File(“path/to/screenshots/failure_screenshot.png”));
    } catch (IOException e) {
        e.printStackTrace(); // Handle the exception appropriately
    }
}
Ensure screenshot path flexibility
Try to make sure that the screenshot path is configurable or even dynamic so that it can allow flexibility during the management of screenshots.
Integrate with test framework
If you are using a testing framework such as TestNG or Junit, then try to integrate the screenshot capture Within the right test listeners or even hooks:-
@AfterMethod
Public void tearDown(ITestResult result) {
    If (result.getStatus() == ITestResult.FAILURE) {
        captureScreenshot();
    }
}