selenium实现屏幕截图

selenium实现屏幕截图

 System.out.println("strart selenium");
		System.setProperty("webdriver.firefox.bin", "F:/Mozilla Firefox/firefox.exe");//指定浏览器地址
		WebDriver driver = new FirefoxDriver();//声明一个web driver 并new一个FirefoxDriver
		driver.get("http://www.baidu.com/");//打开百度首页
		
		try {
			File srcFile = ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE);
			org.apache.commons.io.FileUtils.copyFile(srcFile, new File("d:\\screenshot.png"));  //使用copyFile()方法保存获取到的截图文件
		} catch (IOException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}