smartbear_cart_popup.py

from selenium import webdriver
from selenium.common import TimeoutException
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
"""The class is for the PopUp cart window on the right side of the screen"""
class CartPopUp:
    def __init__(self,driver:webdriver.Edge):
        self.driver = driver

    def cart_total_items_amount(self):
        """Return the total number of items in the cart."""
        try:
            element = WebDriverWait(self.driver, 5).until(
                EC.presence_of_element_located(
                    (By.CSS_SELECTOR, '#cart-tab > span.badge.badge-pill.label-cart-amount.badge-warning'))
            )
            # Additional wait for text content to be updated
            WebDriverWait(self.driver, 2).until(lambda driver: element.text != '')
            return element.text
        except TimeoutException:
            return '0'  # Returning '0' to indicate the cart is empty

    def cart_product_name(self):
        """Locate and return all product names in the cart popup."""
        return self.driver.find_elements(By.CSS_SELECTOR, "div.col.col-data > a")

    def cart_product_price(self):
        """Locate and return all product prices in the cart popup."""
        return self.driver.find_elements(By.CLASS_NAME, "price.unit-price")

    def cart_product_quantity(self):
        """Locate and return all product quantities in the cart popup."""
        elements = WebDriverWait(self.driver, 5).until(
            EC.presence_of_all_elements_located((By.ID, "item_EnteredQuantity")))
        return elements

    def find_subtotal(self):
        subtotal_element = WebDriverWait(self.driver, 5).until(
            EC.presence_of_element_located((By.CSS_SELECTOR, ".sub-total.price"))  # Ensure selector is correct
        )
        return subtotal_element

    def go_to_cart(self):
        cart = WebDriverWait(self.driver, 5).until(
            EC.presence_of_element_located((By.CSS_SELECTOR, 'a.btn.btn-success'))
        )
        self.driver.execute_script("arguments[0].click();", cart)

    def go_to_checkout(self):
        checkout = WebDriverWait(self.driver, 5).until(EC.presence_of_element_located((By.CSS_SELECTOR, 'a.btn.btn-clear.btn-block.btn-action > span')))
        self.driver.execute_script("arguments[0].click();", checkout)