如何避免pytest中测试用例间的模拟数据冲突？

核心原因 #

你的问题本质是两个测试用例的Mock实例发生了交叉污染——要么是Mock被全局共享，要么是Patch的路径不正确，导致第二个测试复用了第一个测试的Mock返回值。以下是几种直接有效的解决方法：

1. 每个测试用例独立Patch目标函数 #

不要在测试类级别共享Patch装饰器，而是给每个测试方法单独使用@patch装饰，或者用with patch(...)上下文管理器，确保每个测试都生成全新的Mock实例。

关键注意：Patch的路径必须是函数被调用的模块位置，而非函数定义的原模块。比如两个命令各自导入了同名函数，就要分别Patch命令模块内的那个引用，而非原工具模块的函数。

示例代码：

# test_command_one.py
from unittest.mock import patch
from django.test import TestCase
from myapp.management.commands.command_one import Command

class TestCommandOne(TestCase):
    def test_valid_output(self):
        # Patch command_one模块里的get_api_data引用
        with patch('myapp.management.commands.command_one.get_api_data') as mock_get:
            mock_get.return_value = {'time1': '2024-01-01'}
            # 执行命令并断言
            cmd = Command()
            cmd.handle()
            ...

# test_command_two.py
from unittest.mock import patch
from django.test import TestCase
from myapp.management.commands.command_two import Command

class TestCommandTwo(TestCase):
    def test_valid_output(self):
        # Patch command_two模块里的get_api_data引用
        with patch('myapp.management.commands.command_two.get_api_data') as mock_get:
            mock_get.return_value = {'time2': '2024-01-02'}
            # 执行命令并断言
            cmd = Command()
            cmd.handle()
            ...

2. 用setUp/tearDown重置Mock状态 #

如果必须在类级别共享Patch，可以在setUp()方法中重置Mock的返回值和调用记录，彻底清除上一个测试的残留状态：

from unittest.mock import patch
from django.test import TestCase
from myapp.management.commands.command_two import Command

class TestCommandTwo(TestCase):
    @classmethod
    def setUpClass(cls):
        super().setUpClass()
        # 启动类级别的Patch
        cls.mock_get = patch('myapp.management.commands.command_two.get_api_data').start()

    @classmethod
    def tearDownClass(cls):
        # 停止Patch
        cls.mock_get.stop()
        super().tearDownClass()

    def setUp(self):
        super().setUp()
        # 每个测试前重置Mock状态，重新设置返回值
        self.mock_get.reset_mock()
        self.mock_get.return_value = {'time2': '2024-01-02'}

    def test_valid_output(self):
        cmd = Command()
        cmd.handle()
        ...

reset_mock()会清除Mock的调用次数、参数记录等状态，确保每个测试用的是干净的Mock实例。

3. 禁止全局Mock实例 #

不要在测试模块的全局作用域定义Mock，比如这种写法会导致所有测试共享同一个Mock：

# 错误示例：全局Mock会被所有测试复用
from unittest.mock import Mock
mock_get = Mock(return_value={'time1': '2024-01-01'})

所有Mock的创建和配置都要放在测试方法内部或setUp中，保证测试间完全隔离。

4. 检查跨模块引用的一致性 #

如果两个命令都是从同一个公共模块导入get_api_data，比如：

# command_one.py
from myapp.utils import get_api_data

# command_two.py
from myapp.utils import get_api_data

此时如果直接Patchmyapp.utils.get_api_data，两个测试的Mock会冲突。必须分别Patch每个命令模块内的引用，也就是command_one.get_api_data和command_two.get_api_data，才能实现隔离。

内容的提问来源于stack exchange，提问作者markwalker_