有没有办法使用模拟来模拟 pytest 中的完整代码？

Question

例如，每当测试找到

database.db.session.using_bind("reader")

我要删除lust_bind（“读取器”））并与

database.db.session

使用模拟器

一起使用conftest.py.py尝试使用它来删除）>

@pytest.fixture(scope='function')
def session(mocker):
    mocker.patch('store.database.db.session.using_bind', return_value=_db.db.session)

但是到目前为止什么都没有。

正在测试的代码：

from store import database     
results = database.db.session.using_bind("reader").query(database.Order.id).join(database.Shop).filter(database.Shop.deleted == False).all(), 

我得到

AttributeError: 'scoped_session' object has no attribute 'using_bind' as an error.

Answer 1

让我们从一个MRE开始，其中正在测试的代码使用假数据库：

from unittest.mock import Mock, patch

class Session:
    def using_bind(self, bind):
        raise NotImplementedError(f"Can't bind {bind}")

    def query(self):
        return "success!"


database = Mock()
database.db.session = Session()


def code_under_test():
    return database.db.session.using_bind("reader").query()


def test():
    assert code_under_test() == "success!"

运行此测试失败：

E       NotImplementedError: Can't bind reader

因此，我们要模拟session.used_bind in code_under_teste_test，以便它返回<<代码>会话 - 这将使我们的测试通过。

我们使用patch这样做，例如：

@patch("test.database.db.session.using_bind")
def test(mock_bind):
    mock_bind.return_value = database.db.session
    assert code_under_test() == "success!"

请注意，我的代码位于名为test.py的文件中，因此我的patch> patch呼叫适用于测试模块 - 您需要对其进行调整，以指向您自己的代码中测试的模块。

还要注意，在调用正在测试的代码之前，我需要设置模拟。