added ZRANK and ZREVRANK commands.

1 files changed, 26 insertions, 1 deletions

diff --git a/tests/server_commands.py b/tests/server_commands.py
index b7ce7e3..0ddb7cb 100644
--- a/tests/server_commands.py
+++ b/tests/server_commands.py
@@ -611,6 +611,19 @@ class ServerCommandsTestCase(unittest.TestCase):
             [('a2', 2.0), ('a3', 3.0), ('a4', 4.0)])
         # a non existant key should return None
         self.assertEquals(self.client.zrangebyscore('b', 0, 1, withscores=True), None)
+        
+    def test_zrank(self):
+        # key is not a zset
+        self.client['a'] = 'a'
+        self.assertRaises(redis.ResponseError, self.client.zrank, 'a', 'a4')
+        del self.client['a']
+        # real logic
+        self.make_zset('a', {'a1': 1, 'a2': 2, 'a3': 3, 'a4': 4, 'a5': 5})
+        self.assertEquals(self.client.zrank('a', 'a1'), 0)
+        self.assertEquals(self.client.zrank('a', 'a2'), 1)
+        self.assertEquals(self.client.zrank('a', 'a3'), 2)
+        self.assertEquals(self.client.zrank('a', 'a4'), 3)
+        self.assertEquals(self.client.zrank('a', 'a5'), 4)
             
     def test_zrem(self):
         # key is not a zset
@@ -653,7 +666,19 @@ class ServerCommandsTestCase(unittest.TestCase):
             [('a2', 2.0), ('a1', 1.0)])
         # a non existant key should return None
         self.assertEquals(self.client.zrange('b', 0, 1, withscores=True), None)
-            
+        
+    def test_zrevrank(self):
+        # key is not a zset
+        self.client['a'] = 'a'
+        self.assertRaises(redis.ResponseError, self.client.zrevrank, 'a', 'a4')
+        del self.client['a']
+        # real logic
+        self.make_zset('a', {'a1': 5, 'a2': 4, 'a3': 3, 'a4': 2, 'a5': 1})
+        self.assertEquals(self.client.zrevrank('a', 'a1'), 0)
+        self.assertEquals(self.client.zrevrank('a', 'a2'), 1)
+        self.assertEquals(self.client.zrevrank('a', 'a3'), 2)
+        self.assertEquals(self.client.zrevrank('a', 'a4'), 3)
+        self.assertEquals(self.client.zrevrank('a', 'a5'), 4)
             
     def test_zscore(self):
         # key is not a zset