如何使用Python进行自动化测试_菜鸟教程python自动化测试-CSDN博客

写测试代码其实一点都不难！说来可笑，我刚开始写代码的时候总觉得测试代码特别麻烦，能运行就完事了。直到有一天，我改了一个小功能，结果不知道改出了什么问题，整个系统都崩溃了。从那以后，我就明白了自动化测试的重要性。

自动化测试入门

自动化测试本质上就是用代码检查另一段代码是否正常工作。打个比方，就像你买了个新手机，肯定要试试各种功能对不对？看看电话能不能打通，微信能不能正常聊天，照相机清不清晰。自动化测试也是这个道理，只不过是用代码来做这些检查。

我记得有次做一个计算器程序，要是按照老方法，得一个功能一个功能手动试，加法、减法、乘法、除法，还得试试负数、小数…光想想就头大。用了自动化测试后，写好测试代码，一键运行就能知道所有功能是否正常，省时省力还不容易出错。

def add(a, b):
   return a + b

# 测试代码
def test_add():
   assert add(1, 1) == 2
   assert add(-1, 1) == 0
   assert add(0.1, 0.2) == 0.3

温馨提示: assert 是Python中用来断言的关键字，可以理解为"断定某事一定是对的"。如果断言的结果是False，Python会立即抛出异常，告诉我们哪里出问题了。

pytest框架快速上手

说实话，光用assert来测试还是太原始了。记得我刚开始写测试的时候，就感觉缺了点什么。后来发现了pytest这个框架，简直就是给测试工作开了外挂！它不仅能自动找到并执行测试代码，还能生成清晰的测试报告，告诉你具体哪些测试通过了，哪些失败了，失败的原因是什么。

import pytest

def test_string_match():
   # 检查字符串是否包含特定内容
   text = "hello world"
   assert "hello" in text

@pytest.mark.parametrize("test_input,expected", [
   ("3+5", 8),
   ("2+4", 6),
   ("6+5", 11),
])
def test_eval(test_input, expected):
   # 参数化测试，一次测试多组数据
   assert eval(test_input) == expected

pytest的参数化测试真是太好用了！不用复制粘贴同样的代码，就能测试多组数据。还有个神奇的地方，它能自动发现文件中所有以test开头的函数，统一执行测试。

模拟真实场景

在实际工作中，经常会遇到需要连接数据库或者调用外部接口的情况。总不能为了测试去真实操作数据库吧？万一测试数据把生产环境搞崩了那就惨了。这时候就需要用到mock技术了，它就像是个替身演员，能模仿真实对象的行为。

from unittest.mock import Mock, patch

def get_user_data(user_id):
   # 假设这是个查询数据库的函数
   pass

def test_get_user():
   # 模拟数据库返回结果
   mock_db = Mock()
   mock_db.query.return_value = {"name": "小明", "age": 18}

   with patch('__main__.get_user_data', mock_db):
       result = get_user_data(1)
       assert result["name"] == "小明"

温馨提示: 使用mock不仅能避免测试对实际环境的影响，还能模拟各种异常情况，比如网络超时、数据库连接失败等，这在真实环境中可能很难触发。

测试用例组织

随着项目越来越大，测试代码也会越来越多。要是不好好组织，过段时间自己都看不懂写的是啥了。我现在习惯用类来组织相关的测试用例，每个类就像一个小专题，专门测试某个模块的功能。

class TestUserModule:
   def setup_method(self):
       # 每个测试方法前都会执行
       self.test_data = {"id": 1, "name": "小明"}

   def test_create_user(self):
       # 测试创建用户
       pass

   def test_delete_user(self):
       # 测试删除用户
       pass

setup_method这个方法特别实用，它会在每个测试方法执行前自动运行，可以用来准备测试数据。这样就不用在每个测试方法里都写一遍准备数据的代码了。

Selenium自动化测试

说到Web应用测试，不得不提Selenium。它就像是个机器人，能帮你自动操作网页，点击按钮、填写表单、获取页面内容，啥都能干。

om selenium import webdriver

driver = webdriver.Chrome()
driver.get("http://www.example.com")

# 查找输入框并输入内容
search_box = driver.find_element_by_id("search")
search_box.send_keys("测试内容")

# 点击搜索按钮
search_button = driver.find_element_by_id("submit")
search_button.click()

# 检查结果
results = driver.find_elements_by_class_name("result")
assert len(results) > 0

用Selenium写的测试就像是请了个不知疲倦的助手，24小时都能帮你测试网页功能。特别是在做回归测试的时候，几百个页面的功能，手工测试哪里比得上自动化效率高。

现在回想起来，以前总觉得写测试代码是在浪费时间，现在才发现这完全是一种误解。好的测试用例就像是项目的守护神，能在代码变更时及时发现问题，避免bug溜进生产环境。而且自动化测试还能作为代码的文档，新来的同事看看测试用例，就知道每个功能该怎么用了。

最后作为一位过来人也是希望大家少走一些弯路，如果你不想再体验一次学习时找不到资料，没人解答问题，坚持几天便放弃的感受的话，在这里我给大家分享一些软件测试的学习资源，希望能给你前进的路上带来帮助。