New command: zcount

1 files changed, 11 insertions, 0 deletions

diff --git a/tests/server_commands.py b/tests/server_commands.py
index 81096e9..efcf2d6 100644
--- a/tests/server_commands.py
+++ b/tests/server_commands.py
@@ -615,6 +615,17 @@ class ServerCommandsTestCase(unittest.TestCase):
         self.make_zset('a', {'a1': 1, 'a2': 2, 'a3': 3})
         self.assertEquals(self.client.zcard('a'), 3)
 
+    def test_zcount(self):
+        # key is not a zset
+        self.client['a'] = 'a'
+        self.assertRaises(redis.ResponseError, self.client.zcount, 'a', 0, 0)
+        del self.client['a']
+        # real logic
+        self.make_zset('a', {'a1': 1, 'a2': 2, 'a3': 3})
+        self.assertEquals(self.client.zcount('a', '-inf', '+inf'), 3)
+        self.assertEquals(self.client.zcount('a', 1, 2), 2)
+        self.assertEquals(self.client.zcount('a', 10, 20), 0)
+
     def test_zincrby(self):
         # key is not a zset
         self.client['a'] = 'a'