這篇文章主要介紹了Java selenium截圖操作的實現,文中通過示例代碼介紹的非常詳細,對大家的學習或者工作具有一定的參考學習價值,需要的朋友們下面隨着小編來一起學習學習吧
方法一:Selenium中截圖類TakeScreenshout,這個類主要是獲取瀏覽器窗體內的內容,不包括瀏覽器的菜單和桌面的任務欄區域,我們用百度首頁來截圖,看看截圖效果。
FileUtils.copyFile(srcFile, new File("屏幕截圖", time + ".png"));“屏幕截圖”是我們自己創建的文件夾用來存放截圖文件,此文件夾在project(工程)的更目錄
;
當然也是可以設置保存到其他目錄下:FileUtils.copyFile(srcFile, new File("D:\\資料圖片", time + ".png"));
示例代碼如下:
package com.sandy; import java.io.File; import java.text.SimpleDateFormat; import java.util.Calendar; import org.apache.commons.io.FileUtils; import org.openqa.selenium.OutputType; import org.openqa.selenium.TakesScreenshot; import org.openqa.selenium.WebDriver; import org.openqa.selenium.chrome.ChromeDriver; public class ScreenShot { private static WebDriver driver; public static void main(String[] args) throws Exception { System.setProperty("webdriver.chrome.driver", "E:\\eclipse_jar\\selenium_jar\\chromedriver.exe"); driver = new ChromeDriver(); driver.get("http://www.baidu.com"); driver.manage().window().maximize(); /** * 截屏操作 * 圖片已當前時間命名 */ SimpleDateFormat dateFormat = new SimpleDateFormat("yyyyMMddHHmmss"); //轉換時間格式 String time = dateFormat.format(Calendar.getInstance().getTime()); //獲取當前時間 File srcFile = ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE); //執行屏幕截取 FileUtils.copyFile(srcFile, new File("屏幕截圖", time + ".png")); //利用FileUtils工具類的copyFile()方法保存getScreenshotAs()返回的文件;"屏幕截圖"即時保存截圖的文件夾 Thread.sleep(2000); driver.quit(); } }
方法二:Robot截屏
示例代碼:(示例中的圖片是保存再該工程的根目錄下)
package com.sandy; import java.awt.Rectangle; import java.awt.Robot; import java.awt.Toolkit; import java.awt.image.BufferedImage; import java.io.File; import java.text.SimpleDateFormat; import java.util.Calendar; import javax.imageio.ImageIO; import org.apache.commons.io.FileUtils; import org.openqa.selenium.By; import org.openqa.selenium.OutputType; import org.openqa.selenium.Point; import org.openqa.selenium.TakesScreenshot; import org.openqa.selenium.WebDriver; import org.openqa.selenium.WebElement; import org.openqa.selenium.chrome.ChromeDriver; import org.openqa.selenium.internal.WrapsDriver; public class ScreenShot { private static WebDriver driver; public static void main(String[] args) throws Exception { System.setProperty("webdriver.chrome.driver", "E:\\eclipse_jar\\selenium_jar\\chromedriver.exe"); driver = new ChromeDriver(); driver.get("http://www.baidu.com"); driver.manage().window().maximize(); robotSnapshot(); Thread.sleep(2000); driver.quit(); } /** * 截屏方法二、Robot實現截屏 * @throws Exception */ public static void robotSnapshot() throws Exception { //調用截圖方法 BufferedImage img = new Robot().createScreenCapture(new Rectangle(Toolkit.getDefaultToolkit().getScreenSize())); ImageIO.write(img, "png", new File("robot_screen01.png")); }
方法三:在測試的過程中,有時候不需要截取整個屏幕,只需要截取某個元素(或者目標區域)的圖片
示例代碼:
package com.sandy; import java.awt.Rectangle; import java.awt.Robot; import java.awt.Toolkit; import java.awt.image.BufferedImage; import java.io.File; import java.text.SimpleDateFormat; import java.util.Calendar; import javax.imageio.ImageIO; import org.apache.commons.io.FileUtils; import org.openqa.selenium.By; import org.openqa.selenium.OutputType; import org.openqa.selenium.Point; import org.openqa.selenium.TakesScreenshot; import org.openqa.selenium.WebDriver; import org.openqa.selenium.WebElement; import org.openqa.selenium.chrome.ChromeDriver; import org.openqa.selenium.internal.WrapsDriver; public class ScreenShot { private static WebDriver driver; public static void main(String[] args) throws Exception { System.setProperty("webdriver.chrome.driver", "E:\\eclipse_jar\\selenium_jar\\chromedriver.exe"); driver = new ChromeDriver(); driver.get("http://www.baidu.com"); driver.manage().window().maximize(); WebElement element = driver.findElement(By.id("su")); elementSnapshot(element); //System.currentTimeMillis()、Calendar.getInstance().getTimeInMillis()獲取時間戳的方法 FileUtils.copyFile(elementSnapshot(element), new File("屏幕截圖", System.currentTimeMillis()+".png")); Thread.sleep(2000); driver.quit(); } /** * 部分截圖(元素截圖) * 有時候需要元素的截圖,不需要整個截圖 * @throws Exception */ public static File elementSnapshot(WebElement element) throws Exception { //創建全屏截圖 WrapsDriver wrapsDriver = (WrapsDriver)element; File screen = ((TakesScreenshot)wrapsDriver.getWrappedDriver()).getScreenshotAs(OutputType.FILE); BufferedImage image = ImageIO.read(screen); //獲取元素的高度、寬度 int width = element.getSize().getWidth(); int height = element.getSize().getHeight(); //創建一個矩形使用上面的高度,和寬度 Rectangle rect = new Rectangle(width, height); //元素座標 Point p = element.getLocation(); BufferedImage img = image.getSubimage(p.getX(), p.getY(), rect.width, rect.height); ImageIO.write(img, "png", screen); return screen; } }
以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支持神馬文庫。