From eb892241ee3bd1f98d2d8c0fe420841ddb21037c Mon Sep 17 00:00:00 2001
From: Ross Lawley <ross.lawley@gmail.com>
Date: Wed, 25 May 2011 13:31:01 +0100
Subject: [PATCH] Added regression test for editting embedded documents

Closes #35
---
 tests/queryset.py | 37 +++++++++++++++++++++++++++++++++++++
 1 file changed, 37 insertions(+)

diff --git a/tests/queryset.py b/tests/queryset.py
index a45aaf3e..296377f3 100644
--- a/tests/queryset.py
+++ b/tests/queryset.py
@@ -1114,6 +1114,9 @@ class QuerySetTest(unittest.TestCase):
             slug = StringField()
             tags = ListField(ReferenceField(BlogTag), required=True)
 
+        BlogPost.drop_collection()
+        BlogTag.drop_collection()
+
         tag_1 = BlogTag(name='code')
         tag_1.save()
         tag_2 = BlogTag(name='mongodb')
@@ -1131,6 +1134,40 @@ class QuerySetTest(unittest.TestCase):
         post.reload()
         self.assertEqual(len(post.tags), 1)
 
+        BlogPost.drop_collection()
+        BlogTag.drop_collection()
+
+    def test_editting_embedded_objects(self):
+
+        class BlogTag(EmbeddedDocument):
+            name = StringField(required=True)
+
+        class BlogPost(Document):
+            slug = StringField()
+            tags = ListField(EmbeddedDocumentField(BlogTag), required=True)
+
+        BlogPost.drop_collection()
+
+        tag_1 = BlogTag(name='code')
+        tag_2 = BlogTag(name='mongodb')
+
+        post = BlogPost(slug="test", tags=[tag_1])
+        post.save()
+
+        post = BlogPost(slug="test-2", tags=[tag_1, tag_2])
+        post.save()
+        self.assertEqual(len(post.tags), 2)
+
+        BlogPost.objects(slug="test-2").update_one(set__tags__0__name="python")
+        post.reload()
+        self.assertEquals(post.tags[0].name, 'python')
+
+        BlogPost.objects(slug="test-2").update_one(pop__tags=-1)
+        post.reload()
+        self.assertEqual(len(post.tags), 1)
+
+        BlogPost.drop_collection()
+
     def test_order_by(self):
         """Ensure that QuerySets may be ordered.
         """