From 87cbaf70aff9ea19b678c80196db02337ddbfcf0 Mon Sep 17 00:00:00 2001
From: Youri Westerman <tetracon@gmail.com>
Date: Sat, 6 Jun 2020 01:52:54 +0200
Subject: [PATCH] Supporting all drivers that can make screenshots

---
 src/Listener/FailedStepListener.php | 9 ++-------
 1 file changed, 2 insertions(+), 7 deletions(-)

diff --git a/src/Listener/FailedStepListener.php b/src/Listener/FailedStepListener.php
index 42a9fe7..0cdbb57 100644
--- a/src/Listener/FailedStepListener.php
+++ b/src/Listener/FailedStepListener.php
@@ -13,9 +13,8 @@
 
 use Behat\Behat\EventDispatcher\Event\AfterStepTested;
 use Behat\Behat\EventDispatcher\Event\StepTested;
-use Behat\Mink\Driver\PantherDriver;
-use Behat\Mink\Driver\Selenium2Driver;
 use Behat\Mink\Exception\Exception as MinkException;
+use Behat\Mink\Exception\UnsupportedDriverActionException;
 use Behat\Mink\Mink;
 use Behat\Mink\Session;
 use Behat\Testwork\Tester\Result\TestResult;
@@ -108,14 +107,10 @@ private function logPageContent()
     private function logScreenshot()
     {
         $session = $this->getSession();
-        $driver = $session->getDriver();
-
-        if (!$driver instanceof Selenium2Driver && !$driver instanceof PantherDriver) {
-            return;
-        }
 
         try {
             $this->saveLog($session->getScreenshot(), 'png');
+        } catch (UnsupportedDriverActionException $unsupportedDriverActionException) {
         } catch (WebDriverException $exception) {}
     }