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

Fixes conversion of null genericreferences in querysets

Browse Source 
closes #211
This commit is contained in:
Ross Lawley 2011-07-11 09:19:27 +01:00
parent 556e620c7a
commit dc5a613bc7
2 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mongoengine/fields.py
View File

@ -653,6 +653,9 @@ class GenericReferenceField(BaseField):
return doc return doc
def to_mongo(self, document): def to_mongo(self, document):
if document is None:
return None
id_field_name = document.__class__._meta['id_field'] id_field_name = document.__class__._meta['id_field']
id_field = document.__class__._fields[id_field_name] id_field = document.__class__._fields[id_field_name]

12
tests/fields.py
View File

@ -1095,6 +1095,18 @@ class FieldTest(unittest.TestCase):
Link.drop_collection() Link.drop_collection()
User.drop_collection() User.drop_collection()
def test_generic_reference_is_none(self):
class Person(Document):
name = StringField()
city = GenericReferenceField()
Person.drop_collection()
Person(name="Wilson Jr").save()
self.assertEquals(repr(Person.objects(city=None)),
"[<Person: Person object>]")
def test_binary_fields(self): def test_binary_fields(self):
"""Ensure that binary fields can be stored and retrieved. """Ensure that binary fields can be stored and retrieved.
""" """