Added support for using custom QuerySet classes

This commit is contained in:
Harry Marr
2010-09-15 09:47:13 +01:00
parent 2af5f3c56e
commit f11ee1f9cf
3 changed files with 23 additions and 1 deletions

View File

@@ -1378,6 +1378,26 @@ class QTest(unittest.TestCase):
self.assertEqual(Post.objects.filter(created_user=user).count(), 1)
self.assertEqual(Post.objects.filter(Q(created_user=user)).count(), 1)
def test_custom_querysets(self):
"""Ensure that custom QuerySet classes may be used.
"""
class CustomQuerySet(QuerySet):
def not_empty(self):
return len(self) > 0
class Post(Document):
meta = {'queryset_class': CustomQuerySet}
Post.drop_collection()
self.assertTrue(isinstance(Post.objects, CustomQuerySet))
self.assertFalse(Post.objects.not_empty())
Post().save()
self.assertTrue(Post.objects.not_empty())
Post.drop_collection()
if __name__ == '__main__':
unittest.main()