kyfx
/
SeleniumBase
mirror of https://github.com/seleniumbase/SeleniumBase.git
5
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix methods for switching drivers in CDP Mode

This commit is contained in:
Michael Mintz 2025-02-01 14:36:22 -05:00
parent a2a6e9974c
commit b64cbd818c
3 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
seleniumbase/core/browser_launcher.py
View File

@ -749,6 +749,7 @@ def uc_open_with_cdp_mode(driver, url=None):
core_items.browser = cdp.browser core_items.browser = cdp.browser
core_items.tab = cdp.tab core_items.tab = cdp.tab
core_items.util = cdp.util core_items.util = cdp.util
cdp._swap_driver = CDPM._swap_driver
cdp.core = core_items cdp.core = core_items
cdp.loop = cdp.get_event_loop() cdp.loop = cdp.get_event_loop()
driver.cdp = cdp driver.cdp = cdp

5
seleniumbase/core/sb_cdp.py
View File

@ -19,6 +19,11 @@ class CDPMethods():
self.page = page self.page = page
self.driver = driver self.driver = driver
def _swap_driver(self, driver):
self.driver = driver
self.page = driver.cdp.page
self.loop = driver.cdp.loop
def __slow_mode_pause_if_set(self): def __slow_mode_pause_if_set(self):
if ( if (
(hasattr(sb_config, "demo_mode") and sb_config.demo_mode) (hasattr(sb_config, "demo_mode") and sb_config.demo_mode)

4
seleniumbase/fixtures/base_case.py
View File

@ -4358,6 +4358,8 @@ class BaseCase(unittest.TestCase):
self.driver = driver self.driver = driver
if self.driver in self._drivers_browser_map: if self.driver in self._drivers_browser_map:
self.browser = self._drivers_browser_map[self.driver] self.browser = self._drivers_browser_map[self.driver]
if self.__is_cdp_swap_needed():
self.cdp._swap_driver(self.driver)
self.bring_active_window_to_front() self.bring_active_window_to_front()
def switch_to_default_driver(self): def switch_to_default_driver(self):
@ -4366,6 +4368,8 @@ class BaseCase(unittest.TestCase):
self.driver = self._default_driver self.driver = self._default_driver
if self.driver in self._drivers_browser_map: if self.driver in self._drivers_browser_map:
self.browser = self._drivers_browser_map[self.driver] self.browser = self._drivers_browser_map[self.driver]
if self.__is_cdp_swap_needed():
self.cdp._swap_driver(self.driver)
self.bring_active_window_to_front() self.bring_active_window_to_front()
def save_screenshot( def save_screenshot(