Fix issue with self.has_exception()

Author: mdmintz

seleniumbase/fixtures/base_case.py

@@ -307,6 +307,7 @@ def open(self, url):
                     pass  # Odd issue where the open did happen. Continue.
             else:
                 raise
+        unittest.has_exception = False
         if (
             self.undetectable
             or (
@@ -4119,6 +4120,7 @@ def wait_for_ready_state_complete(self, timeout=None):
             and settings.SKIP_JS_WAITS
         ):
             time.sleep(0.05)
+        unittest.has_exception = False
         return True
 
     def wait_for_angularjs(self, timeout=None, **kwargs):
@@ -6360,12 +6362,14 @@ def assert_pdf_text(
         )
         if type(page) is int:
             if text not in pdf_text:
+                unittest.has_exception = True
                 raise Exception(
                     "PDF [%s] is missing expected text [%s] on "
                     "page [%s]!" % (pdf, text, page)
                 )
         else:
             if text not in pdf_text:
+                unittest.has_exception = True
                 raise Exception(
                     "PDF [%s] is missing expected text [%s]!" % (pdf, text)
                 )
@@ -6740,40 +6744,52 @@ def assert_downloaded_file(self, file, timeout=None, browser=False):
     def assert_true(self, expr, msg=None):
         """Asserts that the expression is True.
         Will raise an exception if the statement if False."""
+        unittest.has_exception = True
         self.assertTrue(expr, msg=msg)
+        unittest.has_exception = False
 
     def assert_false(self, expr, msg=None):
         """Asserts that the expression is False.
         Will raise an exception if the statement if True."""
+        unittest.has_exception = True
         self.assertFalse(expr, msg=msg)
+        unittest.has_exception = False
 
     def assert_equal(self, first, second, msg=None):
         """Asserts that the two values are equal.
         Will raise an exception if the values are not equal."""
+        unittest.has_exception = True
         self.assertEqual(first, second, msg=msg)
+        unittest.has_exception = False
 
     def assert_not_equal(self, first, second, msg=None):
         """Asserts that the two values are not equal.
         Will raise an exception if the values are equal."""
+        unittest.has_exception = True
         self.assertNotEqual(first, second, msg=msg)
+        unittest.has_exception = False
 
     def assert_in(self, first, second, msg=None):
         """Asserts that the first string is in the second string.
         Will raise an exception if the first string is not in the second."""
+        unittest.has_exception = True
         self.assertIn(first, second, msg=msg)
+        unittest.has_exception = False
 
     def assert_not_in(self, first, second, msg=None):
         """Asserts that the first string is not in the second string.
         Will raise an exception if the first string is in the second string."""
+        unittest.has_exception = True
         self.assertNotIn(first, second, msg=msg)
+        unittest.has_exception = False
 
     def assert_raises(self, *args, **kwargs):
         """Asserts that the following block of code raises an exception.
         Will raise an exception if the block of code has no exception.
         Usage Example =>
-                # Verify that the expected exception is raised.
-                with self.assert_raises(Exception):
-                    raise Exception("Expected Exception!") """
+            # Verify that the expected exception is raised.
+            with self.assert_raises(Exception):
+                raise Exception("Expected Exception!") """
         return self.assertRaises(*args, **kwargs)
 
     def wait_for_attribute(
@@ -6884,9 +6900,11 @@ def assert_title(self, title):
                     self.wait_for_ready_state_complete()
                     time.sleep(2)
                     actual = self.get_page_title().strip()
+                    unittest.has_exception = True
                     self.assertEqual(
                         expected, actual, error % (expected, actual)
                     )
+                    unittest.has_exception = False
         if self.demo_mode and not self.recorder_mode:
             a_t = "ASSERT TITLE"
             if self._language != "English":
@@ -6931,9 +6949,11 @@ def assert_title_contains(self, substring):
                     self.wait_for_ready_state_complete()
                     time.sleep(2)
                     actual = self.get_page_title().strip()
+                    unittest.has_exception = True
                     self.assertIn(
                         expected, actual, error % (expected, actual)
                     )
+                    unittest.has_exception = False
         if self.demo_mode and not self.recorder_mode:
             a_t = "ASSERT TITLE CONTAINS"
             if self._language != "English":
@@ -6968,7 +6988,9 @@ def assert_url(self, url):
                 self.wait_for_ready_state_complete()
                 time.sleep(2)
                 actual = self.get_current_url().strip()
+                unittest.has_exception = True
                 self.assertEqual(expected, actual, error % (expected, actual))
+                unittest.has_exception = False
         if self.demo_mode and not self.recorder_mode:
             a_u = "ASSERT URL"
             if self._language != "English":
@@ -7006,7 +7028,9 @@ def assert_url_contains(self, substring):
                 self.wait_for_ready_state_complete()
                 time.sleep(2)
                 actual = self.get_current_url().strip()
+                unittest.has_exception = True
                 self.assertIn(expected, actual, error % (expected, actual))
+                unittest.has_exception = False
         if self.demo_mode and not self.recorder_mode:
             a_u = "ASSERT URL CONTAINS"
             if self._language != "English":
@@ -7103,6 +7127,7 @@ def assert_no_js_errors(self, exclude=[]):
             er_str = str(errors)
             er_str = er_str.replace("[{", "[\n{").replace("}, {", "},\n{")
             current_url = self.get_current_url()
+            unittest.has_exception = True
             raise Exception(
                 "JavaScript errors found on %s => %s" % (current_url, er_str)
             )
@@ -9812,6 +9837,7 @@ def __get_new_timeout(self, timeout):
 
     def __check_scope(self):
         if hasattr(self, "browser"):  # self.browser stores the type of browser
+            unittest.has_exception = False
             return  # All good: setUp() already initialized variables in "self"
         else:
             message = (
@@ -9827,6 +9853,7 @@ def __check_scope(self):
                 "\n variables, which are initialized during the setUp() method"
                 "\n that runs automatically before all tests called by pytest."
             )
+            unittest.has_exception = True
             raise OutOfScopeException(message)
 
     ############
@@ -10082,6 +10109,7 @@ def process_deferred_asserts(self, print_only=False):
             if print_only:
                 print(exception_output)
             else:
+                unittest.has_exception = True
                 raise Exception(exception_output.replace("\\n", "\n"))
 
     ############
@@ -13874,6 +13902,8 @@ def setUp(self, masterqa_mode=False):
             # Some actions such as hover-clicking are different on mobile.
             self.mobile_emulator = False
 
+        unittest.has_exception = False
+
         # Configure the test time limit (if used).
         self.set_time_limit(self.time_limit)
 
@@ -14158,6 +14188,8 @@ def __has_exception(self):
             has_exception = sys.exc_info()[1] is not None
         if self.__will_be_skipped and hasattr(self, "_using_sb_fixture"):
             has_exception = False
+        if python3_11_or_newer and unittest.has_exception:
+            has_exception = True
         return has_exception
 
     def __get_test_id(self):

seleniumbase/fixtures/page_actions.py

@@ -23,6 +23,7 @@
 import codecs
 import os
 import time
+import unittest
 from selenium.common.exceptions import ElementNotInteractableException
 from selenium.common.exceptions import ElementNotVisibleException
 from selenium.common.exceptions import NoAlertPresentException
@@ -224,6 +225,7 @@ def hover_element(driver, element):
 
 
 def timeout_exception(exception, message):
+    unittest.has_exception = True
     exc, msg = shared_utils.format_exc(exception, message)
     raise exc(msg)