@@ -137,7 +137,7 @@ public FirefoxDriver(FirefoxProfile profile) :
137
137
/// <param name="capabilities">The <see cref="ICapabilities"/> object containing the desired
138
138
/// capabilities of this FirefoxDriver.</param>
139
139
public FirefoxDriver ( ICapabilities capabilities )
140
- : this ( ExtractBinary ( capabilities ) , ExtractProfile ( capabilities ) )
140
+ : this ( ExtractBinary ( capabilities ) , ExtractProfile ( capabilities ) , capabilities , RemoteWebDriver . DefaultCommandTimeout )
141
141
{
142
142
}
143
143
@@ -162,7 +162,12 @@ public FirefoxDriver(FirefoxBinary binary, FirefoxProfile profile)
162
162
/// to be used in starting Firefox.</param>
163
163
/// <param name="commandTimeout">The maximum amount of time to wait for each command.</param>
164
164
public FirefoxDriver ( FirefoxBinary binary , FirefoxProfile profile , TimeSpan commandTimeout )
165
- : base ( CreateExtensionConnection ( binary , profile , commandTimeout ) , DesiredCapabilities . Firefox ( ) )
165
+ : this ( binary , profile , DesiredCapabilities . Firefox ( ) , commandTimeout )
166
+ {
167
+ }
168
+
169
+ private FirefoxDriver ( FirefoxBinary binary , FirefoxProfile profile , ICapabilities capabilities , TimeSpan commandTimeout )
170
+ : base ( CreateExtensionConnection ( binary , profile , commandTimeout ) , RemoveUnneededCapabilities ( capabilities ) )
166
171
{
167
172
this . binary = binary ;
168
173
this . profile = profile ;
@@ -345,6 +350,14 @@ private static ExtensionConnection CreateExtensionConnection(FirefoxBinary binar
345
350
ExtensionConnection extension = new ExtensionConnection ( binary , profileToUse , "localhost" , commandTimeout ) ;
346
351
return extension ;
347
352
}
353
+
354
+ private static ICapabilities RemoveUnneededCapabilities ( ICapabilities capabilities )
355
+ {
356
+ DesiredCapabilities caps = capabilities as DesiredCapabilities ;
357
+ caps . Capabilities . Remove ( FirefoxDriver . ProfileCapabilityName ) ;
358
+ caps . Capabilities . Remove ( FirefoxDriver . BinaryCapabilityName ) ;
359
+ return caps ;
360
+ }
348
361
#endregion
349
362
}
350
363
}
0 commit comments