Ease unitesting with mock

[python-musicpd.git] / musicpd.py

diff --git a/musicpd.py b/musicpd.py
index 864337e711baeb4ffa1cf05841df5c2acd87d3c0..6bb050990c900c6585bcb3c61ade7aa89eb3d0a2 100644 (file)
--- a/musicpd.py
+++ b/musicpd.py
@@ -440,7 +440,7 @@ class MPDClient(object):
             self._rfile.close()
         if hasattr(self._wfile, 'close'):
             self._wfile.close()
-        if isinstance(self._sock, socket.socket):
+        if hasattr(self._sock, 'close'):
             self._sock.close()
         self._reset()