[唐胡璐]Selenium技巧- 抓图并保存到TestNG报告中

这里不讲解怎么在Eclipse安装配置TestNG，网上一搜一大把，大家自己去实践一下。

在这里主要说一下用Java来实现Selenium Webdriver的截图功能和把截图写到TestNG的报告中。

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29

//Capture screenshot public String captureScreenShot() {     String dir = "screenshot";     String date = new SimpleDateFormat("yyyyMMdd").format(new Date());     String time = new SimpleDateFormat("HHmmss").format(new Date());     String screenShotPath = dir + File.separator + date + File.separator + time + ".png";     try     {         File source = ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE);         FileUtils.copyFile(source, new File(screenShotPath));         screenShotPath = screenShotPath.substring(screenShotPath.indexOf("\\"));     }     catch(IOException e)     {         screenShotPath = "Failed to capture screenshot: " + e.getMessage();     }     return screenShotPath; } //Write to TestNG public void writeToTestNG(String proMessage) {     String png = captureScreenShot();     Reporter.log("[" + logTime + "] " + proMessage);     String log = new File("screenshot").getAbsolutePath();     Reporter.log("<br/><img src=\"" + log  + "/" + png + "\" />");  }

用下面的方法来调用：

1 2 3 4 5 6 7 8 9 10 11

@Test public void search() {     openURL();     BaiduSearch yy = new BaiduSearch(driver);     yy.searchFor("searchTest");        writeToTestNG("testing ");          driver.quit(); }

运行结果如下图所示：