Selenium2 Capture Sample

Capture.java

	private WebDriver driver;
	protected void capture(String path) throws IOException {
		WebDriver driver = this.driver;
		if (driver instanceof HtmlUnitDriver) {
			return;
		} else if (driver instanceof RemoteWebDriver) {
			driver = new Augmenter().augment(this.driver);
		} else {
			// 何もしない
		}
		File file = ((TakesScreenshot) driver).getScreenshotAs(OutputType.FILE);
		FileUtils.copyFile(file, new File(path));
	}

Test.java

	@Test
	public void test() throws IOException {
		driver.get("https://www.google.co.jp/");

		capture("./img/google1.jpg");

		driver.findElement(By.id("gbqfq")).clear();
		driver.findElement(By.id("gbqfq")).sendKeys("selenium");

		capture("./img/google2.jpg");

		driver.findElement(By.linkText("Selenium - Web Browser Automation")).click();

		capture("./img/google3.jpg");

	}