From 1d3c7600c8162a2985ddc81d2ff4f52321451afe Mon Sep 17 00:00:00 2001
From: skygo <skygo@unknown>
Date: Fri, 7 Jun 2013 00:27:25 +0000
Subject: [PATCH] add page source capture for fluent content may not be
 accurate

git-svn-id: https://svn.apache.org/repos/asf/archiva/trunk@1490482 13f79535-47bb-0310-9956-ffa450edef68
---
 .../archiva/web/test/WebDriverTest.java       | 30 ++++++++++++++++++-
 1 file changed, 29 insertions(+), 1 deletion(-)

diff --git a/archiva-modules/archiva-web/archiva-webapp-test/src/test/java/org/apache/archiva/web/test/WebDriverTest.java b/archiva-modules/archiva-web/archiva-webapp-test/src/test/java/org/apache/archiva/web/test/WebDriverTest.java
index 94c7f393b..f184dd0b9 100644
--- a/archiva-modules/archiva-web/archiva-webapp-test/src/test/java/org/apache/archiva/web/test/WebDriverTest.java
+++ b/archiva-modules/archiva-web/archiva-webapp-test/src/test/java/org/apache/archiva/web/test/WebDriverTest.java
@@ -33,7 +33,11 @@ import org.openqa.selenium.safari.SafariDriver;
 
 import java.io.File;
 import java.io.FileInputStream;
+import java.io.IOException;
 import java.util.Properties;
+import org.apache.commons.io.FileUtils;
+import org.fluentlenium.core.Fluent;
+import org.junit.Before;
 
 /**
  * @author Olivier Lamy
@@ -42,11 +46,35 @@ public class WebDriverTest
     extends FluentTest
 {
 
+    @Override
+    public Fluent takeScreenShot( String fileName )
+    {
+        try
+        {
+            // save html to have a minimum feedback if jenkins firefox not up
+            File fileNameHTML = new File( fileName + ".html" );
+            FileUtils.writeStringToFile( fileNameHTML, getDriver().getPageSource() );
+        }
+        catch ( IOException e )
+        {
+            System.out.print( e.getMessage() );
+            e.printStackTrace();
+        }
+        return super.takeScreenShot( fileName );
+    }
+    
+    @Before
+    public void init()
+    {
+        setSnapshotMode( Mode.TAKE_SNAPSHOT_ON_FAIL );
+        setSnapshotPath( new File( "target", "errorshtmlsnap" ).getAbsolutePath() );
+    }
+    
     @Test
     public void simpletest()
         throws Exception
     {
-
+        
         Properties tomcatPortProperties = new Properties();
         tomcatPortProperties.load(
             new FileInputStream( new File( System.getProperty( "tomcat.propertiesPortFilePath" ) ) ) );