Improve "driver.highlight(selector)" functionality

Author: mdmintz

seleniumbase/core/browser_launcher.py

@@ -184,6 +184,7 @@ def extend_driver(driver):
     driver.get_origin = DM.get_origin
     driver.get_user_agent = DM.get_user_agent
     driver.highlight = DM.highlight
+    driver.highlight_click = DM.highlight_click
     driver.sleep = time.sleep
     driver.get_page_source = DM.get_page_source
     driver.get_title = DM.get_title

seleniumbase/core/sb_driver.py

@@ -142,8 +142,22 @@ def get_user_agent(self, *args, **kwargs):
         return js_utils.get_user_agent(self.driver, *args, **kwargs)
 
     def highlight(self, *args, **kwargs):
+        w_args = kwargs
+        if "loops" in w_args:
+            w_args.pop("loops")
+        element = page_actions.wait_for_element(self.driver, *args, **w_args)
+        browser = self.driver.capabilities["browserName"].lower()
+        js_utils.slow_scroll_to_element(self.driver, element, browser)
+        if "timeout" in kwargs:
+            kwargs.pop("timeout")
         js_utils.highlight(self.driver, *args, **kwargs)
 
+    def highlight_click(self, *args, **kwargs):
+        self.highlight(*args, **kwargs)
+        if "loops" in kwargs:
+            kwargs.pop("loops")
+        page_actions.click(self.driver, *args, **kwargs)
+
     def set_wire_proxy(self, string):
         """Set a proxy server for selenium-wire mode ("--wire")
         Examples:  (ONLY avilable if using selenium-wire mode!)