Selenium WebDriver- 操作JavaScript的confirm弹窗

#encoding=utf-8

import unittest

import time

from selenium import webdriver

from selenium.webdriver import ActionChains

class VisitSogouByIE(unittest.TestCase):

    def setUp(self):

        #启动IE浏览器

        #self.driver = webdriver.Firefox(executable_path = "e:\\geckodriver")

        self.driver = webdriver.Ie(executable_path = "e:\\IEDriverServer")

    def test_Handleconfirm(self):

        from selenium.common.exceptions import NoAlertPresentException

        import time

        url = "http://127.0.0.1/test_confirm.html"

        # 访问自动以测试网页

        self.driver.get(url)

        # 通过id属性值查找页面上的按钮元素

        button = self.driver.find_element_by_id("button")

        # 单击按钮元素，则会弹出一个confirm提示框，

        # 上面显示“这是一个 confirm 弹出框”、“确定”和“取消”按钮

        button.click()

        try:

            # 较高版本的selenium推荐使用driver.switch_to.alert方法代替

            # driver.switch_to_alert()方法来获取alert对象

            alert = self.driver.switch_to.alert

            time.sleep(2)

            # 使用alert.text属性获取confirm框中的内容，

            # 并断言文字内容是否是“这是一个 confirm 弹出框”

            self.assertAlmostEqual(alert.text, u"这是一个 confirm 弹出框")

            # 调用alert对象的accept()方法，模拟鼠标单击confirm弹窗上的“确定”按钮

            # 以便关闭confirm窗

            #alert.accept()

            # 取消下面一行代码的注释，就会模拟单击confirm框上的“取消”按钮

            alert.dismiss()

        except NoAlertPresentException, e:

            # 如果confirm框未弹出显示在页面上，则会抛出NoAlertPresentException的异常

            self.fail("尝试操作的confirm框未被找到")

            print e

    def tearDown(self):

        # 退出IE浏览器

        self.driver.quit()

if __name__ == '__main__':

    unittest.main()