模拟基类中的函数

Question

我有一个基类，在这个基类中我连接到一个mongodb集合并从hdfs文件中读取一个日期值，然后删除比从该hdfs文件中读取的值更旧的值。我对单元测试有一些问题，所以：

在子类中，我有以下内容：

def remove_old_sources(self):
    input_date = self.get_date_from_input()
    mongo_collection = self.connect_to_collection(db, collection)
    deleted_files_count = self.delete_stale_documents(mongo_collection, input_date)

上面调用的所有函数都来自基类。我的问题是，我如何才能用以下方式模拟它们：

带有datetime对象的input_date和来自mongomock的带有Collection的mongo_collection，例如：

mocked_collection = mongomock.MongoClient().db.collection
mocked_collection.insert_many([
    {'_id': '1', 'time': datetime(2010, 10, 10, 10, 10, 10)},
    {'_id': '2', 'time': datetime(2015, 10, 10, 10, 10, 10)},
    {'_id': '3', 'time': datetime(2020, 10, 10, 10, 10, 10)},
    {'_id': '4', 'time': datetime(2030, 10, 10, 10, 10, 10)},
    {'_id': '5', 'time': datetime(2040, 10, 10, 10, 10, 10)},
])

如果我没理解错的话，这个mongomock在内存中创建了一个本地mongodb。

我试过了：

@mock.patch("path_to_base_class.get_date_from_input")
@mock.patch("path_to_base_class.connect_to_collection")
def test_remove_old_docs(connect_mock, date_mock)

然后初始化子类并运行remove_old_sources，但模拟对象是这样的MagicMock对象，我不确定如何用上面提到的方法覆盖它们。我还尝试使用上下文管理器with mock.patch(BaseClass, "function_name", mocked_stuff)，然后初始化Child类，但它不是这样工作的。

这方面的任何帮助都是非常感谢的！

Answer 1

您将获得MagicMock对象，因为这是调用模拟对象时生成的标准对象类型。如果你想传递其他东西，你可以随时替换它们的return_value属性。

但是，您不需要模拟您的mongo集合，您的代码无论如何都不会接触到任何数据。

您的代码进行了3次调用：

    input_date = self.get_date_from_input()
    mongo_collection = self.connect_to_collection(db, collection)
    deleted_files_count = self.delete_stale_documents(mongo_collection, input_date)

在前两个调用中，代码设置了两个局部变量input_date和mongo_collection，然后将它们传递给第三个方法self.delete_stale_documents()。

如果您的目标是验证代码是否正确地将这两个参数传递给第三个方法，那么您所要做的就是模拟这三个方法，并验证前两个方法的返回值是否传递给了第三个方法：

@mock.patch("path_to_base_class.delete_stale_documents")
@mock.patch("path_to_base_class.connect_to_collection")
@mock.patch("path_to_base_class.get_date_from_input")
def test_remove_old_docs(date_mock, connect_mock, delete_mock):
    input_date_mock = date_mock.return_value  # a specific MagicMock object
    collection_mock = connect_mock.return_value  # another specific MagicMock object

    # call code-under-test here

    delete_mock.assert_called_with(collection_mock, input_date_mock)

仅当connect_mock()和date_mock()调用的结果用于调用delete_mock()时，delete_mock.assert_called_with()才会通过。

如果您的测试比较复杂，例如，使用self.delete_stale_documents()的返回值，则可以通过在测试前设置.return_value，将特定的值设置为该调用的返回值：

    delete_mock.return_value = 42

现在，deleted_files_count将被设置为42，如果您的代码正确处理了该值，则可以进行进一步的断言。