summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJoe Gordon <jogo@pinterest.com>2022-01-21 09:34:13 -0800
committerJon Parise <jon@pinterest.com>2022-02-20 12:10:20 -0800
commit4048ab1a5f0b8839f59bc4c3a4327639c3677adb (patch)
treedafcad316c5194b17dc0e86a45ffdbc270db13ef
parent828394ffabfdeadbae96cc167d3953e71a050a2a (diff)
downloadpymemcache-4048ab1a5f0b8839f59bc4c3a4327639c3677adb.tar.gz
Support noreply for hash flush_all
The regular client supports noreply so pass args, kwargs like for other functions in hash client Fixes issue #360
-rw-r--r--pymemcache/client/hash.py4
-rw-r--r--pymemcache/test/test_client_hash.py4
2 files changed, 6 insertions, 2 deletions
diff --git a/pymemcache/client/hash.py b/pymemcache/client/hash.py
index 72cb9d8..11c4246 100644
--- a/pymemcache/client/hash.py
+++ b/pymemcache/client/hash.py
@@ -457,9 +457,9 @@ class HashClient(object):
def touch(self, key, *args, **kwargs):
return self._run_cmd('touch', key, False, *args, **kwargs)
- def flush_all(self):
+ def flush_all(self, *args, **kwargs):
for client in self.clients.values():
- self._safely_run_func(client, client.flush_all, False)
+ self._safely_run_func(client, client.flush_all, False, *args, **kwargs)
def quit(self):
for client in self.clients.values():
diff --git a/pymemcache/test/test_client_hash.py b/pymemcache/test/test_client_hash.py
index aecc1bd..541e003 100644
--- a/pymemcache/test/test_client_hash.py
+++ b/pymemcache/test/test_client_hash.py
@@ -337,6 +337,10 @@ class TestHashClient(ClientTestMixin, unittest.TestCase):
result = client.set_many(values, noreply=True)
assert result == []
+ def test_noreply_flush(self):
+ client = self.make_client()
+ client.flush_all(noreply=True)
+
def test_set_many_unix(self):
values = {
'key1': 'value1',