summaryrefslogtreecommitdiff
path: root/pymemcache/test
diff options
context:
space:
mode:
authorNick Pope <nick.pope@flightdataservices.com>2020-08-17 23:02:58 +0100
committerNick Pope <nick.pope@flightdataservices.com>2020-08-17 23:35:40 +0100
commitf3bc1e07161fd451c4aa94790e3db7bf1b71a89e (patch)
tree40e253ccc1ea78ae45c7740df8d0b5717fa5c822 /pymemcache/test
parenta5e0dcbe99a4bbb9706531ac66d8131d89d7cb88 (diff)
downloadpymemcache-f3bc1e07161fd451c4aa94790e3db7bf1b71a89e.tar.gz
Added missing HashClient.close() and HashClient.quit().
Also added disconnect_all() as an alias for close().
Diffstat (limited to 'pymemcache/test')
-rw-r--r--pymemcache/test/test_client_hash.py14
1 files changed, 14 insertions, 0 deletions
diff --git a/pymemcache/test/test_client_hash.py b/pymemcache/test/test_client_hash.py
index 814661d..c5f3c4b 100644
--- a/pymemcache/test/test_client_hash.py
+++ b/pymemcache/test/test_client_hash.py
@@ -159,6 +159,20 @@ class TestHashClient(ClientTestMixin, unittest.TestCase):
result = client.touch(b'key', 1, noreply=False)
assert result is True
+ def test_close(self):
+ client = self.make_client([])
+ assert all(c.sock is not None for c in client.clients.values())
+ result = client.close()
+ assert result is None
+ assert all(c.sock is None for c in client.clients.values())
+
+ def test_quit(self):
+ client = self.make_client([])
+ assert all(c.sock is not None for c in client.clients.values())
+ result = client.quit()
+ assert result is None
+ assert all(c.sock is None for c in client.clients.values())
+
def test_no_servers_left(self):
from pymemcache.client.hash import HashClient
client = HashClient(