extentreports testng-adapter:如果测试失败，如何生成带有截图的HTML报告(cucumber，testng)

import com.aventstack.extentreports.ExtentReports;
import com.aventstack.extentreports.ExtentTest;
import com.aventstack.extentreports.Status;
import com.aventstack.extentreports.reporter.ExtentHtmlReporter;
import org.testng.ITestResult;
import org.testng.annotations.AfterMethod;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Listeners;
import org.testng.annotations.Test;

@Listeners(ExtentTestNGITestListener.class)
public class MyTest {
    private ExtentReports extent;
    private ExtentTest test;

    @BeforeMethod
    public void setup() {
        ExtentHtmlReporter htmlReporter = new ExtentHtmlReporter("test-output/extent.html");
        extent = new ExtentReports();
        extent.attachReporter(htmlReporter);
    }

    @Test
    public void myTestMethod() {
        test = extent.createTest("My Test Method");
        // 执行测试步骤
        // ...
        // 如果测试失败，添加截图到报告中
        if (testResult.getStatus() == ITestResult.FAILURE) {
            String screenshotPath = captureScreenshot(); // 截图方法
            test.log(Status.FAIL, "Test Failed");
            test.addScreenCaptureFromPath(screenshotPath);
        }
    }

    @AfterMethod
    public void tearDown() {
        extent.flush();
    }
}