如何在持有MongoClient变量时Mock MongoDB的find方法？

你遇到的问题其实是MongoDB调用的链式结构导致的——mongoclient().db_name.collection.find()是多层嵌套调用，只Mock最外层的mongoclient远远不够，得逐层把每个环节的返回值都设置好，才能让find()返回你想要的模拟数据。

下面给你两种可行的解决方案，都是基于Python的unittest.mock模块实现的：

方案一：逐层Mock链式调用（清晰易懂） #

这种写法把每个环节的Mock对象都明确拆分出来，适合复杂场景的调试和验证：

from unittest.mock import patch, Mock
import dao

def test_my_call():
    # 1. 定义你想要mock的find返回结果
    expected_result = [{"id": 1, "name": "test_data_1"}, {"id": 2, "name": "test_data_2"}]
    
    # 2. 用patch替换dao模块里的mongoclient
    with patch('dao.mongoclient') as mock_mongo_client:
        # Mock mongoclient()实例化后的对象
        mock_db_instance = Mock()
        mock_mongo_client.return_value = mock_db_instance
        
        # Mock db_name对应的数据库下的集合对象
        mock_collection = Mock()
        mock_db_instance.db_name = mock_collection
        
        # 关键：让collection.find()返回我们预设的结果
        mock_collection.find.return_value = expected_result
        
        # 3. 调用被测方法
        actual_result = dao.my_call()
        
        # 4. 断言结果符合预期
        assert actual_result == expected_result
        
        # 可选：验证find方法确实被调用了一次（确保逻辑走对了）
        mock_collection.find.assert_called_once()

方案二：链式设置返回值（简洁写法） #

如果你的调用链比较简单，可以直接用链式的return_value设置，代码更紧凑：

from unittest.mock import patch
import dao

def test_my_call_simpler():
    expected_result = [{"key": "value"}]
    
    with patch('dao.mongoclient') as mock_mongo:
        # 直接链式设置：mongoclient().db_name.collection.find() → 预期结果
        mock_mongo.return_value.db_name.collection.find.return_value = expected_result
        
        actual_result = dao.my_call()
        assert actual_result == expected_result

额外注意：如果被测代码需要处理游标 #

如果你的my_call()方法里是把find()返回的游标转成列表（比如return list(result)），那上面的写法完全没问题；但如果需要模拟游标本身的行为，可以Mock游标对象的__iter__方法：

from unittest.mock import patch, Mock
import dao

def test_my_call_with_cursor():
    expected_list = [{"id": 1}, {"id": 2}]
    # Mock游标对象，让它支持迭代转列表
    mock_cursor = Mock()
    mock_cursor.__iter__.return_value = expected_list
    
    with patch('dao.mongoclient') as mock_mongo:
        mock_mongo.return_value.db_name.collection.find.return_value = mock_cursor
        
        # 假设被测代码是return list(db.collection.find())
        actual_result = dao.my_call()
        assert actual_result == expected_list

最后要提醒你：patch('dao.mongoclient')里的路径必须是被测模块（dao.py）中导入并使用的mongoclient的路径，如果你的dao.py里是from pymongo import MongoClient as mongoclient，那路径就是dao.mongoclient，这个路径错了的话Mock是不会生效的哦。

内容的提问来源于stack exchange，提问作者HideDragon2