fix test

Author: opapy

tests/test_backend.py

@@ -18,7 +18,10 @@
 @patch('django.conf.settings', global_settings)
 @patch('django_elastipymemcache.memcached.get_cluster_info')
 def test_split_servers(get_cluster_info):
-    from django_elastipymemcache.memcached import ElastiPyMemCache
+    from django_elastipymemcache.memcached import (
+        ElastiPyMemCache,
+        deserialize_pickle,
+    )
     backend = ElastiPyMemCache('h:0', {})
     servers = [('h1', 0), ('h2', 0)]
     get_cluster_info.return_value = {
@@ -28,13 +31,20 @@ def test_split_servers(get_cluster_info):
     assert backend._cache
     get_cluster_info.assert_called_once_with(
             'h', '0', socket._GLOBAL_DEFAULT_TIMEOUT)
-    backend._lib.Client.assert_called_once_with(servers, ignore_exc=True)
+    backend._lib.Client.assert_called_once_with(
+        servers,
+        deserializer=deserialize_pickle,
+        ignore_exc=True
+    )
 
 
 @patch('django.conf.settings', global_settings)
 @patch('django_elastipymemcache.memcached.get_cluster_info')
 def test_node_info_cache(get_cluster_info):
-    from django_elastipymemcache.memcached import ElastiPyMemCache
+    from django_elastipymemcache.memcached import (
+        ElastiPyMemCache,
+        deserialize_pickle,
+    )
     servers = [('h1', 0), ('h2', 0)]
     get_cluster_info.return_value = {
         'nodes': servers
@@ -46,7 +56,11 @@ def test_node_info_cache(get_cluster_info):
     backend.get('key1')
     backend.set('key2', 'val')
     backend.get('key2')
-    backend._lib.Client.assert_called_once_with(servers, ignore_exc=True)
+    backend._lib.Client.assert_called_once_with(
+        servers,
+        deserializer=deserialize_pickle,
+        ignore_exc=True
+    )
     eq_(backend._cache.get.call_count, 2)
     eq_(backend._cache.set.call_count, 2)