Skip to content

Commit 45158a1

Browse files
author
opapy
committed
fix test
1 parent 08e38f6 commit 45158a1

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

tests/test_backend.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)