使用mongomock模拟mongoengine时drop_database二次调用失效

你遇到的这个问题是mongomock和mongoengine集成时的典型坑——第一次调用drop_database能正常清空数据，但第二次调用就完全失效，数据依然残留。我复现了你的测试代码，确实能稳定重现这个现象。

原因分析 #

问题根源在于mongoengine的文档类集合缓存机制和mongomock的数据库实例交互逻辑不匹配：

• 第一次调用drop_database时，mongomock确实清空了数据库，但mongoengine的DummyDoc类已经缓存了对旧集合对象的引用。
• 当你再次插入数据时，DummyDoc依然使用缓存的旧集合；而第二次调用drop_database时，操作的是新的数据库上下文，导致两者完全脱节，数据自然清不掉。

解决方案 #

这里有几个可行的修复方法，按简洁性和可靠性排序：

方法1：直接调用mongomock原生的drop方法（推荐）

绕过mongoengine的连接封装，直接操作mongomock的数据库实例，确保彻底清空数据库，同时让mongoengine自动重新绑定集合：

# 替换原有的 db_conn.drop_database('mongotest') 为：
db_conn.get_database('mongotest').client.drop_database('mongotest')

方法2：重置文档类的集合引用

每次drop后，强制让文档类重新绑定到当前数据库的集合：

# 第二次drop后添加这行代码
db_conn.drop_database('mongotest')
DummyDoc._collection = db_conn['mongotest'][DummyDoc._meta['collection']]

方法3：重新注册文档类

通过注销再注册文档类，强制它重新绑定到当前数据库：

from mongoengine import register_document

# 第二次drop后执行
db_conn.drop_database('mongotest')
DummyDoc._collection = None
register_document(DummyDoc)

完整修复后的测试代码 #

这里用方法1的完整实现，最简洁有效：

import mongoengine as me
try:
    import pytest
except:
    pass

class DummyDoc(me.Document):
    pass

def test_mongomock():
    db_conn = me.connect(db='mongotest', host='mongomock://localhost')
    db_conn.get_database('mongotest').client.drop_database('mongotest')

    dd = DummyDoc()
    dd.save()
    assert len(DummyDoc.objects()) == 1
    dd = DummyDoc()
    dd.save()
    assert len(DummyDoc.objects()) == 2

    # First drop_database works as expected!
    db_conn.get_database('mongotest').client.drop_database('mongotest')
    assert len(DummyDoc.objects()) == 0

    dd = DummyDoc()
    dd.save()
    assert len(DummyDoc.objects()) == 1
    dd = DummyDoc()
    dd.save()
    assert len(DummyDoc.objects()) == 2

    # Second drop_database now works!!
    db_conn.get_database('mongotest').client.drop_database('mongotest')
    assert len(DummyDoc.objects()) == 0

if __name__ == '__main__':
    test_mongomock()

额外测试优化提示 #

如果是在pytest环境中，更推荐用fixture来自动管理数据库生命周期，从根源避免状态残留问题：

import pytest
import mongoengine as me

@pytest.fixture(autouse=True)
def clean_mongo_db():
    # 初始化测试数据库
    conn = me.connect(db='test_db', host='mongomock://localhost')
    yield conn
    # 测试结束后清理数据库并断开连接
    conn.get_database('test_db').client.drop_database('test_db')
    me.disconnect()

这样每个测试用例都会拥有一个完全干净的数据库环境，无需手动处理多次drop的问题。

内容的提问来源于stack exchange，提问作者Riccardo Petraglia