Skip to content

Commit f12c7bb

Browse files
committed
Fix nested with statement to flatten
1 parent e966ddd commit f12c7bb

File tree

1 file changed

+21
-7
lines changed

1 file changed

+21
-7
lines changed

tests/test_backend.py

Lines changed: 21 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -121,14 +121,14 @@ def test_client_get_many(get_cluster_info):
121121
ret = backend.get_many(['key2'])
122122
eq_(ret, {})
123123

124-
with patch('django_elastipymemcache.client.Client.get_many'):
125-
with patch('pymemcache.client.hash.HashClient._safely_run_func') as p2:
126-
p2.return_value = {
127-
':1:key3': 1509111630.048594
128-
}
124+
with patch('django_elastipymemcache.client.Client.get_many'), \
125+
patch('pymemcache.client.hash.HashClient._safely_run_func') as p2:
126+
p2.return_value = {
127+
':1:key3': 1509111630.048594
128+
}
129129

130-
ret = backend.get_many(['key3'])
131-
eq_(ret, {'key3': 1509111630.048594})
130+
ret = backend.get_many(['key3'])
131+
eq_(ret, {'key3': 1509111630.048594})
132132

133133
# If False value is included, ignore it.
134134
with patch('pymemcache.client.hash.HashClient.get_many') as p:
@@ -145,3 +145,17 @@ def test_client_get_many(get_cluster_info):
145145
'key3': 1509111630.058594
146146
},
147147
)
148+
149+
with patch('pymemcache.client.hash.HashClient.get_many') as p:
150+
p.return_value = {
151+
':1:key1': None,
152+
':1:key2': 1509111630.048594,
153+
':1:key3': False,
154+
}
155+
ret = backend.get_many(['key1', 'key2', 'key3'])
156+
eq_(
157+
ret,
158+
{
159+
'key2': 1509111630.048594,
160+
},
161+
)

0 commit comments

Comments
 (0)