@@ -21,6 +21,7 @@ def test_split_servers(get_cluster_info):
2121 from django_elastipymemcache .memcached import (
2222 ElastiPyMemCache ,
2323 deserialize_pickle ,
24+ serialize_pickle ,
2425 )
2526 backend = ElastiPyMemCache ('h:0' , {})
2627 servers = [('h1' , 0 ), ('h2' , 0 )]
@@ -34,7 +35,8 @@ def test_split_servers(get_cluster_info):
3435 backend ._lib .Client .assert_called_once_with (
3536 servers ,
3637 deserializer = deserialize_pickle ,
37- ignore_exc = True
38+ ignore_exc = True ,
39+ serializer = serialize_pickle
3840 )
3941
4042
@@ -44,6 +46,7 @@ def test_node_info_cache(get_cluster_info):
4446 from django_elastipymemcache .memcached import (
4547 ElastiPyMemCache ,
4648 deserialize_pickle ,
49+ serialize_pickle ,
4750 )
4851 servers = [('h1' , 0 ), ('h2' , 0 )]
4952 get_cluster_info .return_value = {
@@ -59,7 +62,8 @@ def test_node_info_cache(get_cluster_info):
5962 backend ._lib .Client .assert_called_once_with (
6063 servers ,
6164 deserializer = deserialize_pickle ,
62- ignore_exc = True
65+ ignore_exc = True ,
66+ serializer = serialize_pickle
6367 )
6468 eq_ (backend ._cache .get .call_count , 2 )
6569 eq_ (backend ._cache .set .call_count , 2 )
0 commit comments