Skip to content

Commit 1028377

Browse files
authored
Updated mockito to 1.0.0 in dev requirements (robotframework#740)
mockito 1.0.0 did not anymore work with the BrowserManagementKeywords unit tests. Changed unit test to work with mockito 1.0.0. Unit test do not anymore work with older mockito versions.
1 parent ad5f63f commit 1028377

2 files changed

Lines changed: 18 additions & 22 deletions

File tree

requirements-dev.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,2 @@
1-
mockito == 0.7.1
1+
mockito >= 1.0.0
22
robotstatuschecker

test/unit/keywords/test_browsermanagement.py

Lines changed: 17 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -1,20 +1,20 @@
11
import unittest
22

3-
from mockito import mock, verify, verifyNoMoreInteractions
3+
from mockito import when, mock, verify, verifyNoMoreInteractions
44
from selenium import webdriver
55

66
from 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

Comments
 (0)