Selenium遇到异常自动截图-自动化测试

　　最近要在框架中添加case失败时，要自动截图，今天做下总结。

　　1.只针对webdriver的异常截图，重写onException,该方法由于只针对webdriver抛的异常时才能截图，有一定的限制

　　public void onException(java.lang.Throwable throwable,

　　WebDriver driver){

　　Throwable cause = throwable.getCause();

　　/*

　　String cause = throwable.getClass().getName();

　　ScreenshotException ec = new ScreenshotException(cause);

　　*/

　　System.out.println(throwable.getClass().getName());

　　System.out.println("onException=" + cause);

　　Date currentTime = new Date();

　　SimpleDateFormat formatter = new SimpleDateFormat("yyyy-MM-dd-hh-mm-ss");

　　String dateString = formatter.format(currentTime);

　　File scrFile = ((TakesScreenshot) driver).getScreenshotAs(OutputType.FILE);

　　try {

　　File screenshot = new File("D:/ddd/"

　　+ dateString + ".png");

　　FileUtils.copyFile(scrFile,screenshot);

　　} catch (IOException e) {

　　e.printStackTrace();

　　}

　　}

　　测试类，要用EventFiringWebDriver ，并注册MyListen

　　public static void main(String args[]) {

　　String key = "webdriver.chrome.driver";

　　String value = "D:/BaiduYunDownload/selenium/chromedriver.exe";

　　System.setProperty(key, value);

　　WebDriver dr = new ChromeDriver();

　　EventFiringWebDriver event = new EventFiringWebDriver(dr);

　　MyListen ml = new MyListen();

　　event.register(ml);

　　dr = event;

　　dr.get("http://www.baidu.com");

　　dr.findElement(By.id("kw1")).sendKeys("test");

　　//System.out.println(5/0);

　　Assert.assertEquals("haha", event.findElement(By.id("su")).getAttribute("value"));

　　event.quit();

　　}

　　2.使用testNG的TestListenerAdapter

　　a.先建一个类继承TestListenerAdapter，并重写onTestFailure方法

　　package com.screenshot.exception;

　　import java.io.File;

　　import java.io.IOException;

　　import java.text.SimpleDateFormat;

　　import java.util.Date;

　　import org.apache.commons.io.FileUtils;

　　import org.apache.velocity.runtime.log.LogManager;

　　import org.openqa.selenium.OutputType;

　　import org.openqa.selenium.TakesScreenshot;

　　import org.openqa.selenium.WebDriver;

　　import org.testng.ITestResult;

　　import org.testng.TestListenerAdapter;

　　import com.screenshot.singleton.TestBase;

　　/**

　　* @author QiaoJiaofei

　　* @version 创建时间：2022年4月26日

　　* 类说明

　　*/

　　public class UseTestNg extends TestListenerAdapter{

　　@Override

　　public synchronized void onTestFailure(ITestResult result) {

　　Object currentClass = result.getInstance();

　　WebDriver webDriver = ((TestUsNg) currentClass).getDriver();

　　if (webDriver != null)

　　{

　　Date currentTime = new Date();

　　SimpleDateFormat formatter = new SimpleDateFormat("yyyy-MM-dd-hh-mm-ss");

　　String dateString = formatter.format(currentTime);

　　File scrFile = ((TakesScreenshot) webDriver).getScreenshotAs(OutputType.FILE);

　　try {

　　File screenshot = new File("D:/ddd/"

　　+ dateString + ".png");

　　FileUtils.copyFile(scrFile,screenshot);

　　} catch (IOException e) {

　　e.printStackTrace();

　　}

　　}

　　}

　　}

　　b.创建测试类，注意需要在测试类中写getDriver()方法

　　package com.screenshot.exception;

　　import org.openqa.selenium.WebDriver;

　　import org.openqa.selenium.chrome.ChromeDriver;

　　import org.testng.annotations.Test;

　　/**

　　* @author QiaoJiaofei

　　* @version 创建时间：2022年4月26日

　　* 类说明

　　*/

　　public class TestUsNg {

　　private WebDriver dr;

　　public WebDriver getDriver() {

　　return dr;

　　}

　　@Test

　　public void f() {

　　String key = "webdriver.chrome.driver";

　　String value = "D:/BaiduYunDownload/selenium/chromedriver.exe";

　　System.setProperty(key, value);

　　dr = new ChromeDriver();

　　System.out.println(5/0);

　　}

　　}

　　C.在testng的xml文件中添加监听

　　3.如何将生成的图片连接到reportNG中，将下面的代码放入上面相应重写的方法中，图片路径与上述代码中生成的图片结合一起。

　　String imgName = "";//图片路径

　　Reporter.log("Failed Screen Shot", true);

　　4.使用Robot主动截图，这种可以在自己想截图的时候调用该方法即可截当前界面

　　packagecom.screenshot.book;

　　importjava.awt.Rectangle;

　　importjava.awt.Robot;

　　importjava.awt.Toolkit;

　　importjava.awt.image.BufferedImage;

　　importjava.io.File;

　　importjavax.imageio.ImageIO;

　　importorg.testng.annotations.Test;

　　/**

　　*@authorQiaoJiaofei

　　*@version创建时间：2022年4月26日

　　*类说明

　　*/

　　publicclassTestRobot{

　　@Test

　　publicvoidtakeScreenShotMethod(){

　　try{

　　Thread.sleep(3000);

　　BufferedImageimage=newRobot().createScreenCapture(newRectangle(Toolkit.getDefaultToolkit().getScreenSize()));

　　ImageIO.write(image,"jpg",newFile("D:/ddd/screenshot.jpg"));

　　}

　　catch(Exceptione){

　　e.printStackTrace();

　　}

　　}

　　}

　　备注：

　　使用junit自动截图，可以使用Rule，由于我用的是testNG，所以没有调试junit的方法。详细参考：http://stackoverflow.com/questions/20995722/when-does-

　　onexception-get-triggered-in-webdrivereventlistener

　　@Rule

　　public TestRule testWatcher = new TestWatcher() {

　　@Override

　　public void succeeded(Description test){

　　for (LogEntry log : driver.manage().logs().get(LogType.DRIVER).getAll()) {

　　System.out.println("Level:" + log.getLevel().getName());

　　System.out.println("Message:" + log.getMessage());

　　System.out.println("Time:" + log.getTimestamp());

　　System.out.println("-----------------------------------------------");

　　}

　　System.out.println();

　　@Override

　　public void failed(Throwable t, Description test) {

　　String testName = test.getClassName();

　　String subTestName = test.getMethodName();

　　String screenShotName = String.format("%s\\%s", testName, screenShotName);

　　if (driver instanceof TakesScreenshot) {

　　File tempFile = ((TakesScreenshot) driver).getScreenshotAs(OutputType.FILE);

　　try {

　　System.out.println(">>>>>>>>>>LOGS: " + yourDirForImages + "\\" + screenShotName + ".png");

　　FileUtils.copyFile(tempFile, new File(String.format("%s.png", screenShotName)));

　　} catch (IOException e) {

　　e.printStackTrace();

　　}

　　}

　　以上内容为大家介绍了自动化测试中的Selenium遇到异常自动截图，本文由多测师亲自撰写，希望对大家有所帮助。了解更多自动化测试相关知识：https://www.aichudan.com/xwzx/