使用假的mongoDB进行pytest测试

Question

我有连接到MongoDB客户端的代码，我正在尝试测试它。为了测试，我不想连接到实际的客户端，所以我试图找出一个假的用于测试目的。代码的基本流程是我在某处有一个函数，用于创建一个pymongo客户端，然后查询该函数并生成在其他地方使用的字典。

我想使用 pytest 编写一些测试，这些测试将测试将调用get_stuff的不同函数和类。我的问题是get_stuff调用mongo()这是实际连接到数据库的原因。我试图只使用pytest.fixture(autouse=True)和mongomock.MongoClient()来替换mongo().

但这并不能取代mongo_stuff.mongo().有没有办法告诉pytest替换函数，以便调用我的fixture而不是实际函数？我认为制作fixture会使我的测试在命名空间中mongo()优先级高于实际模块中的函数。

下面是一个带有我示例的示例文件结构：

.
├── project
│   ├── __init__.py
│   ├── mongo_stuff
│   │   ├── __init__.py
│   │   └── mongo_stuff.py
│   └── working_class
│       ├── __init__.py
│       └── somewhere_else.py
└── testing
├── __init__.py
└── test_stuff.py

mongo_stuff.py

import pymongo
def mongo():
return pymongo.MongoClient(connection_params)
def get_stuff():
db = mongo()  # Makes the connection using another function
stuff = query_function(db)  # Does the query and makes a dict
return result

somewhere_else.py

from project.mongo_stuff import mongo_stuff
mongo_dict = mongo_stuff.get_stuff()

test_stuff.py

import pytest
import mongomock
@pytest.fixture(autouse=True)
def patch_mongo(monkeypatch):
db = mongomock.MongoClient()
def fake_mongo():
return db
monkeypatch.setattr('project.mongo_stuff.mongo', fake_mongo)
from poject.working_class import working_class  # This starts by calling project.mongo_stuff.mongo_stuff.get_stuff()

这目前会给我一个连接错误，因为mongo_stuff.py中的connection params只能在生产环境中工作。如果我将test_stuff.py中的import语句放入测试函数中，那么它工作正常，并且mongomockdb 将在测试环境中使用。我还尝试将setattr更改为monkeypatch.setattr('project.working_class.mongo_stuff.mongo', fake_mongo)这也不起作用。

Answer 1

你已经完成了一半：你已经为数据库客户端创建了一个模拟，现在你必须修补mongo_stuff.mongo函数以返回模拟而不是真正的连接：

@pytest.fixture(autouse=True)
def patch_mongo(monkeypatch):
db = mongomock.MongoClient()
def fake_mongo():
return db
monkeypatch.setattr('mongo_stuff.mongo', fake_mongo)

编辑：

您收到连接错误的原因是您在test_stuff中在模块级别导入somewhere_else，并且somewhere_else也在模块级别运行连接代码。因此，用夹具打补丁会来得太晚，不会有任何效果。如果要在模块级别导入，则必须在导入somewhere_else之前修补 mongo 客户端。这将避免引发错误，但非常丑陋：

from project.mongo_stuff import mongo_stuff
import mongomock
import pytest
from unittest.mock import patch
with patch.object(mongo_stuff, 'mongo', return_value=mongomock.MongoClient()):
from project.working_class import somewhere_else

@patch.object(mongo_stuff, 'mongo', return_value=mongomock.MongoClient())
def test_db1(mocked_mongo):
mongo_stuff.mongo()
assert True

@patch.object(mongo_stuff, 'mongo', return_value=mongomock.MongoClient())
def test_db2(mocked_mongo):
somewhere_else.foo()
assert True

如果可能的话，你应该避免在模块级别运行代码，或者在测试中运行在模块级别执行代码的导入(正如您在注释中已经发现的那样(。