selenium-彈窗操作（八）

本次以筆者公告欄的 打賞 彈窗為例

對彈窗中的一些操作進行封裝后，在測試中使用

作用：減少對彈窗反復操作時進行定位的麻煩，以后使用中都直接調用即可達到目的

# coding=utf-8

from selenium import webdriver
import unittest, time

class dialogManu(unittest.TestCase):


    def dialog(self):
        u"定位dialog"
        dialog = self.dr.find_element_by_class_name('shang_box')
        return dialog

    def dialog_close(self):
        u"關閉彈窗"
        dialog = dialogManu.dialog(self)
        close = dialog.find_element_by_class_name('shang_close').click()
        return close

    def dialog_select_pay(self,text):
        u"彈窗中支付選則"
        dialog = dialogManu.dialog(self)
        pay_items = dialog.find_elements_by_class_name('pay_item')

        if text == "微信":
            pay_select = pay_items[1].click()
        elif text == "支付寶":
            pay_select = pay_items[0].click()
        else:
            print("沒有找到" + text)

        return pay_select


    def test_dialog(self):
        u"testcase"
        self.dr = webdriver.Chrome()
        self.dr.maximize_window()
        url = "https://www.cnblogs.com/tynam/p/10260021.html"
        self.dr.get(url)
        self.dr.implicitly_wait(10)

        # 進入到支付dialog
        self.dr.find_element_by_class_name('dashang').click()
        time.sleep(1)

        dialog = dialogManu.dialog(self)
        # 選則微信支付
        dialogManu.dialog_select_pay(self, "微信")

        # 關閉dialog
        dialogManu.dialog_close(self)

        # 關閉瀏覽器
        self.dr.close()

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

 

這而暫時只是對單個彈窗進行封裝的處理

在實際項目中可以對項目中所有的彈窗進行封裝，都封裝在該py文件中

以后自動化腳本中遇到彈窗均可調用該方法