Improve Demo Mode reliability on sites with restrictive CSPs

Author: mdmintz

seleniumbase/fixtures/base_case.py

@@ -3174,8 +3174,12 @@ def assert_downloaded_file(self, file, timeout=None, browser=False):
             page_actions.timeout_exception("NoSuchFileException", message)
         if self.demo_mode:
             messenger_post = ("ASSERT DOWNLOADED FILE: [%s]" % file)
-            js_utils.post_messenger_success_message(
-                self.driver, messenger_post, self.message_duration)
+            try:
+                js_utils.activate_jquery(self.driver)
+                js_utils.post_messenger_success_message(
+                    self.driver, messenger_post, self.message_duration)
+            except Exception:
+                pass
 
     def assert_true(self, expr, msg=None):
         """ Asserts that the expression is True.

seleniumbase/fixtures/js_utils.py

@@ -634,10 +634,15 @@ def highlight_with_js_2(driver, message, selector, o_bs, msg_dur):
     driver.execute_script(script)
     time.sleep(0.0181)
 
-    post_messenger_success_message(driver, message, msg_dur)
+    try:
+        activate_jquery(driver)
+        post_messenger_success_message(driver, message, msg_dur)
+    except Exception:
+        pass
 
-    script = ("""document.querySelector('%s').style.boxShadow =
-              '%s';""" % (selector, o_bs))
+    script = (
+        """document.querySelector('%s').style.boxShadow = '%s';""" % (
+            selector, o_bs))
     driver.execute_script(script)
 
 
@@ -668,7 +673,11 @@ def highlight_with_jquery_2(driver, message, selector, o_bs, msg_dur):
     driver.execute_script(script)
     time.sleep(0.0181)
 
-    post_messenger_success_message(driver, message, msg_dur)
+    try:
+        activate_jquery(driver)
+        post_messenger_success_message(driver, message, msg_dur)
+    except Exception:
+        pass
 
     script = """jQuery('%s').css('box-shadow', '%s');""" % (selector, o_bs)
     driver.execute_script(script)