当我的测试失败，在chrome浏览器关闭之前，如何截取chrome浏览器的屏幕(@After)

import org.openqa.selenium.OutputType;
import org.openqa.selenium.TakesScreenshot;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.chrome.ChromeDriver;
import org.testng.annotations.AfterMethod;
import org.testng.annotations.Test;

import java.io.File;
import java.io.IOException;

public class ChromeScreenshotExample {
    private WebDriver driver;

    @Test
    public void testMethod() {
        // 执行测试的代码
        // ...
        // 这里是你的测试代码
        // ...
        // 如果测试失败，则在此处进行屏幕截图
        takeScreenshot();
    }

    @AfterMethod
    public void teardown() {
        if (driver != null) {
            driver.quit();
        }
    }

    private void takeScreenshot() {
        File screenshotFile = ((TakesScreenshot) driver).getScreenshotAs(OutputType.FILE);
        // 将截图保存到指定路径
        try {
            // 这里假设你已经有一个保存截图的目录，比如"screenshots"
            String screenshotDirectory = "screenshots";
            String screenshotName = "failed_test.png";
            File destinationFile = new File(screenshotDirectory, screenshotName);
            org.apache.commons.io.FileUtils.copyFile(screenshotFile, destinationFile);
            System.out.println("Screenshot saved to: " + destinationFile.getAbsolutePath());
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}