selenium remotewebdriver screenshot

selenium_remotewebdriver_screenshot

import org.openqa.selenium.TakesScreenshot;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.remote.Augmenter;
import org.openqa.selenium.remote.DesiredCapabilities;
import org.openqa.selenium.remote.RemoteWebDriver;

public class NavigateToAUrl {
	public static void main(String[] args) {
		DesiredCapabilities capabilities = DesiredCapabilities.firefox();

		RemoteWebDriver remoteWD = null;
		try {
			remoteWD = new RemoteWebDriver(new URL("http://192.168.137.225:4444/wd/hub"), capabilities);
			remoteWD.get("file://C:/selenium/Chapter 2/HTML/Selectable.html");

			Thread.sleep(1000);

	        WebDriver augmentedDriver = new Augmenter().augment(remoteWD);
	        File screenshot = ((TakesScreenshot)augmentedDriver)
	                            .getScreenshotAs(OutputType.FILE);
	        FileUtils.copyFile(screenshot, new File("C:/work/test.png"));
	        
	        /*
	        File screenshot = (File) ((TakesScreenshot)remoteWD)
					.getScreenshotAs(OutputType.FILE);
		FileUtils.copyFile(screenshot, new File("C:/work/test.png"));
		*/
			Thread.sleep(2000);

			remoteWD.quit();

		} catch (InterruptedException | IOException e) {
			e.printStackTrace();
		}

	}
}