Add new CDP Mode methods

Author: mdmintz

examples/cdp_mode/ReadMe.md

@@ -413,6 +413,7 @@ sb.cdp.send_keys(selector, text, timeout=None)
 sb.cdp.press_keys(selector, text, timeout=None)
 sb.cdp.type(selector, text, timeout=None)
 sb.cdp.set_value(selector, text, timeout=None)
+sb.cdp.submit(selector)
 sb.cdp.evaluate(expression)
 sb.cdp.js_dumps(obj_name)
 sb.cdp.maximize()
@@ -438,6 +439,8 @@ sb.cdp.get_page_source()
 sb.cdp.get_user_agent()
 sb.cdp.get_cookie_string()
 sb.cdp.get_locale_code()
+sb.cdp.get_local_storage_item(key)
+sb.cdp.get_session_storage_item(key)
 sb.cdp.get_screen_rect()
 sb.cdp.get_window_rect()
 sb.cdp.get_window_size()
@@ -454,6 +457,8 @@ sb.cdp.get_element_attribute(selector, attribute)
 sb.cdp.get_attribute(selector, attribute)
 sb.cdp.get_element_html(selector)
 sb.cdp.set_locale(locale)
+sb.cdp.set_local_storage_item(key, value)
+sb.cdp.set_session_storage_item(key, value)
 sb.cdp.set_attributes(selector, attribute, value)
 sb.cdp.gui_press_key(key)
 sb.cdp.gui_press_keys(keys)

seleniumbase/core/browser_launcher.py

@@ -662,6 +662,7 @@ def uc_open_with_cdp_mode(driver, url=None):
     cdp.press_keys = CDPM.press_keys
     cdp.type = CDPM.type
     cdp.set_value = CDPM.set_value
+    cdp.submit = CDPM.submit
     cdp.evaluate = CDPM.evaluate
     cdp.js_dumps = CDPM.js_dumps
     cdp.maximize = CDPM.maximize
@@ -670,6 +671,8 @@ def uc_open_with_cdp_mode(driver, url=None):
     cdp.set_window_rect = CDPM.set_window_rect
     cdp.reset_window_size = CDPM.reset_window_size
     cdp.set_locale = CDPM.set_locale
+    cdp.set_local_storage_item = CDPM.set_local_storage_item
+    cdp.set_session_storage_item = CDPM.set_session_storage_item
     cdp.set_attributes = CDPM.set_attributes
     cdp.gui_press_key = CDPM.gui_press_key
     cdp.gui_press_keys = CDPM.gui_press_keys
@@ -705,6 +708,8 @@ def uc_open_with_cdp_mode(driver, url=None):
     cdp.get_user_agent = CDPM.get_user_agent
     cdp.get_cookie_string = CDPM.get_cookie_string
     cdp.get_locale_code = CDPM.get_locale_code
+    cdp.get_local_storage_item = CDPM.get_local_storage_item
+    cdp.get_session_storage_item = CDPM.get_session_storage_item
     cdp.get_text = CDPM.get_text
     cdp.get_title = CDPM.get_title
     cdp.get_page_title = CDPM.get_title

seleniumbase/core/sb_cdp.py

@@ -955,6 +955,20 @@ def set_value(self, selector, text, timeout=None):
         self.__slow_mode_pause_if_set()
         self.loop.run_until_complete(self.page.sleep(0.025))
 
+    def submit(self, selector):
+        submit_script = (
+            """elm = document.querySelector('%s');
+            const event = new KeyboardEvent("keydown", {
+                key: "Enter",
+                keyCode: 13,
+                code: "Enter",
+                which: 13,
+                bubbles: true
+            });
+            elm.dispatchEvent(event);""" % selector
+        )
+        self.loop.run_until_complete(self.page.evaluate(submit_script))
+
     def evaluate(self, expression):
         """Run a JavaScript expression and return the result."""
         expression = expression.strip()
@@ -1115,6 +1129,16 @@ def get_locale_code(self):
             self.page.evaluate("navigator.language || navigator.languages[0]")
         )
 
+    def get_local_storage_item(self, key):
+        js_code = """localStorage.getItem('%s');""" % key
+        with suppress(Exception):
+            return self.loop.run_until_complete(self.page.evaluate(js_code))
+
+    def get_session_storage_item(self, key):
+        js_code = """sessionStorage.getItem('%s');""" % key
+        with suppress(Exception):
+            return self.loop.run_until_complete(self.page.evaluate(js_code))
+
     def get_screen_rect(self):
         coordinates = self.loop.run_until_complete(
             self.page.js_dumps("window.screen")
@@ -1302,6 +1326,16 @@ def set_locale(self, locale):
         """(Settings will take effect on the next page load)"""
         self.loop.run_until_complete(self.page.set_locale(locale))
 
+    def set_local_storage_item(self, key, value):
+        js_code = """localStorage.setItem('%s','%s');""" % (key, value)
+        with suppress(Exception):
+            self.loop.run_until_complete(self.page.evaluate(js_code))
+
+    def set_session_storage_item(self, key, value):
+        js_code = """sessionStorage.setItem('%s','%s');""" % (key, value)
+        with suppress(Exception):
+            self.loop.run_until_complete(self.page.evaluate(js_code))
+
     def set_attributes(self, selector, attribute, value):
         """This method uses JavaScript to set/update a common attribute.
         All matching selectors from querySelectorAll() are used.

seleniumbase/fixtures/base_case.py

@@ -1164,6 +1164,9 @@ def submit(self, selector, by="css selector"):
         """Alternative to self.driver.find_element_by_*(SELECTOR).submit()"""
         self.__check_scope()
         selector, by = self.__recalculate_selector(selector, by)
+        if self.__is_cdp_swap_needed():
+            self.cdp.submit(selector)
+            return
         element = self.wait_for_element_clickable(
             selector, by=by, timeout=settings.SMALL_TIMEOUT
         )
@@ -8800,6 +8803,9 @@ def set_local_storage_item(self, key, value):
         self.__check_scope()
         if not self.__is_valid_storage_url():
             raise WebDriverException("Local Storage is not available here!")
+        if self.__is_cdp_swap_needed():
+            self.cdp.set_local_storage_item(key, value)
+            return
         self.execute_script(
             "window.localStorage.setItem('{}', '{}');".format(key, value)
         )
@@ -8808,6 +8814,8 @@ def get_local_storage_item(self, key):
         self.__check_scope()
         if not self.__is_valid_storage_url():
             raise WebDriverException("Local Storage is not available here!")
+        if self.__is_cdp_swap_needed():
+            return self.cdp.get_local_storage_item(key)
         return self.execute_script(
             "return window.localStorage.getItem('{}');".format(key)
         )
@@ -8859,6 +8867,9 @@ def set_session_storage_item(self, key, value):
         self.__check_scope()
         if not self.__is_valid_storage_url():
             raise WebDriverException("Session Storage is not available here!")
+        if self.__is_cdp_swap_needed():
+            self.cdp.set_session_storage_item(key, value)
+            return
         self.execute_script(
             "window.sessionStorage.setItem('{}', '{}');".format(key, value)
         )
@@ -8867,6 +8878,8 @@ def get_session_storage_item(self, key):
         self.__check_scope()
         if not self.__is_valid_storage_url():
             raise WebDriverException("Session Storage is not available here!")
+        if self.__is_cdp_swap_needed():
+            return self.cdp.get_session_storage_item(key)
         return self.execute_script(
             "return window.sessionStorage.getItem('{}');".format(key)
         )