|
18 | 18 | package org.openqa.selenium.firefox;
|
19 | 19 |
|
20 | 20 | import static org.openqa.selenium.firefox.FirefoxDriver.SystemProperty.DRIVER_USE_MARIONETTE;
|
21 |
| -import static org.openqa.selenium.remote.CapabilityType.BROWSER_NAME; |
22 |
| -import static org.openqa.selenium.remote.CapabilityType.BROWSER_VERSION; |
23 |
| -import static org.openqa.selenium.remote.CapabilityType.PLATFORM_NAME; |
24 | 21 | import static org.openqa.selenium.remote.CapabilityType.PROXY;
|
25 | 22 |
|
26 | 23 | import com.google.common.collect.Maps;
|
@@ -196,11 +193,15 @@ private static Capabilities dropCapabilities(Capabilities capabilities) {
|
196 | 193 | return new ImmutableCapabilities();
|
197 | 194 | }
|
198 | 195 |
|
199 |
| - final Set<String> toRemove = isLegacy(capabilities) |
200 |
| - ? Sets.newHashSet(BINARY, PROFILE) |
201 |
| - : Sets.newHashSet(BROWSER_NAME, BROWSER_VERSION, PLATFORM_NAME); |
202 |
| - MutableCapabilities caps = new MutableCapabilities( |
203 |
| - Maps.filterKeys(capabilities.asMap(), key -> !toRemove.contains(key))); |
| 196 | + MutableCapabilities caps; |
| 197 | + |
| 198 | + if (isLegacy(capabilities)) { |
| 199 | + final Set<String> toRemove = Sets.newHashSet(BINARY, PROFILE); |
| 200 | + caps = new MutableCapabilities( |
| 201 | + Maps.filterKeys(capabilities.asMap(), key -> !toRemove.contains(key))); |
| 202 | + } else { |
| 203 | + caps = new MutableCapabilities(capabilities); |
| 204 | + } |
204 | 205 |
|
205 | 206 | // Ensure that the proxy is in a state fit to be sent to the extension
|
206 | 207 | Proxy proxy = Proxy.extractFrom(capabilities);
|
|
0 commit comments