以头歌实践教学平台为例，编写登录功能测试脚本，使用selenium实现自动化测试。要求如下：（1）输入错误用户名，输入正确密码，点击登录按钮，获取错误提示信息； （2）输入正确用户名，输入错误密

在以头歌实践教学平台的登录功能自动化测试中，我们可以使用Selenium WebDriver库来模拟用户操作并验证结果。这是一个基本的Python示例，假设我们已经在环境中安装了Selenium、ChromeDriver，并且页面元素的ID或XPath已知：

from selenium import webdriver
from selenium.webdriver.common.keys import Keys
import time

# 初始化浏览器驱动
driver = webdriver.Chrome()

# 定义登录函数
def login(username, password):
    driver.get("http://example.com/login")  # 替换为实际URL
    username_input = driver.find_element_by_id("username")
    password_input = driver.find_element_by_id("password")

    # 输入用户名和密码
    username_input.send_keys(username)
    password_input.send_keys(password + Keys.RETURN)  # 加上回车键

    # 等待页面加载并检查错误消息
    error_message = None
    try:
        error_message = driver.find_element_by_xpath("//div[@class='error-message']")
    except Exception as e:
        print(f"Error finding error message: {e}")

    return error_message.text if error_message else "Login successful"

# 测试用例1：输入错误用户名
username_error = "invalid_user"
error_msg1 = login(username_error, "correct_password")
print(f"Error for invalid username: {error_msg1}")
assert "无效用户名" in error_msg1, f"Expected '无效用户名', got '{error_msg1}'"

# 测试用例2：输入正确用户名，错误密码
password_error = "wrong_password"
error_msg2 = login("valid_user", password_error)
print(f"Error for wrong password: {error_msg2}")
assert "无效密码" in error_msg2 or "密码错误" in error_msg2, f"Expected '无效密码' or '密码错误', got '{error_msg2}'"

# 关闭浏览器
driver.quit()