def __init__(self, wait: WebDriverWait, root: WebElement):
self._wait = wait
self._title = BasePageElement(CartItemLoc.TITLE, wait=wait, root=root)
self._description = BasePageElement(CartItemLoc.DESCRIPTION,
wait=wait,
root=root)
self._price = BasePageElement(CartItemLoc.PRICE, wait=wait, root=root)
self._inv_btn = BasePageElement(CartItemLoc.BTN, wait=wait, root=root)
class Checkout:
def __init__(self, wait: WebDriverWait):
self._wait = wait
self._subtotal = BasePageElement(CheckoutLoc.SUBTOTAL_LABEL, wait=wait)
self._total = BasePageElement(CheckoutLoc.TOTAL_LABEL, wait=wait)
self._cancel_btn = BasePageElement(CheckoutLoc.CANCEL_BTN, wait=wait)
self._finish_btn = BasePageElement(CheckoutLoc.FINISH_BTN, wait=wait)
self._success_msg = BasePageElement(CheckoutLoc.SUCCESS_MSG, wait=wait)
self._title = BasePageElement(CheckoutLoc.TITLE_LABEL, wait=wait)
def cancel_checkout(self):
self._cancel_btn.click()
def finish(self):
self._finish_btn.click()
def get_subtotal(self) -> str:
return self._subtotal.get_text()
def get_total(self) -> str:
return self._total.get_text()
def get_success_msg(self) -> str:
return self._success_msg.get_text()
def get_title(self) -> str:
return self._title.get_text()
class InventoryDetailsPage(InventoryItemMixin, BasePage):
"""Implements inventory item details"""
def __init__(self, driver: WebDriver, timeout: int = 5):
super().__init__(driver, _URL, timeout)
self._title = BasePageElement(InventoryDetailsLoc.TITLE, wait=self._wait)
self._description = BasePageElement(InventoryDetailsLoc.DESCRIPTION, wait=self._wait)
self._price = BasePageElement(InventoryDetailsLoc.PRICE, wait=self._wait)
self._inv_btn = BasePageElement(InventoryDetailsLoc.BTN, wait=self._wait)
self._back_btn = BasePageElement(InventoryDetailsLoc.BACK_BTN, wait=self._wait)
self.header = Header(self._wait)
def back(self):
"""Go back to details page."""
self._back_btn.click()
def __init__(self, driver: WebDriver, timeout: int = 5):
super().__init__(driver, _URL, timeout)
self.__user = BasePageElement(LoginPageLoc.USER, wait=self._wait)
self.__password = BasePageElement(LoginPageLoc.PASSWORD,
wait=self._wait)
self.__login = BasePageElement(LoginPageLoc.LOGIN, wait=self._wait)
self.__error_msg = BasePageElement(LoginPageLoc.ERROR_MSG,
wait=self._wait)
def __init__(self, driver: WebDriver, timeout: int = 5):
super().__init__(driver, _URL, timeout)
self.header = Header(self._wait)
self.products = CartItems(CartLoc.CART_ITEMS, self._wait)
self.__title = BasePageElement(CartLoc.TITLE, self._wait)
self.__ctn_shopping = BasePageElement(CartLoc.CTN_SHOPPING, self._wait)
self.__remove_button = BasePageElement(CartLoc.REMOVE_BUTTON, self._wait)
self.__checkout_btn = BasePageElement(CartLoc.CHECKOUT_BTN, self._wait)
class InventoryItem(InventoryItemMixin):
"""Represents inventory item."""
def __init__(self, wait: WebDriverWait, root: WebElement):
self._wait = wait
self._title = BasePageElement(InventoryItemLoc.TITLE,
wait=wait,
root=root)
self._description = BasePageElement(InventoryItemLoc.DESCRIPTION,
wait=wait,
root=root)
self._price = BasePageElement(InventoryItemLoc.PRICE,
wait=wait,
root=root)
self._inv_btn = BasePageElement(InventoryItemLoc.BTN,
wait=wait,
root=root)
def open_details(self):
"""Open product details"""
self._title.click()
return InventoryDetailsPage(self._wait._driver, self._wait._timeout)
class Header:
"""Represents inventory item."""
def __init__(self, wait: WebDriverWait):
self._wait = wait
self._link = BasePageElement(HeaderLoc.LINK, wait=wait)
self._badge = BasePageElement(HeaderLoc.BADGE, wait=wait)
self._burger_btn = BasePageElement(HeaderLoc.BURGER_BTN, wait=wait)
def get_total_cart_items(self) -> int:
"""Get total items in cart"""
try:
return int(self._badge.get_text())
except Exception:
return 0
def goto_cart(self):
"""Go to cart."""
self._link.click()
def open_menu(self):
"""Open menu"""
self._burger_btn.click()
def __init__(self, driver: WebDriver, timeout: int = 5):
super().__init__(driver, _URL, timeout)
self.__search_textbox = BasePageElement(GoogleLocators.SEARCH_TEXT_BOX,
wait=self._wait)
self.__search_btn = BasePageElement(GoogleLocators.SEARCH_BTN,
wait=self._wait)
self.__feeling_lucky_btn = BasePageElement(
GoogleLocators.FEELING_LUCKY_BTN, wait=self._wait)
def __init__(self, driver: WebDriver, timeout: int = 5):
super().__init__(driver, _URL, timeout)
self._title = BasePageElement(InventoryDetailsLoc.TITLE, wait=self._wait)
self._description = BasePageElement(InventoryDetailsLoc.DESCRIPTION, wait=self._wait)
self._price = BasePageElement(InventoryDetailsLoc.PRICE, wait=self._wait)
self._inv_btn = BasePageElement(InventoryDetailsLoc.BTN, wait=self._wait)
self._back_btn = BasePageElement(InventoryDetailsLoc.BACK_BTN, wait=self._wait)
self.header = Header(self._wait)
class Google(BasePage):
"""Google base page."""
def __init__(self, driver: WebDriver, timeout: int = 5):
super().__init__(driver, _URL, timeout)
self.__search_textbox = BasePageElement(GoogleLocators.SEARCH_TEXT_BOX,
wait=self._wait)
self.__search_btn = BasePageElement(GoogleLocators.SEARCH_BTN,
wait=self._wait)
self.__feeling_lucky_btn = BasePageElement(
GoogleLocators.FEELING_LUCKY_BTN, wait=self._wait)
def search(self, value):
"""Simple search."""
self.__search_textbox.write(value)
self.__search_btn.click()
def feeling_lucky(self, value):
"""Feeling lucky search."""
self.__search_textbox.write(value)
self.__feeling_lucky_btn.click()
class LoginPage(BasePage):
"""Sauce lab login."""
def __init__(self, driver: WebDriver, timeout: int = 5):
super().__init__(driver, _URL, timeout)
self.__user = BasePageElement(LoginPageLoc.USER, wait=self._wait)
self.__password = BasePageElement(LoginPageLoc.PASSWORD,
wait=self._wait)
self.__login = BasePageElement(LoginPageLoc.LOGIN, wait=self._wait)
self.__error_msg = BasePageElement(LoginPageLoc.ERROR_MSG,
wait=self._wait)
def login(self, user, password):
"""Login to sauce lab"""
self.__user.write(user)
self.__password.write(password)
self.__login.click()
return InventoryPage(self._driver, self._timeout)
def get_error_message(self) -> str:
"""Get login error message
:return: Error message
"""
return self.__error_msg.get_text()
class CartPage(BasePage):
def __init__(self, driver: WebDriver, timeout: int = 5):
super().__init__(driver, _URL, timeout)
self.header = Header(self._wait)
self.products = CartItems(CartLoc.CART_ITEMS, self._wait)
self.__title = BasePageElement(CartLoc.TITLE, self._wait)
self.__ctn_shopping = BasePageElement(CartLoc.CTN_SHOPPING, self._wait)
self.__remove_button = BasePageElement(CartLoc.REMOVE_BUTTON, self._wait)
self.__checkout_btn = BasePageElement(CartLoc.CHECKOUT_BTN, self._wait)
def back_shopping(self):
self.__ctn_shopping.click()
def checkout(self):
self.__checkout_btn.click()
return ContactInfoPage(self._wait._driver, self._wait._timeout)
def get_title(self) -> str:
return self.__title.get_text()
def remove_item(self):
self.__remove_button.click()
def __init__(self, wait: WebDriverWait):
self._wait = wait
self._link = BasePageElement(HeaderLoc.LINK, wait=wait)
self._badge = BasePageElement(HeaderLoc.BADGE, wait=wait)
self._burger_btn = BasePageElement(HeaderLoc.BURGER_BTN, wait=wait)
def __init__(self, wait: WebDriverWait):
self._wait = wait
self._firstname = BasePageElement(ContactInfoLoc.FIRSTNAME, wait=wait)
self._lastname = BasePageElement(ContactInfoLoc.LASTNAME, wait=wait)
self._postal_code = BasePageElement(ContactInfoLoc.POSTAL_CODE,
wait=wait)
self._warning_msg = BasePageElement(ContactInfoLoc.WARNING_MSG,
wait=wait)
self._cancel_btn = BasePageElement(ContactInfoLoc.CANCEL_BTN,
wait=wait)
self._continue_btn = BasePageElement(ContactInfoLoc.CONTINUE_BTN,
wait=wait)
class ContactInfo:
"""Represent Contact Info section"""
def __init__(self, wait: WebDriverWait):
self._wait = wait
self._firstname = BasePageElement(ContactInfoLoc.FIRSTNAME, wait=wait)
self._lastname = BasePageElement(ContactInfoLoc.LASTNAME, wait=wait)
self._postal_code = BasePageElement(ContactInfoLoc.POSTAL_CODE,
wait=wait)
self._warning_msg = BasePageElement(ContactInfoLoc.WARNING_MSG,
wait=wait)
self._cancel_btn = BasePageElement(ContactInfoLoc.CANCEL_BTN,
wait=wait)
self._continue_btn = BasePageElement(ContactInfoLoc.CONTINUE_BTN,
wait=wait)
def fill_info(self, firstname="", lastname="", postal_code=""):
self._firstname.write(firstname)
self._lastname.write(lastname)
self._postal_code.write(postal_code)
def checkout(self):
self._continue_btn.click()
def back_to_cart(self):
self._cancel_btn.click()
def get_error_msg(self):
return self._warning_msg.get_text()
def __init__(self, wait: WebDriverWait):
self._wait = wait
self._subtotal = BasePageElement(CheckoutLoc.SUBTOTAL_LABEL, wait=wait)
self._total = BasePageElement(CheckoutLoc.TOTAL_LABEL, wait=wait)
self._cancel_btn = BasePageElement(CheckoutLoc.CANCEL_BTN, wait=wait)
self._finish_btn = BasePageElement(CheckoutLoc.FINISH_BTN, wait=wait)
self._success_msg = BasePageElement(CheckoutLoc.SUCCESS_MSG, wait=wait)
self._title = BasePageElement(CheckoutLoc.TITLE_LABEL, wait=wait)