Question

我试图在我的测试运行中从json文件加载测试夹具（以匹配Django中使用的类似样式的夹具加载）。这是我到目前为止的代码;

from django.utils import simplejson as json
from mongoengine import base as mongobase
from mongoengine import connect
from pymongo import json_util

db = connect("DATABASENAME")

# Clear out the current collections.
for collection in db.collection_names():
    if collection == 'system.indexes':
        continue
    db.drop_collection(collection)


# Open json fixtures

f = open('path/to/test_fixtures.json', 'r')
datas = json.loads(f.read(), object_hook=json_util.object_hook)

# For each serialised model instance, loop through and save to the database.

for data in datas:

    print data

    if data['_cls'] not in mongobase._document_registry:
        print "Skipping %s" % data['_cls2']
        continue

    model = mongobase._document_registry[data['_cls']]
    model_instance = model._from_son(data)
    model_instance.save(force_insert=True)

这似乎很有效，但是，当其中一个模型有ReferenceField时，它会失败。抱怨重复的密钥。您会注意到那里有一个print语句来显示dict内容，在一个示例运行中我会得到以下输出，它们在错误之前看起来都很好

{u'_types': [u'Account'], u'status': u'ok', u'name': u'Alice', u'local_id': u'3', u'_cls': u'Account', u'members': [], u'_id': ObjectId('4f17f0855585d32457000001'), u'email': u'alice@example.com', u'permissions': []}
{u'_types': [u'Account'], u'status': u'ok', u'name': u'Bob', u'local_id': u'2', u'_cls': u'Account', u'members': [], u'_id': ObjectId('4f17f0855585d32457000000'), u'email': u'bob@example.com', u'permissions': []}
{u'_types': [u'Account'], u'status': u'ok', u'name': u'company', u'_cls': u'Account', u'members': [], u'_id': ObjectId('4f17f0855585d32457000002'), u'email': u'org@example.com', u'permissions': []}
{u'_types': [u'Membership'], u'parent_account': DBRef(u'account', ObjectId('4f17f0855585d32457000002')), u'member': DBRef(u'account', ObjectId('4f17f0855585d32457000001')), u'role': u'member', u'_cls': u'Membership', u'_id': ObjectId('4f17f0855585d32457000003')}

最后，错误是：

Traceback (most recent call last):
  File "/home/vagrant/.virtualenvs/engineclub/lib/python2.6/site-packages/django/test/testcases.py", line 292, in __call__
    self._pre_setup()
  File "/home/vagrant/.virtualenvs/engineclub/lib/python2.6/site-packages/django/test/testcases.py", line 259, in _pre_setup
    self._fixture_setup()
  File "/vagrant/engineclub/engineclub/apps/notifications/tests.py", line 67, in _fixture_setup
    model_instance.save(force_insert=True)
  File "/home/vagrant/.virtualenvs/engineclub/lib/python2.6/site-packages/mongoengine/document.py", line 177, in save
    _refs=_refs)
  File "/home/vagrant/.virtualenvs/engineclub/lib/python2.6/site-packages/mongoengine/document.py", line 183, in save
    raise OperationError(message % unicode(err))
OperationError: Tried to save duplicate unique keys (E11000 duplicate key error index: test_aliss.account.$_id_  dup key: { : ObjectId('4f17f0855585d32457000001') })

在插入仅包含对帐户集合的引用的成员资格实例时，似乎在抱怨帐户上的重复密钥。$ id 。

让我知道，如果我可以包含任何其他信息，我会添加JSON文件，但它非常像打印的dicts（并且只有3个帐户，后跟一个成员资格）。有一件事我注意到，当我从保存中删除force_insert时，它似乎根本没有保存任何东西（因此看起来很好）。

Answer 1

尝试使用Mixer：

from mixer.backend.mongoengine import mixer

mixer.blend(MyDocument)

创建保存mongoengine ReferenceField（DBRef）时MongoDB“E11000重复键错误索引”

1 个答案: