Revert "[AIRFLOW-3353] Upgrade redis client (apache#4203)" (apache#4205)

This reverts commit ae62987.

Author: ashb

airflow/contrib/hooks/redis_hook.py

@@ -20,7 +20,7 @@
 """
 RedisHook module
 """
-from redis import Redis
+from redis import StrictRedis
 
 from airflow.exceptions import AirflowException
 from airflow.hooks.base_hook import BaseHook
@@ -69,7 +69,7 @@ def get_conn(self):
                 self.redis_conn_id, self.host, self.port, self.db
             )
             try:
-                self.client = Redis(
+                self.client = StrictRedis(
                     host=self.host,
                     port=self.port,
                     password=self.password,

setup.py

@@ -218,7 +218,7 @@ def write_version(filename=os.path.join(*['airflow',
 postgres = ['psycopg2-binary>=2.7.4']
 qds = ['qds-sdk>=1.9.6']
 rabbitmq = ['librabbitmq>=1.6.1']
-redis = ['redis>=3.0.0,<4.0.0']
+redis = ['redis>=2.10.5,<3.0.0']
 s3 = ['boto3>=1.7.0, <1.8.0']
 salesforce = ['simple-salesforce>=0.72']
 samba = ['pysmbclient>=0.1.3']

tests/contrib/hooks/test_redis_hook.py

@@ -35,7 +35,7 @@ def test_get_conn(self):
         self.assertEqual(
             repr(hook.get_conn()),
             (
-                'Redis<ConnectionPool'
+                'StrictRedis<ConnectionPool'
                 '<Connection<host=localhost,port=6379,db=0>>>'
             )
         )

tests/contrib/sensors/test_redis_sensor.py

@@ -55,7 +55,7 @@ def test_poke(self, key_exists):
         key_exists.return_value = False
         self.assertFalse(self.sensor.poke(None))
 
-    @patch("airflow.contrib.hooks.redis_hook.Redis.exists")
+    @patch("airflow.contrib.hooks.redis_hook.StrictRedis.exists")
     def test_existing_key_called(self, redis_client_exists):
         self.sensor.run(
             start_date=DEFAULT_DATE,