Add "ProxyConnectionException" with a retry

Author: mdmintz

seleniumbase/common/exceptions.py

@@ -5,6 +5,7 @@
     NotUsingChromeException => Used by Chrome-only methods if not using Chrome.
     NotUsingChromiumException => Used by Chromium-only methods if not Chromium.
     OutOfScopeException => Used by BaseCase methods when setUp() is skipped.
+    ProxyConnectionException => Called when the proxy connection failed.
     TextNotVisibleException => Called when expected text fails to appear.
     TimeLimitExceededException => Called when exceeding "--time-limit=SECONDS".
     TimeoutException => Called when some timeout limit has been exceeded.
@@ -36,6 +37,10 @@ class OutOfScopeException(Exception):
     pass
 
 
+class ProxyConnectionException(Exception):
+    pass
+
+
 class TextNotVisibleException(Exception):
     pass
 

seleniumbase/fixtures/base_case.py

@@ -65,6 +65,7 @@ def test_anything(self):
     NotConnectedException,
     NotUsingChromeException,
     NotUsingChromiumException,
+    ProxyConnectionException,
     OutOfScopeException,
     VisualException,
 )
@@ -261,6 +262,7 @@ def open(self, url):
             elif (
                 "ERR_INTERNET_DISCONNECTED" in e.msg
                 or "neterror?e=dnsNotFound" in e.msg
+                or "ERR_PROXY_CONNECTION_FAILED" in e.msg
             ):
                 shared_utils.check_if_time_limit_exceeded()
                 self.__check_browser()
@@ -272,8 +274,13 @@ def open(self, url):
                         "ERR_INTERNET_DISCONNECTED" in e2.msg
                         or "neterror?e=dnsNotFound" in e2.msg
                     ):
-                        message = "Internet unreachable!"
+                        message = "ERR_INTERNET_DISCONNECTED: "
+                        message += "Internet unreachable!"
                         raise NotConnectedException(message)
+                    elif "ERR_PROXY_CONNECTION_FAILED" in e2.msg:
+                        message = "ERR_PROXY_CONNECTION_FAILED: "
+                        message += "Internet unreachable and/or invalid proxy!"
+                        raise ProxyConnectionException(message)
                     else:
                         raise
             elif "Timed out receiving message from renderer" in e.msg: