第一步:在基礎類BaseTest中,寫一個「運行結果截圖」方法。java
代碼以下:測試
//運行結果截圖 public void screenshot(String fileName, WebDriver driver){ try{ File screenFile = ((TakesScreenshot) driver) .getScreenshotAs(OutputType.FILE); FileUtils.copyFile(screenFile, new File("screen/" + fileName)); } catch(Exception e){ e.printStackTrace(); } }
第二步:在測試case中,繼承BaseTest基礎類(extends BaseTest)。this
在要截圖的地方調用「運行結果截圖」方法。code
代碼以下:繼承
this.screenshot("RoleCreate添加成功.jpg",driver);
第三步:截圖後的文件是存放在工程目錄下的「screen」文件夾中。get