41 lines
1.1 KiB
Python
41 lines
1.1 KiB
Python
from pypom import Page, Region
|
|
from selenium.webdriver.common.by import By
|
|
|
|
|
|
class Base(Page):
|
|
"""Base object model."""
|
|
|
|
_url = '{base_url}'
|
|
_send_logo_locator = (By.CLASS_NAME, 'logo')
|
|
|
|
def __init__(self, selenium, base_url, locale='en-US', **kwargs):
|
|
super(Base, self).__init__(
|
|
selenium, base_url, locale=locale, timeout=20, **kwargs)
|
|
|
|
def wait_for_page_to_load(self):
|
|
self.wait.until(
|
|
lambda _: self.find_element(
|
|
*self._send_logo_locator).is_displayed())
|
|
return self
|
|
|
|
@property
|
|
def footer(self):
|
|
return self.Footer(self)
|
|
|
|
class Footer(Region):
|
|
_root_element = (By.CLASS_NAME, 'footer')
|
|
_legal_links = (By.CLASS_NAME, 'legalSection__link')
|
|
|
|
@property
|
|
def links(self):
|
|
return [self.Links(self, el) for el in self.find_elements(
|
|
*self._legal_links)]
|
|
|
|
class Links(Region):
|
|
|
|
@property
|
|
def name(self):
|
|
return self.root.text.split()[0]
|
|
|
|
def click(self):
|
|
self.root.click()
|