Python使用Selenium进行Web自动化测试详解

pip install selenium

from selenium import webdriver  
  
# 创建Chrome浏览器的WebDriver实例  
driver = webdriver.Chrome()  
  
# 打开网页  
driver.get('https://www.example.com')  
  
# 关闭浏览器  
driver.quit()

# 使用ID定位  
element = driver.find_element_by_id("element_id")  
  
# 使用XPath定位  
element = driver.find_element_by_xpath("//div[@class='example']")  
  
# 使用CSS选择器定位  
element = driver.find_element_by_css_selector(".example")  
  
# 使用类名定位（注意：这在新版Selenium中已不推荐使用）  
# element = driver.find_element_by_class_name("example")  
  
# 使用标签名定位  
element = driver.find_element_by_tag_name("div")

# 输入文本  
element.send_keys("Hello, Selenium!")  
  
# 点击按钮  
button = driver.find_element_by_id("submit_button")  
button.click()  
  
# 获取元素文本  
text = element.text

from selenium.webdriver.common.by import By  
from selenium.webdriver.support.ui import WebDriverWait  
from selenium.webdriver.support import expected_conditions as EC  
  
# 显式等待  
element = WebDriverWait(driver, 10).until(  
    EC.presence_of_element_located((By.ID, "element_id"))  
)  
  
# 隐式等待（设置全局等待时间，对后续所有find_element方法生效）  
driver.implicitly_wait(10)  # 设置最大等待时间为10秒

from selenium import webdriver  
from selenium.webdriver.common.keys import Keys  
from selenium.webdriver.common.by import By  
from selenium.webdriver.support.ui import WebDriverWait  
from selenium.webdriver.support import expected_conditions as EC  
  
def test_login(driver, valid_username, valid_password):  
    # 打开登录页面  
    driver.get('https://www.example.com/login')  
  
    # 正确的用户名和密码  
    test_cases = [  
        (valid_username, valid_password, True),  # 预期登录成功  
        (valid_username, 'wrong_password', False),  # 预期显示错误  
        ('wrong_username', valid_password, False),  # 假设系统不允许这种情况，但为演示  
        ('', '', False)  # 空的用户名和密码  
    ]  
  
    for username, password, expect_success in test_cases:  
        # 清除输入框内容（如果有的话）  
        driver.find_element_by_id('username').clear()  
        driver.find_element_by_id('password').clear()  
  
        # 输入用户名和密码  
        driver.find_element_by_id('username').send_keys(username)  
        driver.find_element_by_id('password').send_keys(password)  
  
        # 点击登录按钮  
        driver.find_element_by_id('login_button').click()  
  
        # 根据预期结果验证  
        if expect_success:  
            # 预期登录成功，验证是否跳转到主页（这里以URL变化为例）  
            assert "https://www.example.com/home" in driver.current_url, "登录成功但未跳转到主页"  
        else:  
            # 预期登录失败，验证是否显示错误提示  
            try:  
                error_message = WebDriverWait(driver, 5).until(  
                    EC.presence_of_element_located((By.ID, 'error_message'))  
                )  
                assert "error" in error_message.text.lower(), "登录失败但未显示错误提示"  
            except Exception as e:  
                print(f"登录失败验证失败: {e}")  
                assert False, "登录失败但未捕获到错误提示"  
  
        # 如果需要，可以在这里添加代码来重置测试环境（如清除cookies、重定向到登录页面等）  
  
# 实例化WebDriver  
driver = webdriver.Chrome()  
  
# 调用测试函数，假设有有效的用户名和密码  
test_login(driver, 'valid_user', 'valid_password')  
  
# 关闭浏览器  
driver.quit()