TestNG失败用例自动截图

参考:https://blog.csdn.net/wangxin1982314/article/details/50247245

1. 首先写一个截屏方法　

public class ScreenShotOnFailure {
    public static final String SCREEN_SHOT_PATH = "test-output/screen-shot";
    public static String SCREEN_SHOT_NAME = null;
    
    public static void takeScreenShot() throws IOException {
        File screenshotDir = new File(SCREEN_SHOT_PATH);
        if(!screenshotDir.exists()) {
            screenshotDir.mkdirs();
        }
        
        SimpleDateFormat smf = new SimpleDateFormat("yyyyMMddHHmmss");        
        SCREEN_SHOT_NAME = String.valueOf(smf.format(new Date())) + ".jpg";
        FileUtils.copyFile(((TakesScreenshot)browser.getWebDriver()).getScreenshotAs(OutputType.FILE),
                new File(SCREEN_SHOT_PATH + "/"+ SCREEN_SHOT_NAME));    
    }
}

2. 新建一个监听类,重写onTestFailure方法　　

import org.testng.ITestContext;
import org.testng.ITestListener;
import org.testng.ITestResult;
import static com.crewbudget.bean.ScreenShotOnFailure.SCREEN_SHOT_NAME;
import static com.crewbudget.bean.ScreenShotOnFailure.SCREEN_SHOT_PATH;

public class TestngRetryListener implements ITestListener{
    public void onTestFailure(ITestResult result) {
        try {
            ScreenShotOnFailure.takeScreenShot();
            System.out.println(result.getMethod().getMethodName()+"failed, the screenshot saved in "
                    + SCREEN_SHOT_PATH +" screenshot name: "
                    + SCREEN_SHOT_NAME);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    @Override
    public void onTestStart(ITestResult result) {
        // TODO Auto-generated method stub
        
    }

    @Override
    public void onTestSuccess(ITestResult result) {
        // TODO Auto-generated method stub
        
    }

    @Override
    public void onTestSkipped(ITestResult result) {
        // TODO Auto-generated method stub
        
    }

    @Override
    public void onTestFailedButWithinSuccessPercentage(ITestResult result) {
        // TODO Auto-generated method stub
        
    }

    @Override
    public void onStart(ITestContext context) {
        // TODO Auto-generated method stub
        
    }

    @Override
    public void onFinish(ITestContext context) {
        // TODO Auto-generated method stub
        
    }
}

3. 在测试脚本中添加监听