如何使用IReporter在selenium中获取失败测试的详细信息？

import java.io.File;
import java.util.List;
import org.testng.IReporter;
import org.testng.ISuite;
import org.testng.ISuiteResult;
import org.testng.ITestContext;
import org.testng.ITestResult;
import org.testng.Reporter;
import org.testng.xml.XmlSuite;

public class CustomReporter implements IReporter {

    @Override
    public void generateReport(List<XmlSuite> xmlSuites, List<ISuite> suites, String outputDirectory) {
        for (ISuite suite : suites) {
            for (ISuiteResult suiteResult : suite.getResults().values()) {
                ITestContext testContext = suiteResult.getTestContext();

                // 获取所有测试结果
                for (ITestResult testResult : testContext.getFailedTests().getAllResults()) {
                    String testName = testResult.getName();
                    String testClassName = testResult.getTestClass().getName();
                    String screenshotFilePath = captureScreenshot();

                    // 在测试报告中记录失败测试的详细信息和截图
                    Reporter.log("Test: " + testName + " in class: " + testClassName + " failed.");
                    Reporter.log("Failure Reason: " + testResult.getThrowable().getMessage());
                    Reporter.log("Screenshot: " + screenshotFilePath);

                    // 可以将获取到的信息进行其他处理，如发送邮件等
                }
            }
        }
    }

    private String captureScreenshot() {
        // 使用Selenium WebDriver的相关方法来获取当前页面的截图
        // 返回截图文件的路径或者Base64编码字符串
        // 可以使用Tencent Cloud COS等存储服务来保存截图文件

        return "screenshot.png";
    }
}