模拟补丁函数

Question

我是python mock的新手，最近我正试图为我的函数编写一个测试代码。 该函数用于分析Mongo数据库和CSV报告。我想修补“get_collection（）”函数。结构如下：

蒙哥_报表.py文件：

import pymongo

def get_collection(): # used to get mongo collection

def from_report():  #used to from a report
    get_collection(mongodb, mongo_collection_name)
    .....

在我的测试文件中：

^{pr2}$

这个补丁是根据我在一个网页https://blog.fugue.co/2016-02-11-python-mocking-101.html
但事实并非如此工作。所以我的问题是：

1. 这是使用补丁的正确方法吗？如果不是，我如何修补它？

2. 在mongo_report_test（mock_call）中，mock_call来自何处，它在这里没有任何链接，我如何声明它

3. 补丁之后，我如何调用这个测试函数？

开始向装修师学习，花了一整天的时间把我的头撞到桌子上，但还是没学会窍门。：<

Answer 1

我花了一段时间才发现原来的答案是有误导性的。这是一个mock的工作副本，与您的问题非常相似，我唯一不知道的是您的get_connection（）的返回值。python 2.6.6版

    #/usr/bin/env pthon
    import unittest
    import mock
    from mock import MagicMock

    class Mongo(object):
        def __init__(self, *args, **kwargs):
            # doing nothing for demo
            pass

        def get_collection(self, input1):
            # return fixed list for demo
            return [1, 3, 5]

        def from_report(self, input_a, input_b):
            collection_a = self.get_collection(input_a)
            collection_b = self.get_collection(input_b)
            return collection_a + collection_b

    class TestMongo(unittest.TestCase):

        def setUp(self):
            self.expected = [2, 4, 6, 2, 4, 6]
            self.mock_value = [2, 4, 6]

        def tearDown(self):
            pass

        """ in the format of file_name.class_name.method_name
        @mock.patch('test_mongo.Mongo.get_collection')
        def test_using_mock_patch(self, mock_get_collection):
            mock_get_collection.return_value = self.mock_value
            mongo = Mongo()
            mongo.get_collection = mock_get_collection
            result = mongo.from_report('any', 'where')
            self.assertEquals(result, self.expected)

        """ mock.MagicMock
        def test_mongo_get_collection_using_magicMock(self):
            mock_mongo = MagicMock(name='get_collection')
            mock_mongo.get_collection.return_value = self.mock_value
            mongo = Mongo()
            mongo.get_collection = mock_mongo.get_collection
            result = mongo.from_report('any', 'where')
            self.assertEquals(result, self.expected)

        """ mock.Mock
        def test_mongo_get_collection_using_Mock(self):
            mock_mongo = mock.Mock(name='get_collection')
            mock_mongo.get_collection.return_value = self.mock_value
            mongo = Mongo()
            mongo.get_collection = mock_mongo.get_collection
            result = mongo.from_report('any', 'where')
            self.assertEquals(result, self.expected)


    if __name__ == '__main__':
        suite = unittest.TestLoader().loadTestsFromTestCase(TestMongo)
        unittest.TextTestRunner(verbosity=2).run(suite)

下面是命令和输出

^{pr2}$

Answer 2

除非这是剪切粘贴错误，否则我相信：

mock.count.side_effect = [20, 6, 2]
mock.find.side_effect = [{user: xx}]

应该是：

^{pr2}$