1 files changed, 10 insertions, 6 deletions
diff --git a/tests/test_connection.py b/tests/test_connection.py
index 0b64d2b..5ac76d6 100644
--- a/tests/test_connection.py
+++ b/tests/test_connection.py
@@ -1,4 +1,4 @@
-from redis import ConnectionPool, Redis, UnixDomainSocketConnection
+from redis import ConnectionPool, Redis, SSLConnection, UnixDomainSocketConnection
 
 from rq import Connection, Queue
 from rq.connections import parse_connection
@@ -38,10 +38,14 @@ class TestConnectionInheritance(RQTestCase):
         self.assertEqual(q2.connection, job2.connection)
 
     def test_parse_connection(self):
-        """Test parsing `ssl` and UnixDomainSocketConnection"""
-        _, _, kwargs = parse_connection(Redis(ssl=True))
-        self.assertTrue(kwargs['ssl'])
+        """Test parsing the connection"""
+        conn_class, pool_class, pool_kwargs = parse_connection(Redis(ssl=True))
+        self.assertEqual(conn_class, Redis)
+        self.assertEqual(pool_class, SSLConnection)
+
         path = '/tmp/redis.sock'
         pool = ConnectionPool(connection_class=UnixDomainSocketConnection, path=path)
-        _, _, kwargs = parse_connection(Redis(connection_pool=pool))
-        self.assertTrue(kwargs['unix_socket_path'], path)
+        conn_class, pool_class, pool_kwargs = parse_connection(Redis(connection_pool=pool))
+        self.assertEqual(conn_class, Redis)
+        self.assertEqual(pool_class, UnixDomainSocketConnection)
+        self.assertEqual(pool_kwargs, {"path": path})