com.googlesites.CaptureFailure.java Source code

Introduction

Here is the source code for com.googlesites.CaptureFailure.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package com.googlesites;

/**
 *
 * @author Oana
 */
import java.io.File;
import java.io.IOException;
import java.text.DateFormat;
import java.text.SimpleDateFormat;
import java.util.Date;
import org.apache.commons.io.FileUtils;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.TakesScreenshot;
import org.testng.ITestResult;
import org.testng.TestListenerAdapter;

public class CaptureFailure extends TestListenerAdapter {

    @Override
    public void onTestFailure(ITestResult result) {

        File scrFile = ((TakesScreenshot) WebDriverInstance.getCurrentDriverInstance())
                .getScreenshotAs(OutputType.FILE);
        DateFormat dateFormat = new SimpleDateFormat("dd_MMM_yyyy__hh_mm_ssaa");
        String destDir = Config.getPropertyValue("failedTestsScreenshotsLocation");
        new File(destDir).mkdirs();
        String destFile = result.getMethod().getMethodName() + dateFormat.format(new Date()) + ".png";

        try {
            FileUtils.copyFile(scrFile, new File(destDir + destFile));
        } catch (IOException e) {
            System.out.println("Something wrong happend at the creation of the screenshot: " + e);
        }
        WebDriverInstance.getCurrentDriverInstance().close();
    }
}