python自动化测试selenium核心技术处理弹框

from selenium import webdriver
from time import sleep
from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.support import expected_conditions as ec 
class TestWaitCondition(object):
    def setup(self):
        self.driver = webdriver.Chrome()
        self.driver.get(https://sahitest.com/demo/) 
    def test_alert(self):
        #点页面上的Alert Test
        self.driver.find_element_by_xpath("/html/body/table/tbody/tr/td[3]/a[1]").click()
        #鼠标点击页面中“Click For Alert”按钮
        self.driver.find_element_by_name("b1").click()
         wait = WebDriverWait(self.driver,2)
        #等待并判断警告框是否存在
        wait.until(ec.alert_is_present())
        #切换到alert页面
        alert = self.driver.switch_to.alert
        #获取并打印警告框中文本
        print(alert.text)
        #点弹出警告框中的确定
        alert.accept()
        self.driver.quit() 
    def test_confirm(self):
        #点击Confirm Page链接打开页面
        self.driver.find_element_by_xpath("/html/body/table/tbody/tr/td[3]/a[2]").click()
        #点击Click For Confirm按钮，弹出确认提示框
        self.driver.find_element_by_name("b1").click()
        #切换到警告框
        confirm=self.driver.switch_to.alert
        #获取警告框提示信息并打印
        print(confirm.text)
        #点确定
        # confirm.accept()
        sleep(2)
        #点取消
        confirm.dismiss()
        sleep(2)
        self.driver.quit()
    def test_prompt(self):
        # 点击Prompt Page链接打开页面
        self.driver.find_element_by_xpath("/html/body/table/tbody/tr/td[3]/a[3]").click()
        #点击Click For Prompt按钮，弹出确认提示框
        self.driver.find_element_by_name("b1").click()
        sleep(2)
        #切换到提示框
        prompt = self.driver.switch_to.alert
        #获取提示框中文本并打印
        print(prompt.text)
        #提示框中输入文本test prompt
        prompt.send_keys("test prompt")
        # 点确定
        prompt.accept()
        sleep(2)
        #点取消
        # prompt.dismiss()
        self.driver.quit()