Raises an exception if command contains '\n'

[python-musicpd.git] / test.py

diff --git a/test.py b/test.py
index 7f1890fe9f325ec1b4c5c450ea8048d8b5727cca..840bd4730dd03fd76c18677ba1fa876842e20a12 100755 (executable)
--- a/test.py
+++ b/test.py
@@ -582,5 +582,18 @@ class testConnection(unittest.TestCase):
             sock.connect.assert_called_with('/run/mpd/socket')
 
 
+class testException(unittest.TestCase):
+
+    def test_CommandError_on_newline(self):
+        os.environ['MPD_HOST'] = '/run/mpd/socket'
+        with mock.patch('musicpd.socket') as socket_mock:
+            sock = mock.MagicMock(name='socket')
+            socket_mock.socket.return_value = sock
+            cli = musicpd.MPDClient()
+            cli.connect()
+            with self.assertRaises(musicpd.CommandError):
+                cli.find('(album == "foo\nbar")')
+
+
 if __name__ == '__main__':
     unittest.main()