From 4048ab1a5f0b8839f59bc4c3a4327639c3677adb Mon Sep 17 00:00:00 2001 From: Joe Gordon Date: Fri, 21 Jan 2022 09:34:13 -0800 Subject: 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 --- pymemcache/client/hash.py | 4 ++-- pymemcache/test/test_client_hash.py | 4 ++++ 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', -- cgit v1.2.1