由于 AttributeError，PyTest-Mock 无法正常工作

Question

我正在尝试使用 PyTest_Mock 在我的 Python 项目中进行一些测试。我创建了一个非常简单的测试来尝试它，但是我得到一个 AttributeError 并且我不知道为什么。

model.py

def square(x):
    return x * x

if __name__ == '__main__':
    res = square(5)
    print("result: {}".format(res))

test_model.py

import pytest
from pytest_mock import mocker

import model

def test_model():
    mocker.patch(square(5))

    assert model.square(5) == 25

在运行python -m pytest 后，我遇到了一个失败并出现以下错误：

    def test_model():
>       mocker.patch(square(5))
E       AttributeError: 'function' object has no attribute 'patch'

test_model.py:7: AttributeError

Answer 1

1. 你不需要导入mocker，它可以作为fixture使用，所以你只需在测试函数中将它作为参数传递：

   def test_model(mocker):
       mocker.patch(...)
   

2. square(5) 计算结果为 25，因此mocker.patch(square(5)) 将有效地尝试修补数字 25。相反，将函数名称作为参数传递：要么

   mocker.patch('model.square')
   

   或

   mocker.patch.object(model, 'square')
   

3. 修补后，square(5) 将不再返回 25，因为原始函数被替换为可以返回任何内容的模拟对象，并且默认情况下将返回一个新的模拟对象。 assert model.square(5) == 25 将因此失败。通常，您修补东西以避免复杂的测试设置或模拟测试场景中所需的组件行为（例如，网站不可用）。在您的示例中，您根本不需要嘲笑。

完整的工作示例：

import model

def test_model(mocker):
    mocker.patch.object(model, 'square', return_value='foo')

assert model.square(5) == 'foo'