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