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)