# coding:utf-8
from selenium import webdriver

class BasePage(object):
    """
    1、第一層 - 核心層-BasePage層，定義一個所有頁面都繼承的page層
    2、對將要使用的 selenium 的底層方法進行二次封裝
    """

    def __init__(self, driver, path=None):     # 構(gòu)造函數(shù)，類的初始化
        """
        為了方便編寫將 driver 初始化，
        先使用 "self.driver = webdriver.Chrome()" 后續(xù)改為 self.driver = driver
        """
        self.driver = webdriver.Chrome()
        # self.driver = driver
        self.driver.implicitly_wait(5)  # 定義全局的默認加載時間
        self.load_page(path)            # 訪問并加載網(wǎng)頁

    def load_page(self, path=None):     # 訪問并加載網(wǎng)頁，如果 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 定位元素，并執(zhí)行 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    # 導(dǎo)入 base_page 層

class HomePage(BasePage):      # 定義 FirstPage（繼承 BasePage ）
    """
    1、第二層 - 各個頁面單獨封裝成層，頁面的元素、操作、流程
    """
    def direct_to_login(self):      # 首頁跳轉(zhuǎn)至登錄頁
        return self.by_xpath("http://*[@id='app']/div[1]/div[5]/div[3]")

    def direct_to_product(self):    # 登陸成功后，跳轉(zhuǎn)至首頁
        return self.by_xpath("http://*[@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()    # 點擊 "首頁" 跳轉(zhuǎn)至首頁

# coding:utf-8
from selenium import webdriver
from pages.base_page import BasePage    # 導(dǎo)入 base_page 層

class LoginPage(BasePage):      # 定義 FirstPage（繼承 BasePage ）
    """
    1、頁面層(登錄頁) - 各個頁面單獨封裝成層，頁面的元素、操作、流程
    """
    def login_username(self):    # 登錄頁 - 用戶名輸入框
        return self.by_xpath("http://*[@id='app']/div[1]/form/div[1]/div[2]/div/input")

    def login_password(self):    # 登錄頁 - 密碼輸入框
        return self.by_xpath("http://*[@id='app']/div[1]/form/div[2]/div[2]/div/input")

    def login_button(self):      # 登錄頁 - 登錄按鈕
        return self.by_xpath("http://*[@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    # 導(dǎo)入 base_page 層

class OrderPage(BasePage):      # 定義 FirstPage（繼承 BasePage ）
    """
    1、頁面層(登錄頁) - 各個頁面單獨封裝成層，頁面的元素、操作、流程
    """
    def product(self):    # 下單 - 第一個產(chǎn)品
        return self.by_xpath("http://*[@id='app']/div[1]/div[4]/div[2]/a[1]")

    def ticket_book(self):  # 門票 - 預(yù)定（按鈕）
        return self.by_xpath("http://*[@id='app']/div[1]/div[5]/div[2]/div[2]/a")

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

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

    def pay_off(self):      # 門票下單 - 支付
        return self.by_xpath("http://*[@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、初始化 - 打開瀏覽器，設(shè)置瀏覽器大小
2、最終操作 - 關(guān)閉瀏覽器
3、用例部分 - 登錄 與 購買操作、下訂單、支付
'''

class TestTravel(unittest.TestCase):
    @classmethod
    def setUpClass(cls):                    # 每個測試類在加載之前執(zhí)行一次 setUpClass ，初始化方法
        cls.driver = webdriver.Chrome()
        cls.driver.maximize_window()

    def test_a_order(self):
        #初始化參數(shù)
        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)

        #跳轉(zhuǎn)登錄
        home_page.cross_to_login()

        #登錄
        login_page.login(username, password)

        # 跳轉(zhuǎn)至訂單頁
        home_page.cross_to_product()

        #下單
        order_page.place_order()


    @classmethod
    def tearDownClass(cls):
        cls.driver.quit()       # 徹底退出瀏覽器


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