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文件中

以后自动化脚本中遇到弹窗均可调用该方法