您已经到了一半:您已经为db客户端创建了一个模拟,现在您必须修补该
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)编辑:
出现连接错误的原因是,您正在
somewhere_else中的模块级别导入
test_stuff,并且
somewhere_else也在模块级别上运行连接代码。因此,用灯具打补丁太迟了,不会有任何效果。如果要在模块级别导入,则必须在导入
之前 修补mongo客户端
somewhere_else。这样可以避免引发错误,但是非常丑陋:
from project.mongo_stuff import mongo_stuffimport mongomockimport pytestfrom unittest.mock import patchwith 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
您应该避免在可能的情况下在模块级别运行代码,或者在测试内部运行在模块级别执行代码的导入(如您在注释中已经发现的那样)。



