testng対失敗時スクリーンショット処理

3061 ワード

1.スクリーンショットクラス:
public class ScreenShot {
	public WebDriver driver;

	public ScreenShot(WebDriver driver) {
		this.driver = driver;
	}

	private void takeScreenshot(String screenPath) {
		try {
			File scrFile = ((TakesScreenshot) driver)
					.getScreenshotAs(OutputType.FILE);
			FileUtils.copyFile(scrFile, new File(screenPath));
		} catch (IOException e) {
			System.out.println("Screen shot error: " + screenPath);
		}
	}

	public void takeScreenshot() {
		String screenName = String.valueOf(new Date().getTime()) + ".jpg";
		File dir = new File("test-output/snapshot");
		if (!dir.exists())
			dir.mkdirs();
		String screenPath = dir.getAbsolutePath() + "/" + screenName;
		this.takeScreenshot(screenPath);		
	}
}

 2.testngのリスナーを使用して、エラーのスクリーンショットをリスニングできます.
public class DotTestListener extends TestListenerAdapter {

	@Override
	public void onTestFailure(ITestResult tr) {		
		
	}

}

 3.つまり、onTestFailureメソッドでScreenShotクラスのtakeScreenshotメソッドを呼び出すだけでいいのですが、ScreenShotクラスにはdriverが入る必要があることに気づきました.
今問題が来て、driverの処理に対して、様々で、あるものは単一のサブモードを使って、つまりdriverを1つのグローバルな静的変数として、どこでも使えるので、ScreenShotクラスではdriverオブジェクトにアクセスすることができますが、このようにして1つの問題があります.つまり、全局に1つのdriverしかありません.マルチスレッドを実行したい場合、複数のdriverインスタンスを起動したい場合、このような方法ではできません.次に、各クラスまたは各テストメソッドが、新しいdriverオブジェクトを起動することで、driverオブジェクトはグローバルではなく、クラスオブジェクトのプロパティです.たとえば、次のようにします.
public class TestBase {
	
	public WebDriver driver;	
	
	public WebDriver getDriver() {
		return driver;
	}

	@BeforeClass
	public void setUp(){
		driver = new FirefoxDriver();
		driver.manage().window().maximize();
		driver.navigate().to("http://www.baidu.com");		
	}
	
	@AfterClass
	public void tearDown(){
		driver.close();
		driver.quit();
	}
}

 
public class Test10 extends TestBase{	
	
	@Test
	public void testInput(){
		System.out.println(5/0);
	}	
	
}

では、このクラスのオブジェクトのdriver属性をonTestFailureメソッドにどのように伝えますか?
public class DotTestListener extends TestListenerAdapter {

	@Override
	public void onTestFailure(ITestResult tr) {		
		try {
			TestBase tb = (TestBase) tr.getInstance();
			WebDriver driver = tb.getDriver();			
			System.out.println(driver.getTitle());
		} catch (SecurityException e) {
			e.printStackTrace();
		} catch (IllegalArgumentException e) {			
			e.printStackTrace();
		}
	}

}

最後にリスニングを追加すればいいです.
@Listeners({ DotTestListener.class })
public class Test10 extends TestBase{	
	
	@Test
	public void testInput(){
		System.out.println(5/0);
	}	
	
}