python - 使用变量而不是文字时出现断言错误

Question

我正在使用 Selenium WebDriver 和 Python。我有如下代码。当我使用 assert <some variable> in driver.page_source 时，为什么会出现断言错误？ ？如果我使用文字(例如字符串)而不是变量，测试就会通过。

import unittest
from selenium import webdriver
from selenium.webdriver.common.keys import Keys
import sys
from instream import InStream
from outstream import OutStream
import stdio

class PythonOrgSearch(unittest.TestCase):

    def setUp(self):
        self.driver = webdriver.Firefox()

    def test_search_in_python_org(self):
        driver = self.driver
        s = 'Ireland'
        driver.get("http://www.python.org")
        self.assertIn("Python", driver.title)
        elem = driver.find_element_by_name("q")
        elem.send_keys("pycon")
        elem.send_keys(Keys.RETURN)
        assert s in driver.page_source

    def tearDown(self):
        self.driver.close()       

if __name__ == "__main__":
    unittest.main()

Answer 1

这是一个计时问题(也复制了它)。该断言是在搜索结果尚未加载时做出的，因此 Ireland 尚未出现在 page_source 中。通过 WebDriverWait 添加等待使测试更加可靠:

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


driver = webdriver.Firefox()

s = 'Ireland'
driver.get("http://www.python.org")
wait = WebDriverWait(driver, 10)

elem = driver.find_element_by_name("q")
elem.send_keys("pycon")
elem.send_keys(Keys.RETURN)

# wait for results to be visible
wait.until(EC.visibility_of_element_located((By.CSS_SELECTOR, ".list-recent-events li")))

assert s in driver.page_source

请注意，.list-recent-events li 是一个 CSS 选择器，它与结果事件列表中的第一个事件相匹配。