def test_ie_browser_creating(self, webdriver_create_mock,
browser_create_mock):
"""Test `ie` method.
Note:
Test passing args to create browser.
"""
selenium_config = MagicMock()
s = Selenium(selenium_config)
webdriver = webdriver_create_mock.return_value
browser_create_mock.return_value(None)
s.ie()
browser_create_mock.assert_called_once_with(s, webdriver)
def test_ie_with_lack_ie_config(self):
"""Test `ie` method.
Note:
Test to throw exception when ie config in general selenium config doesn't exist.
"""
selenium_config = {}
mock = MagicMock()
mock.get.side_effect = selenium_config.get
s = Selenium(mock)
with self.assertRaises(SeleniumExError) as context:
s.ie()
self.assertIn('settings for ie browser is not found in config',
context.exception)
def test_ie_return_browser(self, webdriver_create_mock,
browser_create_mock):
"""Test `ie` method.
Note:
Test returning browser by `ie` method.
"""
selenium_config = MagicMock()
s = Selenium(selenium_config)
browser = browser_create_mock.return_value
self.assertEqual(browser, s.ie())
def test_ie(self):
mock = MagicMock().return_value("conf")
mock.browser = drivers.IE
s = Selenium(mock)
with self.assertRaises(WebDriverException) as context:
s.ie()