added background color param to screenshot, no more black background …

…on screenshots
giorgiosironi · Jun 26, 2014 · 07e50f7 · 07e50f7
1 parent a54d922
commit 07e50f7
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/PHPUnit/Extensions/SeleniumTestCase.php b/PHPUnit/Extensions/SeleniumTestCase.php
@@ -551,6 +551,11 @@ abstract class PHPUnit_Extensions_SeleniumTestCase extends PHPUnit_Framework_Tes
      */
     protected $screenshotUrl = '';
 
+    /**
+     * @var    string
+     */
+    protected $screenshotBgColor = '';
+
     /**
      * @var    integer  the number of seconds to wait before declaring
      *                  the Selenium server not reachable
@@ -1187,8 +1192,12 @@ protected function takeScreenshot()
         if (!empty($this->screenshotPath) &&
             !empty($this->screenshotUrl)) {
             $filename = $this->getScreenshotPath() . $this->testId . '.png';
+            $kargs = '';
+            if ($this->screenshotBgColor!='') {
+                $kargs = 'background=' . $this->screenshotBgColor;
+            }
 
-            $this->drivers[0]->captureEntirePageScreenshot($filename);
+            $this->drivers[0]->captureEntirePageScreenshot($filename, $kargs);
 
             return 'Screenshot: ' . $this->screenshotUrl . '/' .
                    $this->testId . ".png\n";