fixed broken tests

Signed-off-by: ahmedsobeh <[email protected]>
valkey-io · Jul 3, 2024 · f7c3a93 · f7c3a93
1 parent 4480492
commit f7c3a93
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 4 deletions.
diff --git a/tests/test_asyncio/test_connection_pool.py b/tests/test_asyncio/test_connection_pool.py
@@ -442,7 +442,8 @@ def test_invalid_scheme_raises_error(self):
         with pytest.raises(ValueError) as cm:
             valkey.ConnectionPool.from_url("localhost")
         assert str(cm.value) == (
-            "Valkey URL must specify one of the following schemes ['valkey', 'valkeys', 'redis', 'rediss']"
+            "Valkey URL must specify one of the following schemes"
+            " ['valkey', 'valkeys', 'redis', 'rediss']"
         )
 
 

diff --git a/tests/test_connection_pool.py b/tests/test_connection_pool.py
@@ -336,14 +336,16 @@ def test_invalid_scheme_raises_error(self):
         with pytest.raises(ValueError) as cm:
             valkey.ConnectionPool.from_url("localhost")
         assert str(cm.value) == (
-            "Valkey URL must specify one of the following schemes ['valkey', 'valkeys', 'redis', 'rediss']"
+            "Valkey URL must specify one of the following schemes "
+            "['valkey', 'valkeys', 'redis', 'rediss']"
         )
 
     def test_invalid_scheme_raises_error_when_double_slash_missing(self):
         with pytest.raises(ValueError) as cm:
             valkey.ConnectionPool.from_url("valkey:foo.bar.com:12345")
         assert str(cm.value) == (
-            "Valkey URL must specify one of the following schemes ['valkey', 'valkeys', 'redis', 'rediss']"
+            "Valkey URL must specify one of the following schemes "
+            "['valkey', 'valkeys', 'redis', 'rediss']"
         )
 
 

diff --git a/valkey/_parsers/url_parser.py b/valkey/_parsers/url_parser.py
@@ -82,7 +82,7 @@ def parse_url(url: str, async_connection: bool):
             except (AttributeError, ValueError):
                 pass
 
-        if parsed.scheme in supported_schemes:
+        if parsed.scheme in ("valkeys", "rediss"):
             kwargs["connection_class"] = (
                 SSLConnectionAsync if async_connection else SSLConnection
             )