def test_wait_until_first_element_is_found_none(self):
page_element = PageElement(By.ID, 'element_id')
page_element._web_element = 'mock_element'
element = self.utils.wait_until_first_element_is_found([None, page_element])
assert_equal(page_element, element)
def test_get_web_element_from_page_element(driver_wrapper, utils):
# Mock Driver.save_web_element = True
driver_wrapper.config = mock.MagicMock()
driver_wrapper.config.getboolean_optional.return_value = True
element = PageElement(By.ID, 'element_id')
element._web_element = 'mock_element'
web_element = utils.get_web_element(element)
assert 'mock_element' == web_element
def test_wait_until_first_element_is_found_none(utils):
# Mock Driver.save_web_element = True
driver_wrapper.config = mock.MagicMock()
driver_wrapper.config.getboolean_optional.return_value = True
page_element = PageElement(By.ID, 'element_id')
page_element._web_element = 'mock_element'
element = utils.wait_until_first_element_is_found([None, page_element])
assert page_element == element
def test_get_web_element_from_page_element(self):
element = PageElement(By.ID, 'element_id')
element._web_element = 'mock_element'
web_element = self.utils.get_web_element(element)
assert_equal('mock_element', web_element)
def test_get_web_element_from_page_element(self):
element = PageElement(By.ID, 'element_id')
element._web_element = 'mock_element'
web_element = self.utils.get_web_element(element)
assert_equal('mock_element', web_element)
