一文详解Python中PO模式的设计与实现

# coding:utf-8
from selenium import webdriver

class BasePage(object):
    """
    1、第一层 - 核心层-BasePage层，定义一个所有页面都继承的page层
    2、对将要使用的 selenium 的底层方法进行二次封装
    """

    def __init__(self, driver, path=None):     # 构造函数，类的初始化
        """
        为了方便编写将 driver 初始化，
        先使用 "self.driver = webdriver.Chrome()" 后续改为 self.driver = driver
        """
        self.driver = webdriver.Chrome()
        # self.driver = driver
        self.driver.implicitly_wait(5)  # 定义全局的默认加载时间
        self.load_page(path)            # 访问并加载网页

    def load_page(self, path=None):     # 访问并加载网页，如果 path 不为空的话，直接传给 driver.get() 访问
        if path is not None:
            self.driver.get(path)

    def by_xpath(self, xpath):          # 二次封装 selenium 的 xpath 元素定位
        return self.driver.find_element_by_xpath(xpath)

    def js_click(self, xpath):          # JavaScript 定位元素，并执行 click
        self.driver.execute_script('arguments[0].click()', self.by_xpath(xpath))

# coding:utf-8

from selenium import webdriver
from pages.base_page import BasePage    # 导入 base_page 层

class HomePage(BasePage):      # 定义 FirstPage（继承 BasePage ）
    """
    1、第二层 - 各个页面单独封装成层，页面的元素、操作、流程
    """
    def direct_to_login(self):      # 首页跳转至登录页
        return self.by_xpath("//*[@id='app']/div[1]/div[5]/div[3]")

    def direct_to_product(self):    # 登陆成功后，跳转至首页
        return self.by_xpath("//*[@id='app']/div[1]/div[5]/div[1]")

    # 方法流程
    def cross_to_login(self):
        self.direct_to_login().click()  # 点击 "登录" 按钮进行登录

    def cross_to_product(self):
        self.direct_to_product().click()    # 点击 "首页" 跳转至首页

# coding:utf-8
from selenium import webdriver
from pages.base_page import BasePage    # 导入 base_page 层

class LoginPage(BasePage):      # 定义 FirstPage（继承 BasePage ）
    """
    1、页面层(登录页) - 各个页面单独封装成层，页面的元素、操作、流程
    """
    def login_username(self):    # 登录页 - 用户名输入框
        return self.by_xpath("//*[@id='app']/div[1]/form/div[1]/div[2]/div/input")

    def login_password(self):    # 登录页 - 密码输入框
        return self.by_xpath("//*[@id='app']/div[1]/form/div[2]/div[2]/div/input")

    def login_button(self):      # 登录页 - 登录按钮
        return self.by_xpath("//*[@id='app']/div[1]/form/div[3]/button")

    # 登录Case
    def login(self, username, password):    # 登录方法，传入 username 与 password
        self.login_username().send_keys(username)
        self.login_password().send_keys(password)
        self.login_button().click()

# coding:utf-8

from time import sleep
from pages.base_page import BasePage    # 导入 base_page 层

class OrderPage(BasePage):      # 定义 FirstPage（继承 BasePage ）
    """
    1、页面层(登录页) - 各个页面单独封装成层，页面的元素、操作、流程
    """
    def product(self):    # 下单 - 第一个产品
        return self.by_xpath("//*[@id='app']/div[1]/div[4]/div[2]/a[1]")

    def ticket_book(self):  # 门票 - 预定（按钮）
        return self.by_xpath("//*[@id='app']/div[1]/div[5]/div[2]/div[2]/a")

    def book_date(self):    # 门票 - 选择日期
        return self.by_xpath("//*[@id='app']/div[1]/form/div[1]/div[1]/div[2]/div/input")

    def to_order(self):     # 门票下单
        return self.by_xpath("//*[@id='app']/div[1]/form/div[4]/div/button")

    def pay_off(self):      # 门票下单 - 支付
        return self.by_xpath("//*[@id='app']/div[1]/form/div/div/button")

    def confirm(self):      # 门票下单 - 确认支付
        return self.by_xpath("/html/body/div[5]/div[3]/button[2]")

    # 下单成功Case
    def place_order(self):
        self.product().click()
        self.ticket_book().click()
        self.book_date().send_keys("2022-06-16")
        self.to_order().click()
        sleep(2)
        element = self.pay_off()
        self.driver.execute_script('arguments[0].click()', element)
        sleep(2)

# coding:utf-8

import unittest
from time import sleep
from selenium import webdriver
from pages.home_page import HomePage
from pages.login_page import LoginPage
from pages.order_page import OrderPage

'''
1、初始化 - 打开浏览器，设置浏览器大小
2、最终操作 - 关闭浏览器
3、用例部分 - 登录 与 购买操作、下订单、支付
'''

class TestTravel(unittest.TestCase):
    @classmethod
    def setUpClass(cls):                    # 每个测试类在加载之前执行一次 setUpClass ，初始化方法
        cls.driver = webdriver.Chrome()
        cls.driver.maximize_window()

    def test_a_order(self):
        #初始化参数
        username = '13500000001'
        password = 'Success@2020'

        #初始化界面
        home_page = HomePage(driver=self.driver, path="http://django.t.mukewang.com/#/")
        login_page = LoginPage(driver=self.driver)
        order_page = OrderPage(driver=self.driver)

        #跳转登录
        home_page.cross_to_login()

        #登录
        login_page.login(username, password)

        # 跳转至订单页
        home_page.cross_to_product()

        #下单
        order_page.place_order()


    @classmethod
    def tearDownClass(cls):
        cls.driver.quit()       # 彻底退出浏览器


if __name__ == '__main__':
    unittest.main()