Python-mock：在同一个测试用例中模拟 pymongo 的 Cursor 和 Collection

Question

我想对一些 mongo 查询进行模拟测试，但是当我尝试在同一范围内模拟两个不同的 mongo 对象（Cursor 和 Collection）时遇到了问题。 我不能把原始代码放在这里，但我做了一个通用的有同样问题的。

这是我要测试的函数（使用 find 和 distinct）：

def get_some_info(date1, date2):
    data = collection.find({"timestamp": {"$lt": date1, "$gt": date2})
    id_list = data.distinct("id")

    for id in id_list :
       collection.find({"id" : id})

这是测试代码：

@mock.patch.object(Collection, "find")
@mock.patch.object(Cursor, "distinct")
def test_get_some_info(self, mock_distinct, mock_find):
    date1 = datetime.datetime()
    date2 = datetime.datetime()
    mock_distinct.return_value = ["id1", "id2"]
    find_parameters1 = {"timestamp": {"$lt": date1, "$gt": date2}
    find_parameters2 = {"id" : "id1"}
    self.my_class.get_some_info(date1, date2)
    mock_find.assert_called_with(find_parameters1)
    mock_find.assert_any_call(find_parameters2)

当我运行测试时，我收到以下错误消息：

'%s call not found' % expected_string AssertionError: find({"id" : "id1"}) 调用未找到

所以我打印了 id_list 的值：

def get_some_info(date1, date2):
    data = collection.find({"timestamp": {"$lt": date1, "$gt": date2})
    id_list = data.distinct("id")
    print id_list

    for id in id_list :
        collection.find({"id" : id})

这是 id_list 的值：

MagicMock name='find().distinct()' id='139719585597776'

但预期是：["id1", "id2"]

当我评论Cursor的mock时，id_list的值和之前一样：

#@mock.patch.object(collection, "distinct")
@mock.patch.object(Cursor, "find")
def test_get_some_info(self, mock_find):
    date1 = datetime.datetime()
    date2 = datetime.datetime()
    mock_distinct.return_value = ["id1", "id2"]
    find_parameters1 = {"timestamp": {"$lt": date1, "$gt": date2}
    find_parameters2 = {"id" : "id1"}
    self.my_class.get_some_info(date1, date2)
    mock_find.assert_called_with(find_parameters1)
    mock_find.assert_any_call(find_parameters2)

<MagicMock name='find().distinct()' id='140089689306448'>

我认为代码总是从 Collection 中捕获“不同”值，而不是从 Cursor 中。 有人遇到同样的问题吗？

Answer 1

1. 测试参数中的模拟对象应该是相反的顺序（查看Nesting Patch Decorators了解详细信息）
2. 你所谓的mock_collection应该是mock_find

你的测试代码应该是这样的：

@mock.patch.object(Cursor, "distinct")
@mock.patch.object(Collection, "find")
def test_get_some_info(self, mock_find, mock_distinct):
    date1 = datetime.datetime()
    date2 = datetime.datetime()
    mock_distinct.return_value = ["id1", "id2"]
    find_parameters1 = {"timestamp": {"$lt": date1, "$gt": date2}
    find_parameters2 = {"id" : "id1"}
    self.my_class.get_some_info(date1, date2)
    mock_find.assert_called_with(find_parameters1)
    mock_find.assert_any_call(find_parameters2)