ScreenshotRule

ScreenshotRule.java

import org.junit.rules.TestRule;
import org.junit.runner.Description;
import org.junit.runners.model.Statement;

public class ScreenshotRule implements TestRule {

    public Statement apply(final Statement statement, final Description description) {
        return new Statement() {
            @Override
            public void evaluate() throws Throwable {
                try {
                    statement.evaluate();
                } catch (Exception e){
                    System.out.println("Took screenshot!");
                    throw e;
                }
            }
        };
    }
}