From bb2487914987972c4d7ff6960de1012d53a7f566 Mon Sep 17 00:00:00 2001 From: Harry Marr Date: Sun, 19 Sep 2010 20:00:53 +0100 Subject: [PATCH] Moved custom queryset test to correct place --- tests/queryset.py | 40 ++++++++++++++++++++-------------------- 1 file changed, 20 insertions(+), 20 deletions(-) diff --git a/tests/queryset.py b/tests/queryset.py index 4491be8c..59a8216e 100644 --- a/tests/queryset.py +++ b/tests/queryset.py @@ -1307,6 +1307,26 @@ class QuerySetTest(unittest.TestCase): Event.drop_collection() + 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() + class QTest(unittest.TestCase): @@ -1378,26 +1398,6 @@ 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()