Selenium - 在每行代码之后等待

时间:2014-04-14 19:16:30

标签: python selenium webdriver

我正在使用Selenium(python)来测试Web应用程序。但是,有时一行代码会触发错误,导致第一行没有足够的时间来运行。例如:

...
driver.find_element_by_id("form_widget_date").click()
driver.find_element_by_link_text(str(self.day)).click()
...

结果,第二行(有时)找不到link_text,因为Selenium显然没有时间完成第一行。当我在线之间进行睡眠时,不会发生错误。

我的问题是:有没有办法为每行代码自动等待一段时间而不是像这样解决它:

...
time.sleep(2)
driver.find_element_by_id("form_widget_date").click()
time.sleep(2)
driver.find_element_by_link_text(str(self.day)).click()
time.sleep(2) 
...

3 个答案:

答案 0 :(得分:8)

您想要做的事情基本上有三种选择。

  1. 使用time.sleep(...)

    这是现实世界发展最糟糕的选择。这很糟糕,因为你保证每次等待 你指定的时间如果您发现有时需要5秒钟才能通过测试,那么如果您等待的条件在1秒内准备好,您仍然需要等待5秒钟。在一个实质性的测试套件中,这些等待将累计并增加运行套件所需的总时间。

  2. 使用Selenium的隐式等待。

    优点是您可以设置隐式等待而忘记它。在等待之后,您不必在等待之后指定等待。但问题是隐式等待不会与显式等待混合。在this answer中很好地解释了它们不混合的原因的细节。根据我使用隐式等待的经验,仅在非常简单的情况下才能正常工作。

  3. 使用Selenium的明确等待。

    如果您的应用程序属于动态类型,如果您有Ajax,如果您动态修改DOM,您将不得不迟早使用显式等待,并在此时你将不再依赖隐式等待。缺点是你的代码会更加冗长。您可以使用包装函数来缓解它。假设driver设置为WebDriver对象,以下是您可以执行此操作的示例:

    from selenium.webdriver.common.by import By
    from selenium.webdriver.support.ui import WebDriverWait
    
    
    class MyDriver(object):
        def __init__(self, driver, default_timeout=2):
            self.driver = driver
            self.timeout = default_timeout
    
        def find_element(self, locator):
            return WebDriverWait(self.driver, self.timeout).until(
                EC.presence_of_element_located(locator))
    
    mydriver = MyDriver(driver)
    foo = mydriver.find_element((By.CSS_SELECTOR, ".foo"))
    

答案 1 :(得分:2)

有两种等待 explicit waitsimplicit waits

Dan提供的解决方案是一个显式等待的示例,我认为您正在寻找的是隐式等待。

  

隐式等待是告诉WebDriver对DOM进行轮询   尝试查找一个或多个元素的时间量   没有立即可用。默认设置为0.一旦设置,   隐式等待是为WebDriver对象实例的生命周期设置的。

您可以在开头定义一次隐式等待时间,如下所示:

driver = webdriver.Firefox()
driver.implicitly_wait(10) # seconds

参考:Explicit and Implicit waits

答案 2 :(得分:1)

现实情况是,你的代码经常会以很多等待结束。你确实想避免使用睡眠思想。看一下显式等待的使用。这允许设定的时间段等待或执行将超时,但是当找到该元素时,无论在等待期间还剩多少时间,它将自动更多地开启。

来自selenium HQ文档:

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait # available since 2.4.0
from selenium.webdriver.support import expected_conditions as EC # available since 2.26.0

ff = webdriver.Firefox()
ff.get("http://somedomain/url_that_delays_loading")
try:
    element = WebDriverWait(ff, 10).until(EC.presence_of_element_located((By.ID, "myDynamicElement")))
finally:
    ff.quit()