Skip to content

Commit 519ea1e

Browse files
committed
Do some refactoring
1 parent fbadbdd commit 519ea1e

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

seleniumbase/fixtures/base_case.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -117,7 +117,6 @@ def __init__(self, *args, **kwargs):
117117
self.__device_width = None
118118
self.__device_height = None
119119
self.__device_pixel_ratio = None
120-
self.__driver_browser_map = {}
121120
self.__changed_jqc_theme = False
122121
self.__jqc_default_theme = None
123122
self.__jqc_default_color = None
@@ -131,6 +130,7 @@ def __init__(self, *args, **kwargs):
131130
self._html_report_extra = [] # (Used by pytest_plugin.py)
132131
self._default_driver = None
133132
self._drivers_list = []
133+
self._drivers_browser_map = {}
134134
self._chart_data = {}
135135
self._chart_count = 0
136136
self._chart_label = {}
@@ -3043,7 +3043,7 @@ def get_new_driver(
30433043
device_pixel_ratio=d_p_r,
30443044
)
30453045
self._drivers_list.append(new_driver)
3046-
self.__driver_browser_map[new_driver] = browser_name
3046+
self._drivers_browser_map[new_driver] = browser_name
30473047
if switch_to:
30483048
self.driver = new_driver
30493049
self.browser = browser_name
@@ -3124,16 +3124,16 @@ def switch_to_driver(self, driver):
31243124
You may need this if using self.get_new_driver() in your code."""
31253125
self.__check_scope()
31263126
self.driver = driver
3127-
if self.driver in self.__driver_browser_map:
3128-
self.browser = self.__driver_browser_map[self.driver]
3127+
if self.driver in self._drivers_browser_map:
3128+
self.browser = self._drivers_browser_map[self.driver]
31293129
self.bring_active_window_to_front()
31303130

31313131
def switch_to_default_driver(self):
3132-
""" Sets self.driver to the default/original driver. """
3132+
""" Sets self.driver to the default/initial driver. """
31333133
self.__check_scope()
31343134
self.driver = self._default_driver
3135-
if self.driver in self.__driver_browser_map:
3136-
self.browser = self.__driver_browser_map[self.driver]
3135+
if self.driver in self._drivers_browser_map:
3136+
self.browser = self._drivers_browser_map[self.driver]
31373137
self.bring_active_window_to_front()
31383138

31393139
def save_screenshot(

0 commit comments

Comments
 (0)