Fix crash on Python 3 in touch() logging

1 files changed, 13 insertions, 1 deletions

diff --git a/tests/test_memcache.py b/tests/test_memcache.py
index 6408e81..58024cd 100644
--- a/tests/test_memcache.py
+++ b/tests/test_memcache.py
@@ -2,9 +2,10 @@ from __future__ import print_function
 
 import unittest
 
+import mock
 import six
 
-from memcache import Client, SERVER_MAX_KEY_LENGTH, SERVER_MAX_VALUE_LENGTH  # noqa: H301
+from memcache import Client, _Host, SERVER_MAX_KEY_LENGTH, SERVER_MAX_VALUE_LENGTH  # noqa: H301
 from .utils import captured_stderr
 
 
@@ -183,6 +184,17 @@ class TestMemcache(unittest.TestCase):
             "'NOT_FOUND'\n"
         )
 
+    @mock.patch.object(_Host, 'readline')
+    def test_touch_unexpected_reply(self, mock_readline):
+        """touch() logs an error upon receiving an unexpected reply."""
+        mock_readline.return_value = 'SET'  # the unexpected reply
+        with captured_stderr() as output:
+            self.mc.touch('key')
+        self.assertEqual(
+            output.getvalue(),
+            "MemCached: touch expected %s, got: 'SET'\n" % b'TOUCHED'
+        )
+
 
 if __name__ == '__main__':
     unittest.main()