From ddbd5c9e9d0d3343f38906906793bb84a672ad5b Mon Sep 17 00:00:00 2001
From: jlaine <jlaine@e071eeec-0327-468d-9b6a-08194a12b294>
Date: Tue, 1 Jun 2010 16:33:20 +0000
Subject: [PATCH] add tests for qs.count()

git-svn-id: https://svn.bolloretelecom.eu/opensource/django-ldapdb/trunk@921 e071eeec-0327-468d-9b6a-08194a12b294
---
 examples/tests.py | 34 +++++++++++++++++++++++++++++++++-
 1 file changed, 33 insertions(+), 1 deletion(-)

diff --git a/examples/tests.py b/examples/tests.py
index af8867c..abdc648 100644
--- a/examples/tests.py
+++ b/examples/tests.py
@@ -68,13 +68,25 @@ class GroupTestCase(BaseTestCase):
         g.usernames = ['wizuser', 'baruser']
         g.save()
 
-    def test_filter(self):
+    def test_count(self):
+        # empty query
+        qs = LdapGroup.objects.none()
+        self.assertEquals(qs.count(), 0)
+
         qs = LdapGroup.objects.none()
         self.assertEquals(len(qs), 0)
 
+        # all query
+        qs = LdapGroup.objects.all()
+        self.assertEquals(qs.count(), 3)
+
         qs = LdapGroup.objects.all()
         self.assertEquals(len(qs), 3)
 
+    def test_filter(self):
+        qs = LdapGroup.objects.filter(name='foogroup')
+        self.assertEquals(qs.count(), 1)
+
         qs = LdapGroup.objects.filter(name='foogroup')
         self.assertEquals(len(qs), 1)
 
@@ -85,6 +97,9 @@ class GroupTestCase(BaseTestCase):
         self.assertEquals(g.usernames, ['foouser', 'baruser'])
 
         # try to filter non-existent entries
+        qs = LdapGroup.objects.filter(name='does_not_exist')
+        self.assertEquals(qs.count(), 0)
+
         qs = LdapGroup.objects.filter(name='does_not_exist')
         self.assertEquals(len(qs), 0)
 
@@ -141,18 +156,35 @@ class GroupTestCase(BaseTestCase):
         self.assertEquals(objs[1].gid, 1001)
         self.assertEquals(objs[2].gid, 1002)
 
+        # limit only
         qs = LdapGroup.objects.all()
+        objs = qs[:2]
+        self.assertEquals(objs.count(), 2)
+
         objs = qs[:2]
         self.assertEquals(len(objs), 2)
         self.assertEquals(objs[0].gid, 1000)
         self.assertEquals(objs[1].gid, 1001)
 
+        # offset only
         qs = LdapGroup.objects.all()
+        objs = qs[1:]
+        self.assertEquals(objs.count(), 2)
+
         objs = qs[1:]
         self.assertEquals(len(objs), 2)
         self.assertEquals(objs[0].gid, 1001)
         self.assertEquals(objs[1].gid, 1002)
 
+        # offset and limit
+        qs = LdapGroup.objects.all()
+        objs = qs[1:2]
+        self.assertEquals(objs.count(), 1)
+
+        objs = qs[1:2]
+        self.assertEquals(len(objs), 1)
+        self.assertEquals(objs[0].gid, 1001)
+
     def test_update(self):
         g = LdapGroup.objects.get(name='foogroup')
 
-- 
2.30.2