Added unit test for EmbeddedDocument

2009-11-17 01:48:43 +00:00
parent 5a87534c22
commit 0bb0ec0114
2 changed files with 15 additions and 3 deletions
--- a/tests/document.py
+++ b/tests/document.py
@@ -1,6 +1,6 @@
 import unittest

-from mongomap import Document, StringField, IntField
+from mongomap import *


 class DocumentTest(unittest.TestCase):
@@ -61,6 +61,14 @@ class DocumentTest(unittest.TestCase):
        person['name'] = 'Another User'
        self.assertEquals(person['name'], 'Another User')

+    def test_embedded_document(self):
+        """Ensure that embedded documents are set up correctly.
+        """
+        class Comment(EmbeddedDocument):
+            content = StringField()
+        
+        self.assertTrue('content' in Comment._fields)
+        self.assertFalse(hasattr(Comment, '_meta'))


 if __name__ == '__main__':