[파이썬] Selenium 페이지 로딩 대기 (Implicit Wait)

Selenium is a popular automation tool for web testing. When writing Selenium test scripts, one important aspect to consider is waiting for the page to load fully before performing any actions on the elements. This ensures that the necessary elements are present on the page and ready to be interacted with.

Why do we need to wait for the page to load?

Web pages can have different loading times depending on various factors like network speed, server response time, and complexity of the page itself. If we don’t wait for the page to load, there is a risk of trying to interact with elements that are not yet accessible or visible, resulting in errors and flaky tests.

Implicit Wait in Selenium

Selenium provides the concept of “implicit wait” to handle page loading delays. Implicit wait instructs the web driver to wait for a certain amount of time before throwing a NoSuchElementException if an element is not found immediately. By setting this parameter, we can ensure that Selenium waits for a certain amount of time for the elements to appear on the page before giving up.

Usage of Implicit Wait in Python

To use implicit wait in Selenium with Python, we need to first import the WebDriverWait and ExpectedConditions classes from the selenium.webdriver.support.ui module.

from selenium import webdriver
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC

Once we have imported the necessary classes, we can create an instance of the WebDriverWait class and set the desired timeout value.

driver = webdriver.Chrome()
wait = WebDriverWait(driver, 10) # Set the timeout to 10 seconds

Now, we can use the until method of the WebDriverWait instance along with an expected condition to wait for an element to be visible or present on the page.

element = wait.until(EC.visibility_of_element_located((By.ID, "myElement")))

In the above example, we wait for the element with ID “myElement” to become visible on the page. If the element is not found within the specified timeout (10 seconds in this case), a TimeoutException will be raised.

Conclusion

By using the implicit wait feature in Selenium, we can ensure that our test scripts wait for the page to load fully before interacting with the elements. This helps in creating more stable and robust automation tests. Remember to set an appropriate timeout value based on the expected loading time of your web page. Happy web testing with Selenium!