我有连接到 MongoDB 客户端的代码,我正在尝试测试它。为了进行测试,我不想连接到实际的客户端,因此我试图找出一个假客户端来进行测试。代码的基本流程是我在某处有一个函数创建一个pymongo
客户端,然后查询并创建一个在其他地方使用的字典。
我想使用 pytest 编写一些测试来测试将调用的不同函数和类get_stuff
。我的问题是get_stuff
calls mongo()
这就是实际建立与数据库的连接的原因。我试图只使用pytest.fixture(autouse=True)
and 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
某处_else.py
from project.mongo_stuff import mongo_stuff
mongo_dict = mongo_stuff.get_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()
这目前会给我一个连接错误,因为connection params
in mongo_stuff.py仅适用于生产环境。如果我把import
声明来自测试材料.py进入测试功能,然后它工作正常并且mongomock
测试环境中将使用db。我也尝试过改变setattr
to monkeypatch.setattr('project.working_class.mongo_stuff.mongo', fake_mongo)
这也行不通。