如何模拟MongoClient进行python单元测试？

Question

我将以下代码发送给UT，这使我很麻烦：

  def initialize():
    try :
        self.client = MongoClient("127.0.0.1", 27017)
        self.conn = self.client["DB_NAME"]
    except Exception:
        print "Except in initialize!"
        return False
    return True

我编写以下测试用例来覆盖上述功能，希望获得返回值“ True”：

def mock_mongodb_mongoclient_init(self, para1, para2):
    pass

def mock_mongodb_mongoclient_getitem(self, name):
    return {"DB_NAME":"Something"}

def test_mongodb_initialize_true(self):
    self.patcher1 = patch('pymongo.MongoClient.__init__', new=self.mock_mongodb_mongoclient_init)
    self.patcher2 = patch('pymongo.MongoClient.__getitem__', new=self.mock_mongodb_mongoclient_getitem)
    self.patcher1.start()
    self.patcher2.start()
    self.assertEqual(initialize(), True)
    self.patcher1.stop()
    self.patcher2.stop()

但是，这永远行不通！ 它总是报告“初始化异常！”！ 并返回“ False”。

我如何才能使用此MongoClient并使函数返回“ True”？

Answer 1

由于您正在进行单元测试，并且不尝试以任何方式实际连接到Mongo，因此我认为您应该只关心已调用客户端API。 所以我建议以下内容-

from unittest import mock
@mock.patch("pymongo.MongoClient")
def test_mongodb_initialize(self, mock_pymongo):
    MyMongo.initialize()
    self.assertTrue(mock_pymongo.called)

（如果我的语法关闭，请原谅我，我使用pytest而不是unittest。）

Answer 2

为了简单地传递initialize（），我们可以模拟pymongo.MongoClient，如下所示：

import unittest
import pymongo
from mock import patch


class MyMongo(object):
    def initialize(self):
        try :
            self.client = pymongo.MongoClient("127.0.0.1", 27017)
            self.conn = self.client["DB_NAME"]
        except Exception:
            print "Except in initialize!"
            return False
        return True


class TestMyMongo(unittest.TestCase):
    def test_mongodb_initialize_true(self):
        with patch('pymongo.MongoClient') as mock_mongo:
            self.mymongo = MyMongo()
            self.assertEqual(self.mymongo.initialize(), True)

但是我不确定您是要模拟MongoClient还是仅模拟MongoClient。 初始化部分？

Answer 3

MongoClient并非旨在以这种方式被嘲笑。 它必须在__init__中初始化其属性才能起作用，因此，如果您跳过调用__init__ ，则所有其他操作都将引发各种异常。 在您的特定情况下，MongoClient需要访问__slave_okay属性，但未设置。

设置一个实际的MongoDB服务器并对其进行测试，或者使用一个伪造的库模拟所有PyMongo。 仅仅在PyMongo中覆盖少数方法是不可行的。