Support for SELECT (#1825)

Author: chayim

redis/commands/core.py

@@ -780,6 +780,13 @@ def swapdb(self, first, second, **kwargs):
         """
         return self.execute_command("SWAPDB", first, second, **kwargs)
 
+    def select(self, index, **kwargs):
+        """Select the Redis logical database at index.
+
+        See: https://redis.io/commands/select
+        """
+        return self.execute_command("SELECT", index, **kwargs)
+
     def info(self, section=None, **kwargs):
         """
         Returns a dictionary containing information about the Redis server

tests/test_commands.py

@@ -695,6 +695,12 @@ def test_role(self, r):
         assert isinstance(r.role()[1], int)
         assert isinstance(r.role()[2], list)
 
+    @pytest.mark.onlynoncluster
+    def test_select(self, r):
+        assert r.select(5)
+        assert r.select(2)
+        assert r.select(9)
+
     @pytest.mark.onlynoncluster
     def test_slowlog_get(self, r, slowlog):
         assert r.slowlog_reset()