FIX: Fix error message thrown by selenium exceptions upon wait

Currently we were casting entire exception to the string (including not readable stacktrace) from selenium to error message. Instead, we should pass just the error message
yashaka · Feb 10, 2024 · bae38aa · bae38aa
1 parent 1dd26e2
commit bae38aa
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 11 deletions.
diff --git a/selene/core/wait.py b/selene/core/wait.py
@@ -119,11 +119,9 @@ def logic(fn: Callable[[E], R]) -> R:
                     return fn(self.entity)
                 except Exception as reason:
                     if time.time() > finish_time:
-                        reason_message = str(reason)
-
                         reason_string = '{name}: {message}'.format(
                             name=reason.__class__.__name__,
-                            message=reason_message,
+                            message=getattr(reason, "msg", str(reason)),
                         )
                         # TODO: think on how can we improve logging failures in selene, e.g. reverse msg and stacktrace
                         # stacktrace = getattr(reason, 'stacktrace', None)

diff --git a/tests/integration/error_messages_test.py b/tests/integration/error_messages_test.py
@@ -92,9 +92,10 @@ def test_element_search_fails_with_message_when_implicitly_waits_for_condition(
         'Timed out after 0.1s, while waiting for:',
         "browser.element(('css selector', '#hidden-button')).click",
         '',
-        'Reason: ElementNotInteractableException: Message: element not interactable',
+        'Reason: ElementNotInteractableException: element not interactable',
         '(Session info: *)',
-        'Stacktrace: *',
+        'Screenshot: *.png',
+        'PageSource: *.html',
     ]
 
 
@@ -119,9 +120,10 @@ def test_inner_element_search_fails_with_message_when_implicitly_waits_for_condi
         "browser.element(('css selector', '#container')).element(('css selector', "
         "'#hidden-button')).click",
         '',
-        'Reason: ElementNotInteractableException: Message: element not interactable',
+        'Reason: ElementNotInteractableException: element not interactable',
         '(Session info: *)',
-        'Stacktrace: *',
+        'Screenshot: *.png',
+        'PageSource: *.html',
     ]
 
 
@@ -146,9 +148,10 @@ def test_inner_element_search_fails_with_message_when_implicitly_waits_for_condi
         "browser.element(('css selector', '#hidden-container')).element(('css "
         "selector', '#button')).click",
         '',
-        'Reason: ElementNotInteractableException: Message: element not interactable',
+        'Reason: ElementNotInteractableException: element not interactable',
         '(Session info: *)',
-        'Stacktrace: *',
+        'Screenshot: *.png',
+        'PageSource: *.html',
     ]
 
 
@@ -173,11 +176,12 @@ def test_inner_element_search_fails_with_message_when_implicitly_waits_for_condi
         "browser.element(('css selector', '#not-existing')).element(('css selector', "
         "'#button')).click",
         '',
-        'Reason: NoSuchElementException: Message: no such element: Unable to locate '
+        'Reason: NoSuchElementException: no such element: Unable to locate '
         'element: {"method":"css selector","selector":"#not-existing"}',
         '(Session info: *); For documentation on this error, please visit: '
         'https://www.selenium.dev/documentation/webdriver/troubleshooting/errors#no-such-element-exception',
-        'Stacktrace: *',
+        'Screenshot: *.png',
+        'PageSource: *.html',
     ]