Selenium+Python Web 自动化测试：元素定位、操作模拟与断言验证

• 键盘操作（需导入 Keys）：

from selenium.webdriver.common.keys import Keys
# 输入后按回车键（替代点击按钮）
driver.find_element_by_id("kw").send_keys(Keys.ENTER)

• 鼠标悬停（需 ActionChains）：

from selenium.webdriver.common.action_chains import ActionChains
# 悬停到'设置'菜单
setting = driver.find_element_by_link_text("设置")
ActionChains(driver).move_to_element(setting).perform()

三、实战案例：模拟电商网站登录

1. 测试用例

目标：自动输入用户名/密码，点击登录，验证是否跳转至首页。

前置条件：准备测试账号（testuser/testpwd123）。

2. 代码实现

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

# 1. 启动浏览器并打开登录页
driver = webdriver.Chrome()
driver.get("https://www.example.com/login") # 替换为实际登录页 URL
driver.maximize_window() # 最大化窗口

# 2. 输入账号密码（显式等待元素加载，避免 time.sleep）
WebDriverWait(driver, 10).until(
    EC.presence_of_element_located(("id", "username"))
).send_keys("test_user")
driver.find_element_by_id("password").send_keys("test_pwd123")

# 3. 点击登录按钮
driver.find_element_by_xpath("//button[@type='submit']").click()

# 4. 验证登录结果（断言 URL 是否包含"home"）
WebDriverWait(driver, 10).until(
    EC.url_contains("home")
)
assert "home" in driver.current_url, "登录失败，未跳转至首页"

# 5. 关闭浏览器
driver.quit()

四、提升效率：避免重复劳动

1. 显式等待（替代 time.sleep）

# 等待 10 秒，直到'首页欢迎语'出现
welcome_text = WebDriverWait(driver, 10).until(
    EC.presence_of_element_located(("id", "welcome"))
)
assert "欢迎回来" in welcome_text.text

2. 封装成测试类（便于维护）

import unittest

class TestLogin(unittest.TestCase):
    def setUp(self):
        self.driver = webdriver.Chrome()

    def test_login_success(self):
        self.driver.get("https://www.example.com/login")
        # 登录逻辑...

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

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

3. 容器化部署（确保环境一致性）

将 Selenium 测试环境打包为 Docker 镜像，避免'本地能跑，服务器跑不了'的问题：

# Dockerfile 示例
FROM python:3.9
RUN pip install selenium
COPY . /app
CMD ["python", "/app/test_login.py"]