如何使用WebDriverWait.until增加代碼穩定性和可靠性

一、WebDriverWait介紹

WebDriverWait是Selenium庫中的一個工具，用於在特定條件下等待DOM元素加載。在使用Selenium構建UI自動化測試時，WebDriverWait被廣泛使用，它可以等待一個特定的頁面元素被加載，這樣我們就可以在該元素存在時執行其他的操作。WebDriverWait類似於Thread.sleep，但是它可以設定時間限制。

二、使用WebDriverWait.until的好處

即使頁面加載正常，但是仍有可能出現頁面元素未加載完成的情況，這是因為頁面加載順序和速度不可預測。WebDriverWait.until方法可以增加代碼的穩定性和可靠性，使得我們在頁面元素確實完全加載完成之前，程序不會執行下一步操作。

三、如何使用WebDriverWait.until

在使用WebDriverWait.until方法時，我們需要提供一個期望的條件，一旦這個條件被滿足，程序就會繼續執行。這裡有兩個例子，分別是等待一個指定的元素被加載和等待頁面完全加載。

1. 等待指定元素被加載

from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.common.by import By

driver = webdriver.Chrome()
driver.get("https://www.example.com")

element = WebDriverWait(driver, 10).until(
    EC.presence_of_element_located((By.ID, "example_element_id"))
)

這個例子中，我們首先使用WebDriverWait方法，並將WebDriver對象和等待時間傳遞給它，然後傳遞一個期望的條件給until方法。在這裡，期望的條件是一個元素出現在頁面上。直到該元素出現，程序只會等待最多10秒鐘。

2. 等待頁面完全加載

from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.common.by import By
from selenium.webdriver.support import expected_conditions as EC

driver = webdriver.Chrome()
driver.get("https://www.example.com")

WebDriverWait(driver, 10).until(EC.title_contains("Example Domain"))

在這個例子中，我們使用WebDriverWait.until等待完全加載頁面。我們使用了EC.title_contains方法作為條件，這個方法會等待頁面標題中包含”Example Domain”的字符。直到包含這個字符，程序最多只會等待10秒。

四、小結

在使用Selenium測試工具時，我們經常需要等待一個頁面加載或元素出現。這個過程中，我們使用WebDriverWait.until方法增加了代碼的可靠性和穩定性。在等待過程中，我們需要指定期望的條件，一旦期望被滿足，程序將會繼續執行。