WebDriver测试失败后自动获取截图

最新推荐文章于 2022-11-15 10:03:53 发布

weixin_33895516

最新推荐文章于 2022-11-15 10:03:53 发布

阅读量378

点赞数

文章标签：测试 python ui

UI自动化测试其实并不是那么稳定，可能是因为UI元素的改动，也可能是因为网络的不稳定，在测试失败的时候，WebDriver通常会抛出一些异常；通过异常信息通常都能知道大概是哪里出错了，但是如果能加上截屏，那就更加好了。尤其是用Remote WebDriver运行测试，所有测试都是通过Selenium Grid分发到各个节点来运行，不同节点的配置还有可能不是完全一样。

如果是使用RemoteWebDriver的话，它提供了一个很好的功能，就是会把运行测试发生异常时候的截图也放到异常里面，具体可以参考RemoteWebDriver的简介。代码很简单：

public String extractScreenShot(WebDriverException e) {     Throwable cause = e.getCause();     if (cause instanceof ScreenshotException) {         return ((ScreenshotException) cause).getBase64EncodedScreenshot();     }     return null; }

怎么样才能在异常发生的时候自动把异常抓住呢？简单来说就是要对写测试代码的人是透明了，在写测试代码的时候不需要特别去处理异常。这里需要实现 WebDriverEventListener接口，然后把RemoteWebDriver对象和实现WebDriverEventListener接口的对象包到一起，实例化一个EventFiringWebDriver对象。之后的事情就跟用一个普通的RemoteWebDriver对象没有任何区别。

实现WebDriverEventListener接口的一个例子：

public class MyEventListener implements WebDriverEventListener {     public void onException(Throwable ex, WebDriver arg1) {         String filename = generateRandomFilename(ex);         try {             byte[] btDataFile = Base64.decodeBase64(extractScreenShot(ex).getBytes());             File of = new File(filename);             FileOutputStream osf = new FileOutputStream(of);             osf.write(btDataFile);             osf.flush();             osf.close();         } catch (IOException e) {             e.printStackTrace();         }     }       private String generateRandomFilename(Throwable ex) {         Calendar c = Calendar.getInstance();         String filename = ex.getMessage();         int i = filename.indexOf('\n');         filename = filename.substring(0, i).replaceAll("\\s", "_")                     .replaceAll(":", "")                     + ".png";                     filename = "" + c.get(Calendar.YEAR) + "-" + c.get(Calendar.MONTH)                     + "-" + c.get(Calendar.DAY_OF_MONTH) + "-"                     + c.get(Calendar.HOUR_OF_DAY) + "-" + c.get(Calendar.MINUTE)                     + "-" + c.get(Calendar.SECOND) + "-" + filename;         return filename;     }       private String extractScreenShot(Throwable ex) {         Throwable cause = ex.getCause();         if (cause instanceof ScreenshotException) {             return ((ScreenshotException) cause).getBase64EncodedScreenshot();         }         return null;     }       @Override     public void afterChangeValueOf(WebElement arg0, WebDriver arg1) {     // TODO Auto-generated method stub       } }

实例化一个EventFiringWebDriver对象：

@Test public void setup(){     String remote_driver_url = "http://localhost:4444/wd/hub";     DesiredCapabilities capability = null;     capability = DesiredCapabilities.firefox();     WebDriverEventListener eventListener = new MyEventListener();     WebDriver driver = new EventFiringWebDriver(new RemoteWebDriver(new URL(                     remote_driver_url), capability)).register(eventListener); }