Skip to content

Commit 2a0a7f8

Browse files
authored
Improve handling of desired capabilities in the remote instance
1 parent f3b01ee commit 2a0a7f8

File tree

1 file changed

+3
-31
lines changed

1 file changed

+3
-31
lines changed

seleniumbase/core/browser_launcher.py

Lines changed: 3 additions & 31 deletions
Original file line numberDiff line numberDiff line change
@@ -995,9 +995,6 @@ def get_remote_driver(
995995
# Set custom desired capabilities
996996
selenoid = False
997997
selenoid_options = None
998-
screen_resolution = None
999-
browser_version = None
1000-
platform_name = None
1001998
for key in desired_caps.keys():
1002999
capabilities[key] = desired_caps[key]
10031000
if key == "selenoid:options":
@@ -1318,36 +1315,11 @@ def get_remote_driver(
13181315
keep_alive=True,
13191316
)
13201317
elif browser_name == constants.Browser.REMOTE:
1321-
selenoid = False
1322-
selenoid_options = None
1323-
screen_resolution = None
1324-
browser_version = None
1325-
platform_name = None
1326-
for key in desired_caps.keys():
1327-
if key == "selenoid:options":
1328-
selenoid = True
1329-
selenoid_options = desired_caps[key]
1330-
elif key == "screenResolution":
1331-
screen_resolution = desired_caps[key]
1332-
elif key == "version" or key == "browserVersion":
1333-
browser_version = desired_caps[key]
1334-
elif key == "platform" or key == "platformName":
1335-
platform_name = desired_caps[key]
13361318
if selenium4:
13371319
remote_options = ArgOptions()
1338-
remote_options.set_capability("cloud:options", desired_caps)
1339-
if selenoid:
1340-
snops = selenoid_options
1341-
remote_options.set_capability("selenoid:options", snops)
1342-
if screen_resolution:
1343-
scres = screen_resolution
1344-
remote_options.set_capability("screenResolution", scres)
1345-
if browser_version:
1346-
br_vers = browser_version
1347-
firefox_options.set_capability("browserVersion", br_vers)
1348-
if platform_name:
1349-
plat_name = platform_name
1350-
firefox_options.set_capability("platformName", plat_name)
1320+
# shovel caps into remote options.
1321+
for cap_name, cap_value in desired_caps.items():
1322+
remote_options.set_capability(cap_name, cap_value)
13511323
return webdriver.Remote(
13521324
command_executor=address,
13531325
options=remote_options,

0 commit comments

Comments
 (0)