diff options
| author | Nick Pope <nick.pope@flightdataservices.com> | 2020-08-17 23:02:58 +0100 |
|---|---|---|
| committer | Nick Pope <nick.pope@flightdataservices.com> | 2020-08-17 23:35:40 +0100 |
| commit | f3bc1e07161fd451c4aa94790e3db7bf1b71a89e (patch) | |
| tree | 40e253ccc1ea78ae45c7740df8d0b5717fa5c822 /pymemcache/test | |
| parent | a5e0dcbe99a4bbb9706531ac66d8131d89d7cb88 (diff) | |
| download | pymemcache-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.py | 14 |
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( |
