def wait_page_load(self, timeout=TIMEOUT):
def _wait_page_load():
return self.is_document_ready
if Waiter.poll(timeout, _wait_page_load):
return
else:
raise TimeoutException(
"Page wasn't load after %s seconds." % timeout)
def find_element(self, id_selector, timeout=TIMEOUT):
def _is_element_present():
return self.web_driver.find_element_by_id(id_selector)
element = Waiter.poll(timeout, _is_element_present)
if element:
return element
raise TimeoutException('Element "%s" was present after %s seconds.' %
(id_selector, timeout))
