Tests run in python3

Author: CM Lubinski

django_elasticache/cluster_utils.py

@@ -41,7 +41,7 @@ def get_cluster_info(host, port):
     client.write(cmd)
     res = client.read_until('\n\r\nEND\r\n')
     client.close()
-    ls = filter(None, re.compile(r'\r?\n').split(res))
+    ls = list(filter(None, re.compile(r'\r?\n').split(res)))
     if len(ls) != 4:
         raise WrongProtocolData(cmd, res)
 

tests/test_backend.py

@@ -1,6 +1,11 @@
 from django.conf import global_settings, settings
-from mock import patch, Mock
 from nose.tools import eq_, raises
+import sys
+if sys.version < '3':
+    from mock import patch, Mock
+else:
+    from unittest.mock import patch, Mock
+
 
 # Initialize django 1.7
 settings.configure()

tests/test_protocol.py

@@ -1,7 +1,12 @@
-from mock import patch, call, MagicMock
 from django_elasticache.cluster_utils import (
     get_cluster_info, WrongProtocolData)
 from nose.tools import eq_, raises
+import sys
+if sys.version < '3':
+    from mock import patch, call, MagicMock
+else:
+    from unittest.mock import patch, call, MagicMock
+
 
 TEST_PROTOCOL_1 = [
     'VERSION 1.4.14',