python - 如何为 python 单元测试模拟 mongodb？

Question

我正在使用 Python 2.7 的 mock 模块来模拟我的其他函数并使用

unittest 用于编写单元测试。

我想知道模拟 MongoDB 是否不同于使用模拟功能(mock.patch 一个被调用的函数？)或者我需要为此目的使用另一个不同的包？

我不认为我想要运行测试 mongodb 实例。我想要的只是一些速度数据并能够调用 pymongo 功能。我只是有点迷失在想是否有一种方法可以为模块编写模拟(如 pymongo)，或者任何可以通过 mock 模块实现的东西。

如果您能提供相关示例或教程，我们将不胜感激。

测试代码

from pymongo import MongoClient

monog_url = 'mongodb://localhost:27017'
client = MongoClient(monog_url)
db = client.db

class Dao(object):
   def __init__(self):
      pass

   def save(self, user):
      db_doc = {
        'name': user.name,
        'email': user.email
      }
      db.users.save(db_doc)

   def getbyname(self, user):
      db_doc = {
        'name': user.name,
      }
      return db.users.find(db_doc)

为了测试这个，我真的不想测试 mongodb 启动并运行!而且，我想我不想模拟 db.userssave 和 db.users.find，因为我希望实际上能够检索我保存的数据并确保它在数据库中。我想我需要为我内存中的每个模型创建一些 fixture 并使用它们。只是我需要一个外部工具吗？

我想保留一些这样的假数据，只是不知道如何正确处理。

users = { 
    {'name' : 'Kelly', 'email' : 'kelly@gmail.com'},
    {'name': 'Sam', 'email': 'sam@gmail.com'}
}

Answer 1

我推荐使用 mongomock 来模拟 mongodb。它基本上是一个内存中的 mongodb，带有 pymongo 接口(interface)，专门为此目的而制作。

https://github.com/mongomock/mongomock