pub/mongoengine
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

bug fix for RefferenceField.to_mongo when dbref=False

Browse Source
This commit is contained in:
yak 2012-11-13 13:02:07 +01:00
parent c31488add9
commit 787fc1cd8b
2 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mongoengine/fields.py
View File

@ -777,7 +777,7 @@ class ReferenceField(BaseField):
def to_mongo(self, document): def to_mongo(self, document):
if isinstance(document, DBRef): if isinstance(document, DBRef):
if not self.dbref: if not self.dbref:
return DBRef.id return document.id
return document return document
elif not self.dbref and isinstance(document, basestring): elif not self.dbref and isinstance(document, basestring):
return document return document

9
tests/test_fields.py
View File

@ -1104,6 +1104,15 @@ class FieldTest(unittest.TestCase):
p = Person.objects.get(name="Ross") p = Person.objects.get(name="Ross")
self.assertEqual(p.parent, p1) self.assertEqual(p.parent, p1)
def test_dbref_to_mongo(self):
class Person(Document):
name = StringField()
parent = ReferenceField('self', dbref=False)
p1 = Person._from_son({'name':"Yakxxx", 'parent': "50a234ea469ac1eda42d347d"})
mongoed = p1.to_mongo()
self.assertIsInstance(mongoed['parent'], ObjectId)
def test_objectid_reference_fields(self): def test_objectid_reference_fields(self):