如何使用AbstractTestNGCucumberTests将屏幕截图附加到步骤？

import io.cucumber.testng.AbstractTestNGCucumberTests;
import io.cucumber.testng.CucumberOptions;
import io.cucumber.java.AfterStep;
import io.cucumber.java.Scenario;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.TakesScreenshot;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.chrome.ChromeDriver;

@CucumberOptions(features = "src/test/resources/features")
public class MyTest extends AbstractTestNGCucumberTests {
    private WebDriver driver;

    @BeforeClass
    public void setUp() {
        // 初始化WebDriver，例如ChromeDriver
        System.setProperty("webdriver.chrome.driver", "path/to/chromedriver");
        driver = new ChromeDriver();
    }

    @AfterStep
    public void attachScreenshot(Scenario scenario) {
        // 在每个步骤之后进行截图并附加到步骤中
        if (scenario.isFailed()) {
            byte[] screenshot = ((TakesScreenshot) driver).getScreenshotAs(OutputType.BYTES);
            scenario.attach(screenshot, "image/png", "Screenshot");
        }
    }

    @AfterClass
    public void tearDown() {
        // 关闭WebDriver
        driver.quit();
    }
}