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

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