0

テスト実行で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)

これはほとんど機能しているように見えますが、モデルの1つにReferenceFieldがある場合、失敗します。キーの重複について不平を言う。dictの内容を示すprintステートメントがそこにあることに気付くでしょう。実行例では、エラーの前にすべて正常に見える次の出力が得られます。

{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') })

アカウントコレクションのみへの参照を持つメンバーシップインスタンスを挿入するときに、account。$ idのキーが重複していることに不満があるようです。

他に含めることができる情報がある場合はお知らせください。JSONファイルを追加しますが、印刷されたdictと非常によく似ています（3つのアカウントとそれに続く1つのメンバーシップしかありません）。私が気づいたことの1つは、force_insertを保存から削除すると、何も保存されていないように見えることです（したがって、正常に通過するように見えます）。

4

1 に答える 1

0

ミキサーを使用してみてください：

from mixer.backend.mongoengine import mixer

mixer.blend(MyDocument)
于 2014-01-07T19:34:41.483 に答える