PhantomJS driver will be removed in selenium 4.0.0

pytest_selenium/pytest_selenium.py

@@ -5,12 +5,14 @@
 import argparse
 import copy
 from datetime import datetime
+from distutils.version import LooseVersion
 import os
 import io
 import logging
 
 import pytest
 from requests.structures import CaseInsensitiveDict
+from selenium import __version__ as SELENIUM_VERSION
 from selenium import webdriver
 from selenium.webdriver.common.desired_capabilities import DesiredCapabilities
 from selenium.webdriver.support.event_firing_webdriver import EventFiringWebDriver
@@ -28,14 +30,18 @@
         "Edge": webdriver.Edge,
         "Firefox": webdriver.Firefox,
         "IE": webdriver.Ie,
-        "PhantomJS": webdriver.PhantomJS,
         "Remote": webdriver.Remote,
         "Safari": webdriver.Safari,
         "SauceLabs": webdriver.Remote,
         "TestingBot": webdriver.Remote,
     }
 )
 
+
+# Selenium 4.0.0 deprecated phantom js.
+if LooseVersion(SELENIUM_VERSION) < LooseVersion("4.0.0"):
+    SUPPORTED_DRIVERS["PhantomJS"] = webdriver.PhantomJS
+
 try:
     from appium import webdriver as appiumdriver
 

testing/test_phantomjs.py

@@ -3,8 +3,10 @@
 # file, You can obtain one at http://mozilla.org/MPL/2.0/.
 
 import os
+from distutils.version import LooseVersion
 
 import pytest
+from selenium import __version__ as SELENIUM_VERSION
 
 pytestmark = pytest.mark.nondestructive
 
@@ -20,7 +22,11 @@ def test_pass(webtext):
             assert webtext == u'Success!'
     """
     )
-    testdir.quick_qa("--driver", "PhantomJS", file_test, passed=1)
+    if LooseVersion(SELENIUM_VERSION) < LooseVersion("4.0.0"):
+        testdir.quick_qa("--driver", "PhantomJS", file_test, passed=1)
+    else:
+        reprec = testdir.inline_run("--driver", "PhantomJS")
+        assert reprec.ret == pytest.ExitCode.USAGE_ERROR
 
 
 @pytest.mark.phantomjs
@@ -36,5 +42,8 @@ def driver_args():
         def test_pass(selenium): pass
     """
     )
-    testdir.quick_qa("--driver", "PhantomJS", file_test, passed=1)
-    assert os.path.exists(str(testdir.tmpdir.join("foo.log")))
+    if LooseVersion(SELENIUM_VERSION) < LooseVersion("4.0.0"):
+        testdir.quick_qa("--driver", "PhantomJS", file_test, passed=1)
+        assert os.path.exists(str(testdir.tmpdir.join("foo.log")))
+    else:
+        pass