Trying to see if screenshots will show Travis errors

src/test/java/se/citerus/dddsample/acceptance/AbstractAcceptanceTest.java

@@ -1,7 +1,6 @@
 package se.citerus.dddsample.acceptance;
 
-import org.junit.After;
-import org.junit.Before;
+import org.junit.*;
 import org.openqa.selenium.WebDriver;
 import org.openqa.selenium.phantomjs.PhantomJSDriver;
 
@@ -11,6 +10,7 @@ public abstract class AbstractAcceptanceTest {
     @Before
     public void setup() {
         driver = new PhantomJSDriver();
+        screenshotTestRule.updateDriver(driver);
     }
 
     @After
@@ -18,4 +18,7 @@ public abstract class AbstractAcceptanceTest {
         driver.quit();
 
     }
+    @Rule
+    public ScreenshotTestRule screenshotTestRule = new ScreenshotTestRule();
+
 }

src/test/java/se/citerus/dddsample/acceptance/ScreenshotTestRule.java

@@ -0,0 +1,43 @@
+package se.citerus.dddsample.acceptance;
+
+import org.apache.commons.io.FileUtils;
+import org.junit.rules.MethodRule;
+import org.junit.runners.model.FrameworkMethod;
+import org.junit.runners.model.Statement;
+import org.openqa.selenium.OutputType;
+import org.openqa.selenium.TakesScreenshot;
+import org.openqa.selenium.WebDriver;
+
+import java.io.File;
+
+class ScreenshotTestRule implements MethodRule {
+    private WebDriver driver;
+
+    public Statement apply(final Statement statement, final FrameworkMethod frameworkMethod, final Object o) {
+        return new Statement() {
+            @Override
+            public void evaluate() throws Throwable {
+                try {
+                    statement.evaluate();
+                } catch (Throwable t) {
+                    captureScreenshot(frameworkMethod.getName());
+                    throw t;
+                }
+            }
+
+            public void captureScreenshot(String fileName) {
+                try {
+                    new File("target/surefire-reports/").mkdirs(); // Insure directory is there
+                    File scrFile = ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE);
+                    FileUtils.copyFile(scrFile, new File("target/surefire-reports/screenshot-" + fileName + ".png"));
+                } catch (Exception e) {
+                    System.err.println("ERROR: " + e);
+                }
+            }
+        };
+    }
+
+    public void updateDriver(WebDriver driver) {
+        this.driver = driver;
+    }
+}

src/test/java/se/citerus/dddsample/acceptance/pages/LaunchPage.java

@@ -2,9 +2,6 @@ package se.citerus.dddsample.acceptance.pages;
 
 import org.openqa.selenium.By;
 import org.openqa.selenium.WebDriver;
-import org.openqa.selenium.WebElement;
-import org.openqa.selenium.support.ui.ExpectedConditions;
-import org.openqa.selenium.support.ui.WebDriverWait;
 
 import static org.junit.Assert.assertEquals;
 
@@ -16,9 +13,6 @@ public class LaunchPage {
         this.driver = driver;
         this.baseUrl = baseUrl;
         driver.get(baseUrl + "dddsample");
-        WebDriverWait wait = new WebDriverWait(driver, 10);
-        wait.until(ExpectedConditions.titleContains("DDDSample"));
-
         assertEquals("DDDSample", driver.getTitle());
     }