Selenium校验URL包含指定单词失败问题排查求助

核心问题分析 #

1. 硬等待不稳定：使用time.sleep(2)依赖固定时长等待URL更新，遇到网络延迟或页面加载缓慢时，URL还未完成跳转就执行断言，直接导致校验失败。
2. 搜索框定位精度不足：用input[type="text"]可能匹配页面上多个文本输入框，不如通过搜索框专属的name属性定位更可靠。

修正后的代码 #

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

class MyTestCase(unittest.TestCase):
    def setUp(self):
        self.browser = webdriver.Chrome()
        self.addCleanup(self.browser.quit)

    def test_search_bar(self):
        self.browser.get('http://www.google.com')

        # 用name属性精准定位Google搜索框，比CSS选择器更稳定
        wait = WebDriverWait(self.browser, 10)
        search_bar = wait.until(EC.element_to_be_clickable((By.NAME, 'q')))

        search_bar.send_keys('Amazon.com' + Keys.ENTER)
        self.assertIn('Amazon.com', search_bar.get_attribute('value'))

        # 显式等待URL包含指定关键词，替代不稳定的硬等待
        wait.until(EC.url_contains('Amazon'))
        
        # 执行URL断言
        self.assertIn('Amazon', self.browser.current_url)

if __name__ == '__main__':
    unittest.main(verbosity=2)

关键优化点说明 #

• 替换硬等待为显式等待：使用EC.url_contains('Amazon')让Selenium主动等待URL更新到包含目标关键词，彻底避免因网络波动导致的断言失败。
• 优化元素定位：Google搜索框的标准name属性为q，用By.NAME, 'q'定位比CSS选择器更精准，减少元素定位失败的概率。
• 保留原有断言逻辑：原URL校验逻辑本身无问题，只要等待机制到位就能正常生效。

内容的提问来源于stack exchange，提问作者Reem Khattab