Merge pull request #103 from singingwolfboy/from_url

Add support for `from_url` constructors

Author: srikalyan

mockredis/client.py

@@ -58,6 +58,10 @@ def __init__(self,
         # Dictionary from script to sha ''Script''
         self.shas = dict()
 
+    @classmethod
+    def from_url(cls, url, db=None, **kwargs):
+        return cls(**kwargs)
+
     # Connection Functions #
 
     def echo(self, msg):
@@ -1509,6 +1513,8 @@ def mock_redis_client(**kwargs):
     """
     return MockRedis()
 
+mock_redis_client.from_url = mock_redis_client
+
 
 def mock_strict_redis_client(**kwargs):
     """
@@ -1517,3 +1523,5 @@ def mock_strict_redis_client(**kwargs):
     instead of a StrictRedis object.
     """
     return MockRedis(strict=True)
+
+mock_strict_redis_client.from_url = mock_strict_redis_client

mockredis/tests/test_factories.py

@@ -13,8 +13,22 @@ def test_mock_redis_client():
     ok_(not mock_redis_client(host="localhost", port=6379).strict)
 
 
+def test_mock_redis_client_from_url():
+    """
+    Test that we can pass kwargs to the Redis from_url mock/patch target.
+    """
+    ok_(not mock_redis_client.from_url(host="localhost", port=6379).strict)
+
+
 def test_mock_strict_redis_client():
     """
     Test that we can pass kwargs to the StrictRedis mock/patch target.
     """
     ok_(mock_strict_redis_client(host="localhost", port=6379).strict)
+
+
+def test_mock_strict_redis_client_from_url():
+    """
+    Test that we can pass kwargs to the StrictRedis from_url mock/patch target.
+    """
+    ok_(mock_strict_redis_client.from_url(host="localhost", port=6379).strict)