def test_waitfor_wraps_its_return_value_if_given_eager_arument_is_true(
self):
mock_elem = mock.Mock(WebElement)
self.mock.find_elements_by_xpath.return_value = [mock_elem]
wrapper = SeleniumWrapper(self.mock)
self.assertIsInstance(wrapper.waitfor("xpath", "dummy", eager=True),
SeleniumContainerWrapper)
def test_wrapper_should_respond_to_waitfor(self):
mocked_driver = mock.Mock(WebDriver)
mocked_driver.find_element_by_id = lambda target: target
wrapper = SeleniumWrapper(mocked_driver)
self.assertEqual(wrapper.waitfor('id', 'hoge'), 'hoge')
def test_waitfor_wraps_its_return_value_if_given_eager_arument_is_true(self):
mock_elem = mock.Mock(WebElement)
self.mock.find_elements_by_xpath.return_value = [mock_elem]
wrapper = SeleniumWrapper(self.mock)
self.assertIsInstance(wrapper.waitfor("xpath", "dummy", eager=True), SeleniumContainerWrapper)
def test_waitfor_wraps_its_return_value_if_it_is_wrappable(self):
mock_elem = mock.Mock(WebElement)
self.mock.find_element_by_xpath.return_value = mock_elem
wrapper = SeleniumWrapper(self.mock)
self.assertIsInstance(wrapper.waitfor("xpath", "dummy"), SeleniumWrapper)
def test_wrapper_should_respond_to_waitfor(self):
mocked_driver = mock.Mock(WebDriver)
mocked_driver.find_element_by_id = lambda target: target
wrapper = SeleniumWrapper(mocked_driver)
self.assertEqual(wrapper.waitfor('id', 'hoge'), 'hoge')
def test_waitfor_wraps_its_return_value_if_it_is_wrappable(self):
mock_elem = mock.Mock(WebElement)
self.mock.find_element_by_xpath.return_value = mock_elem
wrapper = SeleniumWrapper(self.mock)
self.assertIsInstance(wrapper.waitfor("xpath", "dummy"),
SeleniumWrapper)
