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()