Improve "Selenium2Driver::applyTimeouts" error handling for Selenium 3 support

Author: aik099

src/Selenium2Driver.php

@@ -14,6 +14,7 @@
 use Behat\Mink\KeyModifier;
 use Behat\Mink\Selector\Xpath\Escaper;
 use WebDriver\Element;
+use WebDriver\Exception\InvalidArgument;
 use WebDriver\Exception\NoSuchElement;
 use WebDriver\Exception\StaleElementReference;
 use WebDriver\Exception\UnknownCommand;
@@ -370,6 +371,10 @@ private function applyTimeouts(): void
                 $this->getWebDriverSession()->timeouts($type, $param);
             }
         } catch (UnknownError $e) {
+            // Selenium 2.x.
+            throw new DriverException('Error setting timeout: ' . $e->getMessage(), 0, $e);
+        } catch (InvalidArgument $e) {
+            // Selenium 3.x.
             throw new DriverException('Error setting timeout: ' . $e->getMessage(), 0, $e);
         }
     }