Adding test for quit_all

2 files changed, 5 insertions, 3 deletions

diff --git a/memcache.py b/memcache.py
index 1eefcf7..bbc2149 100644
--- a/memcache.py
+++ b/memcache.py
@@ -364,7 +364,7 @@ class Client(threading.local):
                     serverData[slab[0]][slab[1]] = item[2]
         return data
 
-    def quit_all(self):
+    def quit_all(self) -> None:
         '''Send a "quit" command to all servers and wait for the connection to close.'''
         for s in self.servers:
             s.quit()
@@ -1480,10 +1480,9 @@ class _Host(object):
         self.buffer = buf[rlen:]
         return buf[:rlen]
 
-    def quit(self):
+    def quit(self) -> None:
         '''Send a "quit" command to remote server and wait for connection to close.'''
         if self.socket:
-            # Using self.send_cmd, so no need for '\r\n'.
             self.send_cmd('quit')
 
             # We can't close the local socket until the remote end processes the quit
diff --git a/tests/test_memcache.py b/tests/test_memcache.py
index e87dd09..3593e03 100644
--- a/tests/test_memcache.py
+++ b/tests/test_memcache.py
@@ -48,6 +48,9 @@ class TestMemcache(unittest.TestCase):
         self.check_setget("an_integer", 42)
         self.check_setget("an_integer_2", 42, noreply=True)
 
+    def test_quit_all(self):
+        self.mc.quit_all()
+
     def test_delete(self):
         self.check_setget("long", int(1 << 30))
         result = self.mc.delete("long")