Redis container password support (#195)

* redis container password support

* fixed test

* fixed test

* fixed test_docker_run_redis_with_password test

Author: spicy-sauce

testcontainers/redis.py

@@ -12,16 +12,18 @@
 #    under the License.
 
 import redis
-
 from testcontainers.core.container import DockerContainer
 from testcontainers.core.waiting_utils import wait_container_is_ready
 
 
 class RedisContainer(DockerContainer):
-    def __init__(self, image="redis:latest", port_to_expose=6379):
+    def __init__(self, image="redis:latest", port_to_expose=6379, password=None):
         super(RedisContainer, self).__init__(image)
         self.port_to_expose = port_to_expose
+        self.password = password
         self.with_exposed_ports(self.port_to_expose)
+        if self.password:
+            self.with_command(f"redis-server --requirepass {self.password}")
 
     @wait_container_is_ready(redis.exceptions.ConnectionError)
     def _connect(self):
@@ -44,7 +46,8 @@ def get_client(self, **kwargs):
         """
         return redis.Redis(
             host=self.get_container_host_ip(),
-            port=self.get_exposed_port(6379),
+            port=self.get_exposed_port(self.port_to_expose),
+            password=self.password,
             **kwargs,
         )
 

tests/test_redis.py

@@ -15,6 +15,14 @@ def test_docker_run_redis():
         assert b'new_msg', msg['data']
 
 
+def test_docker_run_redis_with_password():
+    config = RedisContainer(password="mypass")
+    with config as redis:
+        client = redis.get_client(decode_responses=True)
+        client.set("hello", "world")
+        assert client.get("hello") == "world"
+
+
 def wait_for_message(pubsub, timeout=1, ignore_subscribe_messages=True):
     now = time.time()
     timeout = now + timeout