11import unittest
22
3- from mockito import mock , verify , verifyNoMoreInteractions
3+ from mockito import when , mock , verify , verifyNoMoreInteractions
44from selenium import webdriver
55
66from Selenium2Library .keywords ._browsermanagement import _BrowserManagementKeywords
77
88
9- class BrowserManagementTests (unittest .TestCase ):
9+ class BrowserManagementTests (unittest .TestCase ):
10+
1011
11-
1212 def test_create_firefox_browser (self ):
1313 test_browsers = ((webdriver .Firefox , "ff" ), (webdriver .Firefox , "firEfOx" ))
1414
1515 for test_browser in test_browsers :
1616 self .verify_browser (* test_browser )
17-
17+
1818 def mock_createProfile (self , profile_directory = None ):
1919 self .ff_profile_dir = profile_directory
2020 return self .old_profile_init (profile_directory )
@@ -104,23 +104,19 @@ def test_bad_browser_name(self):
104104
105105 def test_create_webdriver (self ):
106106 bm = _BrowserManagementWithLoggingStubs ()
107- capt_data = {}
108- class FakeWebDriver (mock ):
109- def __init__ (self , some_arg = None ):
110- mock .__init__ (self )
111- capt_data ['some_arg' ] = some_arg
112- capt_data ['webdriver' ] = self
107+ FakeWebDriver = mock ()
108+ driver = mock ()
109+ when (FakeWebDriver ).__call__ (some_arg = 1 ).thenReturn (driver )
110+ when (FakeWebDriver ).__call__ (some_arg = 2 ).thenReturn (driver )
113111 webdriver .FakeWebDriver = FakeWebDriver
114112 try :
115- index = bm .create_webdriver ('FakeWebDriver' , 'fake' , some_arg = 1 )
116- self .assertEquals (capt_data ['some_arg' ], 1 )
117- self .assertEquals (capt_data ['webdriver' ], bm ._current_browser ())
118- self .assertEquals (capt_data ['webdriver' ], bm ._cache .get_connection (index ))
119- self .assertEquals (capt_data ['webdriver' ], bm ._cache .get_connection ('fake' ))
120- capt_data .clear ()
121- my_kwargs = {'some_arg' :2 }
122- bm .create_webdriver ('FakeWebDriver' , kwargs = my_kwargs )
123- self .assertEquals (capt_data ['some_arg' ], 2 )
113+ index = bm .create_webdriver ('FakeWebDriver' , 'fake1' , some_arg = 1 )
114+ self .assertEqual (bm ._current_browser (), driver )
115+ self .assertEquals (bm ._cache .get_connection (index ), driver )
116+ self .assertEquals (bm ._cache .get_connection ('fake1' ), driver )
117+ my_kwargs = {'some_arg' : 2 }
118+ bm .create_webdriver ('FakeWebDriver' , 'fake2' , kwargs = my_kwargs )
119+ self .assertEquals (bm ._current_browser (), driver )
124120 finally :
125121 del webdriver .FakeWebDriver
126122
@@ -129,7 +125,7 @@ def verify_browser(self , webdriver_type , browser_name, **kw):
129125 bm = _BrowserManagementKeywords ()
130126 old_init = webdriver_type .__init__
131127 webdriver_type .__init__ = self .mock_init
132-
128+
133129 try :
134130 self .was_called = False
135131 bm ._make_browser (browser_name , ** kw )
@@ -138,7 +134,7 @@ def verify_browser(self , webdriver_type , browser_name, **kw):
138134 finally :
139135 webdriver_type .__init__ = old_init
140136 self .assertTrue (self .was_called )
141-
137+
142138 def mock_init (self , * args , ** kw ):
143139 self .was_called = True
144140
0 commit comments