是否有内置的模拟对象可以传递给 Python 单元测试？

Question

我通常在单元测试期间实例化一个Mock 对象。我厌倦了：

1. 必须输入from unittest.mock import Mock
2. 然后通过mock = Mock()实例化一个Mock对象

我想知道，pytest、unittest.mock、pytest-mock 等是否有内置的传递 Mock 对象的方式？

这是我当前的工作流程：

from unittest.mock import Mock


def test_something() -> None:
    mock = Mock()
    mock.func()
    mock.func.assert_called_once()

---

研究

我知道pytest-mock，我至少可以通过mocker fixture 绕过导入。

def test_something(mocker) -> None:
    mock = mocker.Mock()
    mock.func()
    mock.func.assert_called_once()

我希望能够做这样的事情：

def test_something(mock) -> None:
    mock.func()
    mock.func.assert_called_once()

Answer 1

如果您使用的是pytest，只需创建一个fixture。

@pytest.fixture(scope="function")
def my_cool_mock():
    mocked = Mock()
    yield mocked

def test_something(my_cool_mock) -> None:
    my_cool_mock.func()
    my_cool_mock.func.assert_called_once()

否则，请使用pytest-mock。