9-利用监听类设置登录失败自动截图

1、引入commons-io jar包

<!-- https://mvnrepository.com/artifact/commons-io/commons-io -->
    <dependency>
      <groupId>commons-io</groupId>
      <artifactId>commons-io</artifactId>
      <version>2.4</version>
    </dependency>

2、功能实现–截图命名封装

用到getScreenshotAs（）

	/**
     * 截图
     */
    public void takeScreenShot() throws IOException {
        //获取当前时间
        long date = System.currentTimeMillis();
        //valueOf将非字符串转换为字符串
        String phoname = String.valueOf(date);
        //获取当前路径
        String currpath = System.getProperty("user.dir") + "/ourprint";
        phoname = phoname + ".png";
        //拼接最终截图路径
        String decpath = currpath + "/" + phoname;
        //截图并存入内存当中
        File screen = ((TakesScreenshot) webDriver).getScreenshotAs(OutputType.FILE);
        //将截图从内存保存到上面的目的路径
        FileUtils.copyFile(screen, new File(decpath));
    }

想在哪截图，就在哪直接调用takeScreenShot()即可

3、改进功能–登录失败自动截图

思路：要做到登录失败自动截图，需要把testcase加入到监听事件，当监听到登录失败后自动截图

3.1、创建监听类

继承TestListenerAdapter这个类，然后对其中的方法进行重写

/**
 * 监听类实现
 */
public class TestListenerScreenShot extends TestListenerAdapter {

    private  static WebDriver driver;
    public static void setDriver(WebDriver driver){
        TestListenerScreenShot.driver = driver;
    }
    //用例执行结束后，用例执行成功时调用
    @Override
    public void onTestSuccess(ITestResult iTestResult) {
        super.onTestSuccess(iTestResult);
    }

    //用例执行结束后，用例执行失败时调用
    @Override
    public void onTestFailure(ITestResult iTestResult) {
        super.onTestFailure(iTestResult);
        //获取当前时间
        long date = System.currentTimeMillis();
        //valueoOf将非字符串转换为字符串
        String phoname = String.valueOf(date);
        //获取当前路径
        String currpath = System.getProperty("user.dir") + "/ourprint";
        phoname = phoname + ".png";
        //拼接最终截图路径
        String decpath = currpath + "/" + phoname;
        //截图并存入内存当中
        File screen = ((TakesScreenshot) driver).getScreenshotAs(OutputType.FILE);
        //将截图从内存保存到上面的目的路径
        try {
            FileUtils.copyFile(screen, new File(decpath));
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    //用例执行结束后，用例执行skip时调用
    @Override
    public void onTestSkipped(ITestResult iTestResult) {
        super.onTestSkipped(iTestResult);
    }

    //在所有测试运行之后调用，并且所有的配置方法都被调用
    @Override
    public void onFinish(ITestContext iTestContext) {
        super.onFinish(iTestContext);
    }
    //每次调用测试@Test之前调用
    @Override
    public void onTestStart(ITestResult iTestResult) {
        super.onTestStart(iTestResult);
    }
}

3.2、测试代码设置监听

加入注释@Listeners({TestListenerScreenShot.class}),并且在进行初始化driver的时候来设置driver即可。
例如：TestListenerScreenShot.setDriver(webDriver);