1. 项目概述与核心价值最近在带团队做自动化测试项目发现很多同学对UI自动化的理解还停留在“录制回放”的初级阶段脚本写出来脆弱不堪维护成本极高。正好手头有个需求需要验证我们产品集成的第三方邮箱登录功能我就顺手用网易邮箱的登录流程作为案例搭建了一个基于Python、Selenium和unittest的自动化框架。这不仅仅是一个“自动登录”脚本更是一个完整的、可复用的、具备良好工程化结构的UI自动化解决方案。对于想从脚本小子进阶为自动化工程师的朋友来说理解这套框架的设计思路远比单纯复制代码更有价值。它能帮你处理元素定位的稳定性、测试用例的组织管理、失败场景的自动截图与日志记录最终实现低成本、高效益的自动化回归验证。2. 环境准备与核心工具选型解析工欲善其事必先利其器。一个稳定的自动化环境是成功的一半。很多人卡在环境配置上其实只要理清依赖关系一步步来并不复杂。2.1 Python环境与包管理首先你需要一个Python环境。我强烈建议使用Python 3.7及以上版本因为很多现代库对旧版本的支持已经减弱。新手可以直接去Python官网下载安装包安装时务必勾选“Add Python to PATH”这样可以省去后续手动配置环境变量的麻烦。安装完成后打开命令行Windows是CMD或PowerShellMac/Linux是Terminal输入python --version确认安装成功。接下来是包管理工具pip通常随Python一起安装用pip --version检查。自动化测试项目强烈建议使用虚拟环境virtual environment它能将项目的依赖包与系统全局的Python环境隔离避免版本冲突。创建虚拟环境很简单# 在当前目录下创建名为 venv 的虚拟环境 python -m venv venv # 激活虚拟环境 # Windows: venv\Scripts\activate # Mac/Linux: source venv/bin/activate激活后命令行提示符前会出现(venv)标识表示你已进入隔离环境。2.2 Selenium与WebDriver的“黄金搭档”Selenium是一个用于Web应用程序测试的强大工具集它本身是一套API通过selenium包提供而真正操控浏览器的则是WebDriver。这是最容易混淆的点。安装Selenium库在激活的虚拟环境中运行pip install selenium。下载并配置WebDriver这是关键一步。WebDriver是一个独立的可执行文件需要与你将要使用的浏览器版本严格匹配。Chrome去ChromeDriver官网下载对应你Chrome浏览器版本的驱动。查看Chrome版本在浏览器地址栏输入chrome://version/找到“版本”信息。将WebDriver放入PATH下载后将chromedriver.exe(Windows) 或chromedriver(Mac/Linux) 文件放在一个目录下并将该目录添加到系统的环境变量PATH中。更简单的做法是在代码中指定WebDriver的绝对路径我们后面会采用这种方式更清晰。注意浏览器会自动更新但WebDriver不会。如果某天你的脚本突然报错“无法启动浏览器”十有八九是浏览器升级后与WebDriver版本不匹配了记得重新下载匹配的版本。2.3 unittestPython自带的测试框架unittest是Python标准库中的单元测试框架我们用它来组织测试用例。它不需要额外安装提供了测试用例TestCase、测试套件TestSuite、测试夹具setUp/tearDown等概念非常适合结构化我们的自动化脚本。虽然现在pytest更流行但unittest对于理解测试框架的基本概念和入门UI自动化来说更加直观和标准。3. 框架设计与项目结构规划直接写一个.py文件把所有代码堆进去是初学者常犯的错误。这样的脚本难以维护、无法复用、出错难排查。我们需要一个清晰的工程结构。下面是我为这个项目设计的目录结构你可以直接套用netease_email_auto_login/ ├── config/ # 配置文件目录 │ └── config.ini # 存放URL、账号等敏感信息 ├── drivers/ # WebDriver存放目录 │ └── chromedriver.exe # Chrome浏览器驱动 ├── logs/ # 日志文件目录自动生成 ├── reports/ # 测试报告目录自动生成 ├── screenshots/ # 失败截图目录自动生成 ├── test_cases/ # 测试用例目录 │ ├── __init__.py │ └── test_login.py # 登录功能测试用例 ├── utils/ # 工具类目录 │ ├── __init__.py │ ├── logger.py # 日志记录模块 │ └── base_page.py # 页面基类 └── run_tests.py # 测试执行入口文件这样设计的好处配置与代码分离账号密码、URL等写在config.ini里不会泄露在代码中也方便不同环境切换。驱动集中管理所有驱动放在drivers下路径清晰。结果自动归档日志、报告、截图自动按时间或用例归类便于回溯。代码复用性高将浏览器操作、日志记录等通用功能抽象到utils下用例脚本只关注业务逻辑。用例独立每个测试用例文件独立方便管理和并行执行。4. 核心模块实现与代码逐行解析有了结构我们来填充核心代码。我会从底层工具类开始向上构建。4.1 日志记录模块utils/logger.py日志是自动化测试的“黑匣子”没有日志的脚本调试起来如同盲人摸象。我们设计一个既能输出到控制台又能保存到文件的日志器。import logging import os from datetime import datetime class Logger: def __init__(self, name__name__): # 创建logger实例 self.logger logging.getLogger(name) self.logger.setLevel(logging.DEBUG) # 设置最低日志级别 # 避免重复添加handler导致日志重复输出 if not self.logger.handlers: # 定义日志格式 formatter logging.Formatter(%(asctime)s - %(name)s - %(levelname)s - %(message)s) # 创建控制台handler并设置级别 ch logging.StreamHandler() ch.setLevel(logging.INFO) ch.setFormatter(formatter) self.logger.addHandler(ch) # 创建文件handler并设置级别 # 确保logs目录存在 log_dir os.path.join(os.path.dirname(os.path.dirname(__file__)), logs) if not os.path.exists(log_dir): os.makedirs(log_dir) log_file os.path.join(log_dir, fautotest_{datetime.now().strftime(%Y%m%d)}.log) fh logging.FileHandler(log_file, encodingutf-8) fh.setLevel(logging.DEBUG) # 文件里记录更详细的DEBUG信息 fh.setFormatter(formatter) self.logger.addHandler(fh) def get_logger(self): return self.logger # 提供一个全局的日志器实例 logger Logger(__name__).get_logger()关键点解析setLevel我们设置控制台只输出INFO及以上级别INFO, WARNING, ERROR, CRITICAL的日志避免刷屏而文件则记录所有DEBUG及以上的日志便于详细排查。FileHandler使用encodingutf-8这是为了防止中文日志内容乱码。检查logs目录是否存在并创建增强代码的健壮性。通过判断logger.handlers是否为空来避免重复添加handler这是一个重要的技巧防止在多次实例化Logger时日志重复打印。4.2 页面基类与通用操作封装utils/base_page.py这是框架的精华部分。我们将Selenium的常用操作如查找元素、点击、输入等进行二次封装并加入显式等待、日志记录和失败截图功能形成所有页面对象的“父类”。from selenium import webdriver from selenium.webdriver.support.ui import WebDriverWait from selenium.webdriver.support import expected_conditions as EC from selenium.common.exceptions import TimeoutException, NoSuchElementException import os import time from utils.logger import logger class BasePage: def __init__(self, driver: webdriver.Chrome): self.driver driver self.timeout 10 # 默认显式等待超时时间 def find_element(self, locator, timeoutNone): 查找单个元素加入显式等待和日志 if timeout is None: timeout self.timeout try: logger.info(f正在查找元素: {locator}) element WebDriverWait(self.driver, timeout).until( EC.presence_of_element_located(locator) ) logger.info(f元素查找成功: {locator}) return element except TimeoutException: error_msg f查找元素超时: {locator} logger.error(error_msg) self._take_screenshot(element_not_found) raise TimeoutException(error_msg) def click(self, locator): 点击元素 element self.find_element(locator) try: logger.info(f点击元素: {locator}) element.click() except Exception as e: error_msg f点击元素失败: {locator}, 错误: {e} logger.error(error_msg) self._take_screenshot(click_failed) raise def input_text(self, locator, text): 向元素输入文本 element self.find_element(locator) try: logger.info(f向元素 {locator} 输入文本: {text}) element.clear() # 先清空避免残留内容 element.send_keys(text) except Exception as e: error_msg f输入文本失败: {locator}, 错误: {e} logger.error(error_msg) self._take_screenshot(input_failed) raise def get_text(self, locator): 获取元素的文本内容 element self.find_element(locator) try: text element.text logger.info(f获取元素 {locator} 的文本: {text}) return text except Exception as e: error_msg f获取文本失败: {locator}, 错误: {e} logger.error(error_msg) raise def _take_screenshot(self, name): 内部方法失败时截图 screenshot_dir os.path.join(os.path.dirname(os.path.dirname(__file__)), screenshots) if not os.path.exists(screenshot_dir): os.makedirs(screenshot_dir) timestamp time.strftime(%Y%m%d_%H%M%S) file_path os.path.join(screenshot_dir, f{name}_{timestamp}.png) self.driver.save_screenshot(file_path) logger.info(f截图已保存至: {file_path})封装的价值显式等待find_element方法内置了显式等待WebDriverWait这是解决因网络或JS加载导致元素找不到问题的核心手段。它比time.sleep()更智能会在条件满足时立即返回而不是死等固定时间。统一日志每个关键操作都记录了日志运行脚本时你能清晰看到执行到了哪一步。自动截图在元素查找失败或操作异常时自动截取当前页面屏幕并保存到screenshots目录文件名包含时间戳。这是定位UI自动化问题最直接的证据。代码复用所有具体的页面类如登录页继承自BasePage就可以直接使用click(),input_text()这些稳定可靠的方法无需再写冗长的WebDriverWait代码。4.3 配置文件读取config.ini我们将可变参数配置化。在config/config.ini中写入[NETEASE_EMAIL] url https://mail.163.com/ username your_test_account163.com password your_test_password [BROWSER] driver_path drivers/chromedriver.exe headless False # 是否无头模式运行True为不打开浏览器界面重要警告千万不要将真实的账号密码提交到Git等版本控制系统.gitignore文件里要忽略config.ini。实际项目中敏感信息应通过环境变量或密钥管理服务传入。4.4 页面对象模型POM实现登录页面POM模式是UI自动化的最佳实践之一它将页面元素定位和页面操作封装成类使测试脚本用例更简洁元素定位变更时只需修改页面类。在test_cases/目录下我们其实更应该创建一个page_objects/目录来存放页面类。为了简化我们先在用例文件中实现。我们来创建test_cases/login_page.pyfrom selenium.webdriver.common.by import By from utils.base_page import BasePage class LoginPage(BasePage): # 页面元素定位器 (Locators) # 使用 (By.策略, 定位表达式) 的元组形式 SWITCH_TO_ACCOUNT_LOGIN (By.ID, switchAccountLogin) # “账号登录”切换链接 USERNAME_INPUT (By.NAME, email) # 用户名输入框 PASSWORD_INPUT (By.NAME, password) # 密码输入框 SUBMIT_BUTTON (By.ID, dologin) # 登录按钮 ERROR_MSG_SPAN (By.CSS_SELECTOR, .error-tt) # 错误信息提示示例 def __init__(self, driver): super().__init__(driver) def switch_to_account_login(self): 切换到账号密码登录方式网易邮箱默认可能是二维码 self.click(self.SWITCH_TO_ACCOUNT_LOGIN) return self # 返回自身支持链式调用 def enter_username(self, username): 输入用户名 self.input_text(self.USERNAME_INPUT, username) return self def enter_password(self, password): 输入密码 self.input_text(self.PASSWORD_INPUT, password) return self def click_submit(self): 点击登录按钮 self.click(self.SUBMIT_BUTTON) # 点击后页面会跳转通常返回下一个页面的对象这里我们简单处理 # 实际项目中可以返回 InboxPage 之类的对象 def get_error_message(self): 获取错误提示信息用于登录失败断言 try: return self.get_text(self.ERROR_MSG_SPAN) except: return # 如果没有找到错误元素说明可能登录成功返回空字符串 def login(self, username, password): 完整的登录流程 self.switch_to_account_login().enter_username(username).enter_password(password).click_submit()POM的优势可维护性如果登录页面的输入框ID从email改成了username你只需要修改USERNAME_INPUT这一个常量的值所有用到它的测试用例都无需改动。可读性在测试用例中你看到的是login_page.enter_username(xxx)这样清晰的业务语句而不是晦涩的find_element(By.NAME, email).send_keys(xxx)。复用性登录操作被封装成一个方法login()任何需要登录的测试用例都可以直接调用。5. 测试用例编写与unittest整合现在我们用unittest框架来编写和组织我们的测试用例。创建test_cases/test_login.py。import unittest import os import sys from selenium import webdriver from configparser import ConfigParser # 将项目根目录添加到Python路径以便导入自定义模块 sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) from utils.logger import logger from test_cases.login_page import LoginPage class TestNeteaseEmailLogin(unittest.TestCase): 网易邮箱登录测试用例类 classmethod def setUpClass(cls): 所有测试用例执行前运行一次用于初始化配置和浏览器 logger.info( * 50) logger.info(开始执行网易邮箱登录测试套件) logger.info( * 50) # 读取配置文件 config ConfigParser() config_path os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), config, config.ini) config.read(config_path, encodingutf-8) cls.url config.get(NETEASE_EMAIL, url) cls.username config.get(NETEASE_EMAIL, username) cls.password config.get(NETEASE_EMAIL, password) driver_path config.get(BROWSER, driver_path) headless config.getboolean(BROWSER, headless) # 初始化浏览器驱动 options webdriver.ChromeOptions() if headless: options.add_argument(--headless) # 无头模式不打开GUI options.add_argument(--disable-gpu) options.add_argument(--window-size1920,1080) # 无头模式需指定窗口大小 # 一些常用选项提升稳定性 options.add_argument(--no-sandbox) # 解决DevToolsActivePort文件不存在的报错 options.add_argument(--disable-dev-shm-usage) # 解决共享内存问题 options.add_experimental_option(excludeSwitches, [enable-logging]) # 禁止控制台无关日志 # 构建driver的绝对路径 abs_driver_path os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), driver_path) cls.driver webdriver.Chrome(executable_pathabs_driver_path, optionsoptions) cls.driver.maximize_window() # 最大化窗口 cls.driver.implicitly_wait(5) # 设置隐式等待全局等待元素出现的最大时间作为显式等待的补充 def setUp(self): 每个测试用例执行前运行这里我们打开登录页面 logger.info(f开始执行测试用例: {self._testMethodName}) self.driver.get(self.url) # 初始化页面对象 self.login_page LoginPage(self.driver) def test_01_login_success(self): 测试用例1使用正确的账号密码登录 logger.info(执行测试: 正确账号密码登录) self.login_page.switch_to_account_login() \ .enter_username(self.username) \ .enter_password(self.password) \ .click_submit() # 添加一个简单的断言登录成功后页面通常会跳转URL或标题会改变 # 这里我们等待一段时间然后检查当前URL是否不再是登录页URL简单示例 import time time.sleep(3) # 等待页面跳转实际应用中应用显式等待判断新页面元素 current_url self.driver.current_url self.assertNotEqual(current_url, self.url, 登录后页面未发生跳转可能登录失败) logger.info(f登录成功当前URL: {current_url}) # 更健壮的断言应该是判断登录后的页面是否存在某个特定元素如“收件箱” # 例如self.assertTrue(self.driver.find_element_by_id(_mail_component_149_149).is_displayed()) def test_02_login_failure_wrong_password(self): 测试用例2使用错误密码登录验证错误提示 logger.info(执行测试: 错误密码登录) wrong_password self.password _wrong self.login_page.switch_to_account_login() \ .enter_username(self.username) \ .enter_password(wrong_password) \ .click_submit() # 显式等待错误信息出现 from selenium.webdriver.support.ui import WebDriverWait from selenium.webdriver.support import expected_conditions as EC try: error_element WebDriverWait(self.driver, 5).until( EC.visibility_of_element_located(LoginPage.ERROR_MSG_SPAN) ) error_text error_element.text self.assertIn(账号或密码错误, error_text, f错误提示信息不符实际为: {error_text}) logger.info(f登录失败捕获到预期错误提示: {error_text}) except Exception as e: self.fail(f未找到预期的错误提示信息。异常: {e}) classmethod def tearDownClass(cls): 所有测试用例执行后运行一次用于清理环境 logger.info(所有测试用例执行完毕正在关闭浏览器...) cls.driver.quit() logger.info(浏览器已关闭。) logger.info( * 50) logger.info(网易邮箱登录测试套件执行结束) logger.info( * 50) if __name__ __main__: # 使用unittest的TestLoader和TextTestRunner来执行测试 suite unittest.TestLoader().loadTestsFromTestCase(TestNeteaseEmailLogin) runner unittest.TextTestRunner(verbosity2) # verbosity2 显示详细结果 runner.run(suite)unittest框架要点解析setUpClass/tearDownClass: 类方法在所有测试开始前和结束后各执行一次适合做一次性的初始化和清理如启动/关闭浏览器。setUp/tearDown: 实例方法在每个测试用例test_开头的方法执行前后运行适合做用例级别的准备和清理如打开特定页面。self.assertXxx: unittest提供的断言方法用于验证测试结果是否符合预期。断言失败则测试用例标记为失败。测试用例命名以test_开头这是unittest自动发现测试的规则。我们在setUp中初始化了LoginPage对象这样每个测试用例都能有一个干净的页面状态开始。6. 测试执行入口与报告生成最后我们创建一个统一的执行入口run_tests.py在项目根目录。这个脚本负责组织测试套件并生成更美观的HTML测试报告使用第三方库HTMLTestRunner。首先安装HTMLTestRunnerpip install html-testRunner然后创建run_tests.pyimport unittest import os import sys import time from datetime import datetime import HtmlTestRunner # 将项目根目录添加到路径 project_root os.path.dirname(os.path.abspath(__file__)) sys.path.append(project_root) def create_suite(): 创建测试套件 # 定义测试用例目录 test_case_dir os.path.join(project_root, test_cases) # 使用TestLoader发现并加载所有测试用例 discover unittest.defaultTestLoader.discover(test_case_dir, patterntest_*.py) return discover if __name__ __main__: # 创建报告目录 report_dir os.path.join(project_root, reports) if not os.path.exists(report_dir): os.makedirs(report_dir) # 生成带时间戳的报告文件名 current_time datetime.now().strftime(%Y%m%d_%H%M%S) report_file os.path.join(report_dir, fTestReport_{current_time}.html) # 使用HTMLTestRunner运行测试并生成报告 with open(report_file, wb) as fp: runner HtmlTestRunner.HTMLTestRunner( streamfp, report_title网易邮箱自动化登录测试报告, descriptions自动化测试执行结果, verbosity2 ) suite create_suite() runner.run(suite) print(f测试报告已生成: {report_file})运行这个脚本它会自动发现test_cases目录下所有以test_开头的测试文件执行里面的测试用例并生成一个带有详细结果、通过率、错误日志和截图的HTML报告放在reports目录下。报告直观易读非常适合发给非技术同事查看测试结果。7. 常见问题排查与实战技巧在实际运行中你肯定会遇到各种问题。这里我总结几个最常见的坑和解决思路。7.1 元素定位失败NoSuchElementException这是UI自动化最常见的问题。原因1页面未加载完成。解决使用WebDriverWait进行显式等待而不是time.sleep。确保等待的是元素可见visibility_of_element_located或可点击element_to_be_clickable而不仅仅是存在presence_of_element_located。原因2元素在iframe或shadow DOM内。解决使用driver.switch_to.frame(frame_reference)切换到对应的iframe后再定位元素。对于Shadow DOM需要使用execute_script执行JavaScript来穿透。原因3元素属性是动态生成的。解决避免使用绝对路径的XPath或依赖动态ID。优先使用相对稳定的属性如name,class, 或者包含部分固定文本的XPath如//button[contains(text(),登录)]。在浏览器开发者工具中使用$x()或$$()测试你的定位表达式。原因4页面有多个匹配的元素。解决find_element只返回第一个。确保你的定位器能唯一标识目标元素。可以使用find_elements获取列表再按索引选取。7.2 脚本运行不稳定有时成功有时失败原因网络波动、资源加载速度、动画效果等导致时机问题。解决增加合理的等待在关键操作前后如点击后页面跳转加入显式等待等待新页面的某个标志性元素出现。重试机制对于非断言性的操作如点击可以封装一个带重试的函数。禁用动画如果可能让开发同学在测试环境提供一个禁用CSS/JS动画的开关可以极大提升稳定性。使用更稳定的定位策略相对于XPathCSS Selector通常性能更好更稳定。7.3 浏览器被检测为自动化工具一些网站如登录页会检测navigator.webdriver属性来识别Selenium。解决可以尝试添加Chrome选项来隐藏自动化特征但请注意这可能违反某些网站的服务条款仅用于合法测试。options.add_experimental_option(excludeSwitches, [enable-automation]) options.add_experimental_option(useAutomationExtension, False) options.add_argument(--disable-blink-featuresAutomationControlled)更高级的绕过需要修改CDPChrome DevTools Protocol参数这里不展开。7.4 验证码处理自动化登录的最大挑战。绝对不要试图去自动识别和破解生产环境的验证码这是不安全且通常违反服务条款的。合法测试策略测试环境屏蔽验证码与开发团队沟通在测试/预发布环境中为特定的测试账号关闭验证码校验。使用万能验证码在测试环境设置一个固定的、已知的验证码如“0000”。人工干预如果必须测试带验证码的流程可以设计脚本在遇到验证码时暂停等待人工输入后再继续。Selenium本身不支持这个但可以结合input()或更复杂的消息通知机制。7.5 提升脚本健壮性的小技巧使用Page Factory模式虽然我们手动定义了定位器但对于大型项目可以考虑使用PageFactory模式源自JavaPython有类似实现如selenium.webdriver.support.pagefactory配合注解来懒加载元素代码更简洁。数据驱动测试将测试数据如多组账号密码从代码中分离存放在CSV、JSON或Excel文件中使用unittest的data装饰器需配合ddt库或自己读取文件来驱动多次测试。这能让一个测试用例覆盖多种数据场景。并发执行使用unittest的TestSuite结合多进程库如multiprocessing或专门的测试运行器如pytest-xdist可以并行执行测试用例大幅缩短测试时间。与CI/CD集成将你的自动化测试框架集成到Jenkins、GitLab CI等持续集成工具中每次代码提交后自动运行测试并及时反馈结果。这套从环境搭建、框架设计、代码实现到问题排查的完整流程基本涵盖了一个可投入实际项目的UI自动化测试框架的核心要素。记住自动化测试的最终目的不是取代手工测试而是将重复、枯燥的回归测试交给机器让测试人员能更专注于探索性测试和复杂场景。从这个小项目开始逐步理解每个环节的设计用意你就能将其扩展到更复杂的Web应用自动化测试中去。
Python+Selenium+unittest构建企业级UI自动化测试框架实战
发布时间：2026/6/30 18:35:41
1. 项目概述与核心价值最近在带团队做自动化测试项目发现很多同学对UI自动化的理解还停留在“录制回放”的初级阶段脚本写出来脆弱不堪维护成本极高。正好手头有个需求需要验证我们产品集成的第三方邮箱登录功能我就顺手用网易邮箱的登录流程作为案例搭建了一个基于Python、Selenium和unittest的自动化框架。这不仅仅是一个“自动登录”脚本更是一个完整的、可复用的、具备良好工程化结构的UI自动化解决方案。对于想从脚本小子进阶为自动化工程师的朋友来说理解这套框架的设计思路远比单纯复制代码更有价值。它能帮你处理元素定位的稳定性、测试用例的组织管理、失败场景的自动截图与日志记录最终实现低成本、高效益的自动化回归验证。2. 环境准备与核心工具选型解析工欲善其事必先利其器。一个稳定的自动化环境是成功的一半。很多人卡在环境配置上其实只要理清依赖关系一步步来并不复杂。2.1 Python环境与包管理首先你需要一个Python环境。我强烈建议使用Python 3.7及以上版本因为很多现代库对旧版本的支持已经减弱。新手可以直接去Python官网下载安装包安装时务必勾选“Add Python to PATH”这样可以省去后续手动配置环境变量的麻烦。安装完成后打开命令行Windows是CMD或PowerShellMac/Linux是Terminal输入python --version确认安装成功。接下来是包管理工具pip通常随Python一起安装用pip --version检查。自动化测试项目强烈建议使用虚拟环境virtual environment它能将项目的依赖包与系统全局的Python环境隔离避免版本冲突。创建虚拟环境很简单# 在当前目录下创建名为 venv 的虚拟环境 python -m venv venv # 激活虚拟环境 # Windows: venv\Scripts\activate # Mac/Linux: source venv/bin/activate激活后命令行提示符前会出现(venv)标识表示你已进入隔离环境。2.2 Selenium与WebDriver的“黄金搭档”Selenium是一个用于Web应用程序测试的强大工具集它本身是一套API通过selenium包提供而真正操控浏览器的则是WebDriver。这是最容易混淆的点。安装Selenium库在激活的虚拟环境中运行pip install selenium。下载并配置WebDriver这是关键一步。WebDriver是一个独立的可执行文件需要与你将要使用的浏览器版本严格匹配。Chrome去ChromeDriver官网下载对应你Chrome浏览器版本的驱动。查看Chrome版本在浏览器地址栏输入chrome://version/找到“版本”信息。将WebDriver放入PATH下载后将chromedriver.exe(Windows) 或chromedriver(Mac/Linux) 文件放在一个目录下并将该目录添加到系统的环境变量PATH中。更简单的做法是在代码中指定WebDriver的绝对路径我们后面会采用这种方式更清晰。注意浏览器会自动更新但WebDriver不会。如果某天你的脚本突然报错“无法启动浏览器”十有八九是浏览器升级后与WebDriver版本不匹配了记得重新下载匹配的版本。2.3 unittestPython自带的测试框架unittest是Python标准库中的单元测试框架我们用它来组织测试用例。它不需要额外安装提供了测试用例TestCase、测试套件TestSuite、测试夹具setUp/tearDown等概念非常适合结构化我们的自动化脚本。虽然现在pytest更流行但unittest对于理解测试框架的基本概念和入门UI自动化来说更加直观和标准。3. 框架设计与项目结构规划直接写一个.py文件把所有代码堆进去是初学者常犯的错误。这样的脚本难以维护、无法复用、出错难排查。我们需要一个清晰的工程结构。下面是我为这个项目设计的目录结构你可以直接套用netease_email_auto_login/ ├── config/ # 配置文件目录 │ └── config.ini # 存放URL、账号等敏感信息 ├── drivers/ # WebDriver存放目录 │ └── chromedriver.exe # Chrome浏览器驱动 ├── logs/ # 日志文件目录自动生成 ├── reports/ # 测试报告目录自动生成 ├── screenshots/ # 失败截图目录自动生成 ├── test_cases/ # 测试用例目录 │ ├── __init__.py │ └── test_login.py # 登录功能测试用例 ├── utils/ # 工具类目录 │ ├── __init__.py │ ├── logger.py # 日志记录模块 │ └── base_page.py # 页面基类 └── run_tests.py # 测试执行入口文件这样设计的好处配置与代码分离账号密码、URL等写在config.ini里不会泄露在代码中也方便不同环境切换。驱动集中管理所有驱动放在drivers下路径清晰。结果自动归档日志、报告、截图自动按时间或用例归类便于回溯。代码复用性高将浏览器操作、日志记录等通用功能抽象到utils下用例脚本只关注业务逻辑。用例独立每个测试用例文件独立方便管理和并行执行。4. 核心模块实现与代码逐行解析有了结构我们来填充核心代码。我会从底层工具类开始向上构建。4.1 日志记录模块utils/logger.py日志是自动化测试的“黑匣子”没有日志的脚本调试起来如同盲人摸象。我们设计一个既能输出到控制台又能保存到文件的日志器。import logging import os from datetime import datetime class Logger: def __init__(self, name__name__): # 创建logger实例 self.logger logging.getLogger(name) self.logger.setLevel(logging.DEBUG) # 设置最低日志级别 # 避免重复添加handler导致日志重复输出 if not self.logger.handlers: # 定义日志格式 formatter logging.Formatter(%(asctime)s - %(name)s - %(levelname)s - %(message)s) # 创建控制台handler并设置级别 ch logging.StreamHandler() ch.setLevel(logging.INFO) ch.setFormatter(formatter) self.logger.addHandler(ch) # 创建文件handler并设置级别 # 确保logs目录存在 log_dir os.path.join(os.path.dirname(os.path.dirname(__file__)), logs) if not os.path.exists(log_dir): os.makedirs(log_dir) log_file os.path.join(log_dir, fautotest_{datetime.now().strftime(%Y%m%d)}.log) fh logging.FileHandler(log_file, encodingutf-8) fh.setLevel(logging.DEBUG) # 文件里记录更详细的DEBUG信息 fh.setFormatter(formatter) self.logger.addHandler(fh) def get_logger(self): return self.logger # 提供一个全局的日志器实例 logger Logger(__name__).get_logger()关键点解析setLevel我们设置控制台只输出INFO及以上级别INFO, WARNING, ERROR, CRITICAL的日志避免刷屏而文件则记录所有DEBUG及以上的日志便于详细排查。FileHandler使用encodingutf-8这是为了防止中文日志内容乱码。检查logs目录是否存在并创建增强代码的健壮性。通过判断logger.handlers是否为空来避免重复添加handler这是一个重要的技巧防止在多次实例化Logger时日志重复打印。4.2 页面基类与通用操作封装utils/base_page.py这是框架的精华部分。我们将Selenium的常用操作如查找元素、点击、输入等进行二次封装并加入显式等待、日志记录和失败截图功能形成所有页面对象的“父类”。from selenium import webdriver from selenium.webdriver.support.ui import WebDriverWait from selenium.webdriver.support import expected_conditions as EC from selenium.common.exceptions import TimeoutException, NoSuchElementException import os import time from utils.logger import logger class BasePage: def __init__(self, driver: webdriver.Chrome): self.driver driver self.timeout 10 # 默认显式等待超时时间 def find_element(self, locator, timeoutNone): 查找单个元素加入显式等待和日志 if timeout is None: timeout self.timeout try: logger.info(f正在查找元素: {locator}) element WebDriverWait(self.driver, timeout).until( EC.presence_of_element_located(locator) ) logger.info(f元素查找成功: {locator}) return element except TimeoutException: error_msg f查找元素超时: {locator} logger.error(error_msg) self._take_screenshot(element_not_found) raise TimeoutException(error_msg) def click(self, locator): 点击元素 element self.find_element(locator) try: logger.info(f点击元素: {locator}) element.click() except Exception as e: error_msg f点击元素失败: {locator}, 错误: {e} logger.error(error_msg) self._take_screenshot(click_failed) raise def input_text(self, locator, text): 向元素输入文本 element self.find_element(locator) try: logger.info(f向元素 {locator} 输入文本: {text}) element.clear() # 先清空避免残留内容 element.send_keys(text) except Exception as e: error_msg f输入文本失败: {locator}, 错误: {e} logger.error(error_msg) self._take_screenshot(input_failed) raise def get_text(self, locator): 获取元素的文本内容 element self.find_element(locator) try: text element.text logger.info(f获取元素 {locator} 的文本: {text}) return text except Exception as e: error_msg f获取文本失败: {locator}, 错误: {e} logger.error(error_msg) raise def _take_screenshot(self, name): 内部方法失败时截图 screenshot_dir os.path.join(os.path.dirname(os.path.dirname(__file__)), screenshots) if not os.path.exists(screenshot_dir): os.makedirs(screenshot_dir) timestamp time.strftime(%Y%m%d_%H%M%S) file_path os.path.join(screenshot_dir, f{name}_{timestamp}.png) self.driver.save_screenshot(file_path) logger.info(f截图已保存至: {file_path})封装的价值显式等待find_element方法内置了显式等待WebDriverWait这是解决因网络或JS加载导致元素找不到问题的核心手段。它比time.sleep()更智能会在条件满足时立即返回而不是死等固定时间。统一日志每个关键操作都记录了日志运行脚本时你能清晰看到执行到了哪一步。自动截图在元素查找失败或操作异常时自动截取当前页面屏幕并保存到screenshots目录文件名包含时间戳。这是定位UI自动化问题最直接的证据。代码复用所有具体的页面类如登录页继承自BasePage就可以直接使用click(),input_text()这些稳定可靠的方法无需再写冗长的WebDriverWait代码。4.3 配置文件读取config.ini我们将可变参数配置化。在config/config.ini中写入[NETEASE_EMAIL] url https://mail.163.com/ username your_test_account163.com password your_test_password [BROWSER] driver_path drivers/chromedriver.exe headless False # 是否无头模式运行True为不打开浏览器界面重要警告千万不要将真实的账号密码提交到Git等版本控制系统.gitignore文件里要忽略config.ini。实际项目中敏感信息应通过环境变量或密钥管理服务传入。4.4 页面对象模型POM实现登录页面POM模式是UI自动化的最佳实践之一它将页面元素定位和页面操作封装成类使测试脚本用例更简洁元素定位变更时只需修改页面类。在test_cases/目录下我们其实更应该创建一个page_objects/目录来存放页面类。为了简化我们先在用例文件中实现。我们来创建test_cases/login_page.pyfrom selenium.webdriver.common.by import By from utils.base_page import BasePage class LoginPage(BasePage): # 页面元素定位器 (Locators) # 使用 (By.策略, 定位表达式) 的元组形式 SWITCH_TO_ACCOUNT_LOGIN (By.ID, switchAccountLogin) # “账号登录”切换链接 USERNAME_INPUT (By.NAME, email) # 用户名输入框 PASSWORD_INPUT (By.NAME, password) # 密码输入框 SUBMIT_BUTTON (By.ID, dologin) # 登录按钮 ERROR_MSG_SPAN (By.CSS_SELECTOR, .error-tt) # 错误信息提示示例 def __init__(self, driver): super().__init__(driver) def switch_to_account_login(self): 切换到账号密码登录方式网易邮箱默认可能是二维码 self.click(self.SWITCH_TO_ACCOUNT_LOGIN) return self # 返回自身支持链式调用 def enter_username(self, username): 输入用户名 self.input_text(self.USERNAME_INPUT, username) return self def enter_password(self, password): 输入密码 self.input_text(self.PASSWORD_INPUT, password) return self def click_submit(self): 点击登录按钮 self.click(self.SUBMIT_BUTTON) # 点击后页面会跳转通常返回下一个页面的对象这里我们简单处理 # 实际项目中可以返回 InboxPage 之类的对象 def get_error_message(self): 获取错误提示信息用于登录失败断言 try: return self.get_text(self.ERROR_MSG_SPAN) except: return # 如果没有找到错误元素说明可能登录成功返回空字符串 def login(self, username, password): 完整的登录流程 self.switch_to_account_login().enter_username(username).enter_password(password).click_submit()POM的优势可维护性如果登录页面的输入框ID从email改成了username你只需要修改USERNAME_INPUT这一个常量的值所有用到它的测试用例都无需改动。可读性在测试用例中你看到的是login_page.enter_username(xxx)这样清晰的业务语句而不是晦涩的find_element(By.NAME, email).send_keys(xxx)。复用性登录操作被封装成一个方法login()任何需要登录的测试用例都可以直接调用。5. 测试用例编写与unittest整合现在我们用unittest框架来编写和组织我们的测试用例。创建test_cases/test_login.py。import unittest import os import sys from selenium import webdriver from configparser import ConfigParser # 将项目根目录添加到Python路径以便导入自定义模块 sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) from utils.logger import logger from test_cases.login_page import LoginPage class TestNeteaseEmailLogin(unittest.TestCase): 网易邮箱登录测试用例类 classmethod def setUpClass(cls): 所有测试用例执行前运行一次用于初始化配置和浏览器 logger.info( * 50) logger.info(开始执行网易邮箱登录测试套件) logger.info( * 50) # 读取配置文件 config ConfigParser() config_path os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), config, config.ini) config.read(config_path, encodingutf-8) cls.url config.get(NETEASE_EMAIL, url) cls.username config.get(NETEASE_EMAIL, username) cls.password config.get(NETEASE_EMAIL, password) driver_path config.get(BROWSER, driver_path) headless config.getboolean(BROWSER, headless) # 初始化浏览器驱动 options webdriver.ChromeOptions() if headless: options.add_argument(--headless) # 无头模式不打开GUI options.add_argument(--disable-gpu) options.add_argument(--window-size1920,1080) # 无头模式需指定窗口大小 # 一些常用选项提升稳定性 options.add_argument(--no-sandbox) # 解决DevToolsActivePort文件不存在的报错 options.add_argument(--disable-dev-shm-usage) # 解决共享内存问题 options.add_experimental_option(excludeSwitches, [enable-logging]) # 禁止控制台无关日志 # 构建driver的绝对路径 abs_driver_path os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), driver_path) cls.driver webdriver.Chrome(executable_pathabs_driver_path, optionsoptions) cls.driver.maximize_window() # 最大化窗口 cls.driver.implicitly_wait(5) # 设置隐式等待全局等待元素出现的最大时间作为显式等待的补充 def setUp(self): 每个测试用例执行前运行这里我们打开登录页面 logger.info(f开始执行测试用例: {self._testMethodName}) self.driver.get(self.url) # 初始化页面对象 self.login_page LoginPage(self.driver) def test_01_login_success(self): 测试用例1使用正确的账号密码登录 logger.info(执行测试: 正确账号密码登录) self.login_page.switch_to_account_login() \ .enter_username(self.username) \ .enter_password(self.password) \ .click_submit() # 添加一个简单的断言登录成功后页面通常会跳转URL或标题会改变 # 这里我们等待一段时间然后检查当前URL是否不再是登录页URL简单示例 import time time.sleep(3) # 等待页面跳转实际应用中应用显式等待判断新页面元素 current_url self.driver.current_url self.assertNotEqual(current_url, self.url, 登录后页面未发生跳转可能登录失败) logger.info(f登录成功当前URL: {current_url}) # 更健壮的断言应该是判断登录后的页面是否存在某个特定元素如“收件箱” # 例如self.assertTrue(self.driver.find_element_by_id(_mail_component_149_149).is_displayed()) def test_02_login_failure_wrong_password(self): 测试用例2使用错误密码登录验证错误提示 logger.info(执行测试: 错误密码登录) wrong_password self.password _wrong self.login_page.switch_to_account_login() \ .enter_username(self.username) \ .enter_password(wrong_password) \ .click_submit() # 显式等待错误信息出现 from selenium.webdriver.support.ui import WebDriverWait from selenium.webdriver.support import expected_conditions as EC try: error_element WebDriverWait(self.driver, 5).until( EC.visibility_of_element_located(LoginPage.ERROR_MSG_SPAN) ) error_text error_element.text self.assertIn(账号或密码错误, error_text, f错误提示信息不符实际为: {error_text}) logger.info(f登录失败捕获到预期错误提示: {error_text}) except Exception as e: self.fail(f未找到预期的错误提示信息。异常: {e}) classmethod def tearDownClass(cls): 所有测试用例执行后运行一次用于清理环境 logger.info(所有测试用例执行完毕正在关闭浏览器...) cls.driver.quit() logger.info(浏览器已关闭。) logger.info( * 50) logger.info(网易邮箱登录测试套件执行结束) logger.info( * 50) if __name__ __main__: # 使用unittest的TestLoader和TextTestRunner来执行测试 suite unittest.TestLoader().loadTestsFromTestCase(TestNeteaseEmailLogin) runner unittest.TextTestRunner(verbosity2) # verbosity2 显示详细结果 runner.run(suite)unittest框架要点解析setUpClass/tearDownClass: 类方法在所有测试开始前和结束后各执行一次适合做一次性的初始化和清理如启动/关闭浏览器。setUp/tearDown: 实例方法在每个测试用例test_开头的方法执行前后运行适合做用例级别的准备和清理如打开特定页面。self.assertXxx: unittest提供的断言方法用于验证测试结果是否符合预期。断言失败则测试用例标记为失败。测试用例命名以test_开头这是unittest自动发现测试的规则。我们在setUp中初始化了LoginPage对象这样每个测试用例都能有一个干净的页面状态开始。6. 测试执行入口与报告生成最后我们创建一个统一的执行入口run_tests.py在项目根目录。这个脚本负责组织测试套件并生成更美观的HTML测试报告使用第三方库HTMLTestRunner。首先安装HTMLTestRunnerpip install html-testRunner然后创建run_tests.pyimport unittest import os import sys import time from datetime import datetime import HtmlTestRunner # 将项目根目录添加到路径 project_root os.path.dirname(os.path.abspath(__file__)) sys.path.append(project_root) def create_suite(): 创建测试套件 # 定义测试用例目录 test_case_dir os.path.join(project_root, test_cases) # 使用TestLoader发现并加载所有测试用例 discover unittest.defaultTestLoader.discover(test_case_dir, patterntest_*.py) return discover if __name__ __main__: # 创建报告目录 report_dir os.path.join(project_root, reports) if not os.path.exists(report_dir): os.makedirs(report_dir) # 生成带时间戳的报告文件名 current_time datetime.now().strftime(%Y%m%d_%H%M%S) report_file os.path.join(report_dir, fTestReport_{current_time}.html) # 使用HTMLTestRunner运行测试并生成报告 with open(report_file, wb) as fp: runner HtmlTestRunner.HTMLTestRunner( streamfp, report_title网易邮箱自动化登录测试报告, descriptions自动化测试执行结果, verbosity2 ) suite create_suite() runner.run(suite) print(f测试报告已生成: {report_file})运行这个脚本它会自动发现test_cases目录下所有以test_开头的测试文件执行里面的测试用例并生成一个带有详细结果、通过率、错误日志和截图的HTML报告放在reports目录下。报告直观易读非常适合发给非技术同事查看测试结果。7. 常见问题排查与实战技巧在实际运行中你肯定会遇到各种问题。这里我总结几个最常见的坑和解决思路。7.1 元素定位失败NoSuchElementException这是UI自动化最常见的问题。原因1页面未加载完成。解决使用WebDriverWait进行显式等待而不是time.sleep。确保等待的是元素可见visibility_of_element_located或可点击element_to_be_clickable而不仅仅是存在presence_of_element_located。原因2元素在iframe或shadow DOM内。解决使用driver.switch_to.frame(frame_reference)切换到对应的iframe后再定位元素。对于Shadow DOM需要使用execute_script执行JavaScript来穿透。原因3元素属性是动态生成的。解决避免使用绝对路径的XPath或依赖动态ID。优先使用相对稳定的属性如name,class, 或者包含部分固定文本的XPath如//button[contains(text(),登录)]。在浏览器开发者工具中使用$x()或$$()测试你的定位表达式。原因4页面有多个匹配的元素。解决find_element只返回第一个。确保你的定位器能唯一标识目标元素。可以使用find_elements获取列表再按索引选取。7.2 脚本运行不稳定有时成功有时失败原因网络波动、资源加载速度、动画效果等导致时机问题。解决增加合理的等待在关键操作前后如点击后页面跳转加入显式等待等待新页面的某个标志性元素出现。重试机制对于非断言性的操作如点击可以封装一个带重试的函数。禁用动画如果可能让开发同学在测试环境提供一个禁用CSS/JS动画的开关可以极大提升稳定性。使用更稳定的定位策略相对于XPathCSS Selector通常性能更好更稳定。7.3 浏览器被检测为自动化工具一些网站如登录页会检测navigator.webdriver属性来识别Selenium。解决可以尝试添加Chrome选项来隐藏自动化特征但请注意这可能违反某些网站的服务条款仅用于合法测试。options.add_experimental_option(excludeSwitches, [enable-automation]) options.add_experimental_option(useAutomationExtension, False) options.add_argument(--disable-blink-featuresAutomationControlled)更高级的绕过需要修改CDPChrome DevTools Protocol参数这里不展开。7.4 验证码处理自动化登录的最大挑战。绝对不要试图去自动识别和破解生产环境的验证码这是不安全且通常违反服务条款的。合法测试策略测试环境屏蔽验证码与开发团队沟通在测试/预发布环境中为特定的测试账号关闭验证码校验。使用万能验证码在测试环境设置一个固定的、已知的验证码如“0000”。人工干预如果必须测试带验证码的流程可以设计脚本在遇到验证码时暂停等待人工输入后再继续。Selenium本身不支持这个但可以结合input()或更复杂的消息通知机制。7.5 提升脚本健壮性的小技巧使用Page Factory模式虽然我们手动定义了定位器但对于大型项目可以考虑使用PageFactory模式源自JavaPython有类似实现如selenium.webdriver.support.pagefactory配合注解来懒加载元素代码更简洁。数据驱动测试将测试数据如多组账号密码从代码中分离存放在CSV、JSON或Excel文件中使用unittest的data装饰器需配合ddt库或自己读取文件来驱动多次测试。这能让一个测试用例覆盖多种数据场景。并发执行使用unittest的TestSuite结合多进程库如multiprocessing或专门的测试运行器如pytest-xdist可以并行执行测试用例大幅缩短测试时间。与CI/CD集成将你的自动化测试框架集成到Jenkins、GitLab CI等持续集成工具中每次代码提交后自动运行测试并及时反馈结果。这套从环境搭建、框架设计、代码实现到问题排查的完整流程基本涵盖了一个可投入实际项目的UI自动化测试框架的核心要素。记住自动化测试的最终目的不是取代手工测试而是将重复、枯燥的回归测试交给机器让测试人员能更专注于探索性测试和复杂场景。从这个小项目开始逐步理解每个环节的设计用意你就能将其扩展到更复杂的Web应用自动化测试中去。
![华为OD机试2025C卷-字符串变换最小次数[100分]（ Java _ Python3 _ C++ _ C语言 _ JsNode _ Go）实现100%通过率](http://pic.xiahunao.cn/yaotu/华为OD机试2025C卷-字符串变换最小次数[100分]（ Java _ Python3 _ C++ _ C语言 _ JsNode _ Go）实现100%通过率)
![华为OD机试2025C卷-内存资源分配[100分]（ Java _ Python3 _ C++ _ C语言 _ JsNode _ Go）实现100%通过率](http://pic.xiahunao.cn/yaotu/华为OD机试2025C卷-内存资源分配[100分]（ Java _ Python3 _ C++ _ C语言 _ JsNode _ Go）实现100%通过率)
