Add more select options for CDP Mode

Author: mdmintz

examples/cdp_mode/ReadMe.md

@@ -399,6 +399,8 @@ sb.cdp.mouse_click(selector, timeout=None)
 sb.cdp.nested_click(parent_selector, selector)
 sb.cdp.get_nested_element(parent_selector, selector)
 sb.cdp.select_option_by_text(dropdown_selector, option)
+sb.cdp.select_option_by_index(dropdown_selector, option)
+sb.cdp.select_option_by_value(dropdown_selector, option)
 sb.cdp.flash(selector, duration=1, color="44CC88", pause=0)
 sb.cdp.highlight(selector)
 sb.cdp.focus(selector)

seleniumbase/core/browser_launcher.py

@@ -808,6 +808,8 @@ def uc_open_with_cdp_mode(driver, url=None, **kwargs):
     cdp.get_window_size = CDPM.get_window_size
     cdp.nested_click = CDPM.nested_click
     cdp.select_option_by_text = CDPM.select_option_by_text
+    cdp.select_option_by_index = CDPM.select_option_by_index
+    cdp.select_option_by_value = CDPM.select_option_by_value
     cdp.flash = CDPM.flash
     cdp.highlight = CDPM.highlight
     cdp.focus = CDPM.focus

seleniumbase/core/sb_cdp.py

@@ -823,6 +823,37 @@ def select_option_by_text(self, dropdown_selector, option):
             % (dropdown_selector, option)
         )
 
+    def select_option_by_index(self, dropdown_selector, option):
+        element = self.find_element(dropdown_selector)
+        element.scroll_into_view()
+        options = element.query_selector_all("option")
+        count = 0
+        for found_option in options:
+            if count == int(option):
+                found_option.select_option()
+                return
+            count += 1
+        raise Exception(
+            "Unable to find index option {%s} in dropdown {%s}!"
+            % (dropdown_selector, option)
+        )
+
+    def select_option_by_value(self, dropdown_selector, option):
+        element = self.find_element(dropdown_selector)
+        element.scroll_into_view()
+        options = element.query_selector_all("option")
+        for found_option in options:
+            if (
+                "value" in found_option.attrs
+                and str(found_option.attrs["value"]) == str(option)
+            ):
+                found_option.select_option()
+                return
+        raise Exception(
+            "Unable to find value option {%s} in dropdown {%s}!"
+            % (dropdown_selector, option)
+        )
+
     def flash(
         self,
         selector,  # The CSS Selector to flash

seleniumbase/fixtures/base_case.py

@@ -3198,6 +3198,9 @@ def select_option_by_index(
             timeout = settings.SMALL_TIMEOUT
         if self.timeout_multiplier and timeout == settings.SMALL_TIMEOUT:
             timeout = self.__get_new_timeout(timeout)
+        if self.__is_cdp_swap_needed():
+            self.cdp.select_option_by_index(dropdown_selector, option)
+            return
         self.__select_option(
             dropdown_selector,
             option,
@@ -3222,6 +3225,9 @@ def select_option_by_value(
             timeout = settings.SMALL_TIMEOUT
         if self.timeout_multiplier and timeout == settings.SMALL_TIMEOUT:
             timeout = self.__get_new_timeout(timeout)
+        if self.__is_cdp_swap_needed():
+            self.cdp.select_option_by_value(dropdown_selector, option)
+            return
         self.__select_option(
             dropdown_selector,
             option,