Python中如何编写一个基于Selenium的自动化测试程序？

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

# 1. 初始化浏览器驱动
driver = webdriver.Chrome()  # 需要先安装对应浏览器的驱动

# 2. 打开目标网页
driver.get("https://www.baidu.com")

# 3. 定位元素并操作
try:
    # 等待搜索框加载完成
    search_box = WebDriverWait(driver, 10).until(
        EC.presence_of_element_located((By.ID, "kw"))
    )
    
    # 输入搜索关键词
    search_box.send_keys("Selenium自动化测试")
    
    # 模拟点击搜索按钮
    search_button = driver.find_element(By.ID, "su")
    search_button.click()
    
    # 等待搜索结果加载
    WebDriverWait(driver, 10).until(
        EC.presence_of_element_located((By.CLASS_NAME, "result"))
    )
    
    # 获取搜索结果标题
    results = driver.find_elements(By.CSS_SELECTOR, "h3.t a")
    print(f"找到 {len(results)} 条结果：")
    for i, result in enumerate(results[:5], 1):
        print(f"{i}. {result.text}")
    
    # 4. 截图保存
    driver.save_screenshot("search_results.png")
    
except Exception as e:
    print(f"执行过程中出现错误: {e}")
    
finally:
    # 5. 关闭浏览器
    time.sleep(2)  # 等待2秒观察结果
    driver.quit()

核心步骤说明：

1. 安装必要组件：pip install selenium 并下载对应浏览器驱动
2. 初始化WebDriver创建浏览器实例
3. 使用定位器（ID、CSS选择器、XPath等）找到页面元素
4. 通过send_keys()、click()等方法模拟用户操作
5. 使用显式等待确保元素加载完成
6. 最后记得调用quit()释放资源

建议用pytest组织测试用例更规范。