package com.gloryroad.Demo;



 import java.io.File;



 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.firefox.FirefoxDriver;

 import org.testng.annotations.AfterMethod;

 import org.testng.annotations.BeforeMethod;

 import org.testng.annotations.Test;







 public class CaptureScreenInCurrentWindow {

String url="http://127.0.0.1:8020/HTMLDemo/HTMLPDir/Temp02/selection.html";

public WebDriver driver;

@BeforeMethod

public void setUp(){

driver=new FirefoxDriver();

driver.get(url);


}



@AfterMethod

public void tearDown(){

driver.close();

}


@Test

public void capture(){

File scrFile=((TakesScreenshot) driver).getScreenshotAs(OutputType.FILE);

try {

FileUtils.copyFile(scrFile, new File("test-output/1.png"));

} catch (Exception e) {

// TODO: handle exception

}



}




 }