Merge pull request #989

This commit is contained in:
Matthew Ellison 2015-05-06 09:38:26 -04:00
commit 01f4dd8f97

View File

@ -1431,16 +1431,26 @@ class FieldTest(unittest.TestCase):
def test_map_field_lookup(self): def test_map_field_lookup(self):
"""Ensure MapField lookups succeed on Fields without a lookup method""" """Ensure MapField lookups succeed on Fields without a lookup method"""
class Action(EmbeddedDocument):
operation = StringField()
object = StringField()
class Log(Document): class Log(Document):
name = StringField() name = StringField()
visited = MapField(DateTimeField()) visited = MapField(DateTimeField())
actions = MapField(EmbeddedDocumentField(Action))
Log.drop_collection() Log.drop_collection()
Log(name="wilson", visited={'friends': datetime.datetime.now()}).save() Log(name="wilson", visited={'friends': datetime.datetime.now()},
actions={'friends': Action(operation='drink', object='beer')}).save()
self.assertEqual(1, Log.objects( self.assertEqual(1, Log.objects(
visited__friends__exists=True).count()) visited__friends__exists=True).count())
self.assertEqual(1, Log.objects(
actions__friends__operation='drink',
actions__friends__object='beer').count())
def test_embedded_db_field(self): def test_embedded_db_field(self):
class Embedded(EmbeddedDocument): class Embedded(EmbeddedDocument):