Fix command not recognized if it is directly followed by a newline (python-telegram-bot#869)

jh0ker · tsnoam · commit ec9b16ac7b55 · 2017-10-14T21:04:02.000+03:00
fixes python-telegram-bot#868
diff --git a/telegram/ext/commandhandler.py b/telegram/ext/commandhandler.py
@@ -134,7 +134,7 @@ def check_update(self, update):
             message = update.message or update.edited_message
 
             if message.text:
-                command = message.text[1:].split(' ')[0].split('@')
+                command = message.text[1:].split(None, 1)[0].split('@')
                 command.append(
                     message.bot.username)  # in case the command was send without a username
 
diff --git a/tests/test_commandhandler.py b/tests/test_commandhandler.py
@@ -165,6 +165,15 @@ def test_pass_args(self, dp, message):
         dp.process_update(Update(0, message=message))
         assert self.test_flag
 
+    def test_newline(self, dp, message):
+        handler = CommandHandler('test', self.callback_basic)
+        dp.add_handler(handler)
+
+        message.text = '/test\nfoobar'
+        assert handler.check_update(Update(0, message))
+        dp.process_update(Update(0, message))
+        assert self.test_flag
+
     def test_pass_user_or_chat_data(self, dp, message):
         handler = CommandHandler('test', self.callback_data_1, pass_user_data=True)
         dp.add_handler(handler)
