如何使用testng listener在每次验证/断言通过和失败时触发屏幕截图

import org.testng.ITestListener;
import org.testng.ITestResult;
import org.testng.Reporter;

public class ScreenshotListener implements ITestListener {
    
    @Override
    public void onTestSuccess(ITestResult result) {
        // 在验证/断言通过时触发屏幕截图
        captureScreenshot(result.getName());
    }
    
    @Override
    public void onTestFailure(ITestResult result) {
        // 在验证/断言失败时触发屏幕截图
        captureScreenshot(result.getName());
    }
    
    private void captureScreenshot(String testName) {
        // 实现屏幕截图的逻辑
        // 这里可以使用各种截图工具或库来进行屏幕截图操作
        // 例如，可以使用Selenium WebDriver的getScreenshotAs方法来进行截图
        // 然后将截图保存到指定的目录中，可以使用文件名中包含测试名称来区分不同的截图
    }
}

import org.testng.annotations.Listeners;
import org.testng.annotations.Test;

@Listeners(ScreenshotListener.class)
public class MyTest {
    
    @Test
    public void myTestMethod() {
        // 测试方法的代码
        // 在验证/断言时，如果通过或失败，监听器会触发屏幕截图
    }
}