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)