Merge pull request #989 from MasterMind2k/master

Add test for query embedded document's field inside MapField
MongoEngine · May 6, 2015 · 7e23e1c · 7e23e1c
2 parents f69ebbe + 4b83440
commit 7e23e1c
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/tests/fields/fields.py b/tests/fields/fields.py
@@ -1431,16 +1431,26 @@ class Test(Document):
     def test_map_field_lookup(self):
         """Ensure MapField lookups succeed on Fields without a lookup method"""
 
+        class Action(EmbeddedDocument):
+            operation = StringField()
+            object    = StringField()
+
         class Log(Document):
             name = StringField()
             visited = MapField(DateTimeField())
+            actions = MapField(EmbeddedDocumentField(Action))
 
         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(
             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):
 
         class Embedded(EmbeddedDocument):