Add back the ability to set profile when using Firefox 45ESR. Fixes S…

…eleniumHQ#3897
ghaseminya · Apr 25, 2017 · 41d9d32 · 41d9d32
1 parent 8f325df
commit 41d9d32
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/py/selenium/webdriver/remote/webdriver.py b/py/selenium/webdriver/remote/webdriver.py
@@ -178,7 +178,10 @@ def start_session(self, capabilities, browser_profile=None):
             raise InvalidArgumentException("Capabilities must be a dictionary")
         w3c_caps = {"firstMatch": [], "alwaysMatch": {}}
         if browser_profile:
-            capabilities["moz:firefoxOptions"]["profile"] = browser_profile.encoded
+            if "moz:firefoxOptions" in capabilities:
+                capabilities["moz:firefoxOptions"]["profile"] = browser_profile.encoded
+            else:
+                capabilities['desiredCapabilities'] = {'firefox_profile': browser_profile.encoded}
         w3c_caps["alwaysMatch"].update(capabilities)
         parameters = {"capabilities": w3c_caps,
                       "desiredCapabilities": capabilities}