Added the SSL option for RedisSettings dataclass (#165)

* Added the SSL option for RedisSettings dataclass

* Updated HISTORY.rst

* Fixed linter errors

Author: Relrin

HISTORY.rst

@@ -6,6 +6,7 @@ History
 v0.18.4 (unreleased)
 ....................
 * Add ``py.typed`` file to tell mypy the package has type hints, #163
+* Added the SSL option for RedisSettings dataclass
 
 v0.18.3 (2019-11-13)
 ....................

arq/connections.py

@@ -4,6 +4,7 @@
 from dataclasses import dataclass
 from datetime import datetime, timedelta
 from operator import attrgetter
+from ssl import SSLContext
 from typing import Any, List, Optional, Tuple, Union
 from uuid import uuid4
 
@@ -29,6 +30,7 @@ class RedisSettings:
     port: int = 6379
     database: int = 0
     password: str = None
+    ssl: [bool, None, SSLContext] = None
     conn_timeout: int = 1
     conn_retries: int = 5
     conn_retry_delay: int = 1
@@ -183,16 +185,17 @@ async def create_pool(
         addr = settings.host
 
         async def pool_factory(*args, **kwargs):
-            client = await aioredis.sentinel.create_sentinel_pool(*args, **kwargs)
+            client = await aioredis.sentinel.create_sentinel_pool(*args, ssl=settings.ssl, **kwargs)
             return client.master_for(settings.sentinel_master)
 
     else:
-        pool_factory = functools.partial(aioredis.create_pool, create_connection_timeout=settings.conn_timeout)
+        pool_factory = functools.partial(
+            aioredis.create_pool, create_connection_timeout=settings.conn_timeout, ssl=settings.ssl
+        )
         addr = settings.host, settings.port
 
     try:
         pool = await pool_factory(addr, db=settings.database, password=settings.password, encoding='utf8')
-
         pool = ArqRedis(pool, job_serializer=job_serializer, job_deserializer=job_deserializer)
 
     except (ConnectionError, OSError, aioredis.RedisError, asyncio.TimeoutError) as e:

tests/test_utils.py

@@ -12,7 +12,7 @@ def test_settings_changed():
     settings = RedisSettings(port=123)
     assert settings.port == 123
     assert (
-        '<RedisSettings host=localhost port=123 database=0 password=None conn_timeout=1 conn_retries=5 '
+        '<RedisSettings host=localhost port=123 database=0 password=None ssl=None conn_timeout=1 conn_retries=5 '
         'conn_retry_delay=1 sentinel=False sentinel_master=mymaster>'
     ) == str(settings)