@@ -252,32 +252,40 @@ def test_phantomjs_remote(self):
252252 self .assertEqual (driver , expected_webdriver )
253253
254254 def test_htmlunit (self ):
255+ caps = webdriver .DesiredCapabilities .HTMLUNIT
255256 expected_webdriver = mock ()
256257 when (webdriver ).Remote (command_executor = 'None' ,
258+ desired_capabilities = caps ,
257259 browser_profile = None ,
258260 options = None ).thenReturn (expected_webdriver )
259261 driver = self .creator .create_htmlunit ({}, None )
260262 self .assertEqual (driver , expected_webdriver )
261263
262264 def test_htmlunit_with_js (self ):
265+ caps = webdriver .DesiredCapabilities .HTMLUNITWITHJS
263266 expected_webdriver = mock ()
264267 when (webdriver ).Remote (command_executor = 'None' ,
268+ desired_capabilities = caps ,
265269 browser_profile = None ,
266270 options = None ).thenReturn (expected_webdriver )
267271 driver = self .creator .create_htmlunit_with_js ({}, None )
268272 self .assertEqual (driver , expected_webdriver )
269273
270274 def test_android (self ):
275+ caps = webdriver .DesiredCapabilities .ANDROID
271276 expected_webdriver = mock ()
272277 when (webdriver ).Remote (command_executor = 'None' ,
278+ desired_capabilities = caps ,
273279 browser_profile = None ,
274280 options = None ).thenReturn (expected_webdriver )
275281 driver = self .creator .create_android ({}, None )
276282 self .assertEqual (driver , expected_webdriver )
277283
278284 def test_iphone (self ):
285+ caps = webdriver .DesiredCapabilities .IPHONE
279286 expected_webdriver = mock ()
280287 when (webdriver ).Remote (command_executor = 'None' ,
288+ desired_capabilities = caps ,
281289 browser_profile = None ,
282290 options = None ).thenReturn (expected_webdriver )
283291 driver = self .creator .create_iphone ({}, None )
0 commit comments