from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
......
# 等待弹窗出现
WebDriverWait(self.driver, 3).until(EC.alert_is_present())
alert = self.driver.switch_to.alert
assert alert.text == expect
alert.accept()
# 等待弹窗消失
WebDriverWait(self.driver, 3).until_not(EC.alert_is_present())