如何使用 pytest-mock 或 magicmock 模拟导入的对象

我正在尝试理解mock/monkeypatch/pytest-mock能力。

让我知道这是否可能。如果没有，请建议我如何测试此代码。

我的代码结构：

/
./app
../__init__.py
../some_module1
.../__init__.py
../some_module2
.../__init__.py
./tests
../test_db.py

The /app/__init__.py是我的应用程序（Flask 应用程序，如果有帮助的话）启动的地方，同时初始化 MongoDB 数据库的数据库连接对象：

# ...

def create_app():
  # ...
  return app

db_conn = DB()

The some_module1 and some_module导入db_conn对象并将其用作其功能的一部分：

## some_module1/__init__.py
from app import db_conn

...
db = db_conn.db_name2.db_collection2

def some_func1():
    data = db.find()
    # check and do something with data
    return boolean_result

...

## some_module2/__init__.py
from app import db_conn

...
db = db_conn.db_name1.db_collection1

def some_func2():
    data = db.find()
    # check and do something with data
    return boolean_result
...

在我的测试中，我想根据从数据库接收到的数据来测试我的代码是否正常工作。 我想模拟数据库，更具体地说是db_conn对象，因为我不想使用真正的数据库（这将需要大量的工作来设置环境并维护它）。

关于如何模仿的任何建议db_conn?

我一直在探索pytest-mock and magicmock但我不认为也不知道如何嘲笑db_conn在我的测试中。

要回答最初的问题“如何使用 pytest-mock 或 magicmock 模拟导入的对象”，您可以执行以下操作：

from unittest import mock  # because unittest's mock works great with pytest

def test_some_func1():
    with mock.patch('some_module1.db', mock.MagicMock(return_value=...)) as magicmock:
        result = some_func1(...)
        assert ... e.g. different fields of magicmock
        assert expected == result

# or alternatively use annotations

@mock.patch('some_module2.db', mock.MagicMock(return_value=...))
def test_some_func2():
        result = some_func2(...)

请注意，您没有修补数据库的实际来源 http://alexmarandon.com/articles/python_mock_gotchas/

对于您的其他用例

我想模拟数据库（使用 mongo 数据库），更具体地说是“db_conn”对象

您同样按照上面链接的提示进行操作：

mock.patch('some_module1.db_conn', mock.MagicMock(return_value=...))

鉴于此，您会在测试中注意到db从 `db = db_conn.db_name2.db_collection2' 将创建另一个模拟对象。对该对象的调用也将被记录。通过这种方式，您还可以跟踪呼叫和值分配的历史记录。

此外，请参阅如何修补 mongodb 的示例。 https://stackoverflow.com/a/51994349/1256837

要测试 Flask 应用程序，请参阅烧瓶的文档 https://flask.palletsprojects.com/en/1.1.x/testing/. Also 这也是一个很好的解释，并且使用数据库连接 https://www.patricksoftwareblog.com/testing-a-flask-application-using-pytest/.

作为一般提示，就像 @MikeMajara 提到的那样 - 将代码更多地分成更小的函数，这些函数也易于测试。 TDD 的传统是：先编写测试，然后实现，然后重构（尤其是 DRY！）