From 11c7a15067b1095f68fb045fd878fdcf4e84c1b7 Mon Sep 17 00:00:00 2001
From: Florian Schlachter <flori@n-schlachter.de>
Date: Fri, 14 May 2010 13:49:13 +0200
Subject: [PATCH] Added test for DictField's basecls.

---
 tests/queryset.py | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/tests/queryset.py b/tests/queryset.py
index aba3bc7d..c43b8f11 100644
--- a/tests/queryset.py
+++ b/tests/queryset.py
@@ -1063,6 +1063,29 @@ class QuerySetTest(unittest.TestCase):
 
         BlogPost.drop_collection()
 
+    def test_dict_with_custom_baseclass(self):
+        """Ensure DictField working with custom base clases.
+        """
+        class Test(Document):
+            testdict = DictField()
+
+        t = Test(testdict={'f': 'Value'})
+        t.save()
+
+        self.assertEqual(len(Test.objects(testdict__f__startswith='Val')), 0)
+        self.assertEqual(len(Test.objects(testdict__f='Value')), 1)
+        Test.drop_collection()
+
+        class Test(Document):
+            testdict = DictField(basecls=StringField)
+
+        t = Test(testdict={'f': 'Value'})
+        t.save()
+
+        self.assertEqual(len(Test.objects(testdict__f='Value')), 1)
+        self.assertEqual(len(Test.objects(testdict__f__startswith='Val')), 1)
+        Test.drop_collection()
+
     def test_bulk(self):
         """Ensure bulk querying by object id returns a proper dict.
         """