Dev: Catch exception if no screenshot can be taken (tests)

LimeSurvey · Nov 23, 2017 · ac977eb · ac977eb
1 parent b3caa5c
commit ac977eb
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/tests/TestHelper.php b/tests/TestHelper.php
@@ -3,6 +3,7 @@
 namespace ls\tests;
 
 use PHPUnit\Framework\TestCase;
+use Facebook\WebDriver\Exception\NoSuchDriverException;
 
 class TestHelper extends TestCase
 {
@@ -300,10 +301,14 @@ public function takeScreenshot($webDriver, $name)
     {
         $tempFolder = \Yii::app()->getBasePath() .'/../tests/tmp';
         $folder     = $tempFolder.'/screenshots/';
-        $screenshot = $webDriver->takeScreenshot();
-        $filename   = $folder . $name . date('YmdHis') . '.png';
-        $result     = file_put_contents($filename, $screenshot);
-        $this->assertTrue($result > 0, 'Could not write screenshot to file ' . $filename);
+        try {
+            $screenshot = $webDriver->takeScreenshot();
+            $filename   = $folder . $name . date('YmdHis') . '.png';
+            $result     = file_put_contents($filename, $screenshot);
+            $this->assertTrue($result > 0, 'Could not write screenshot to file ' . $filename);
+        } catch (NoSuchDriverException $ex) {
+            // No driver.
+        }
     }
 
     /**