Slightly increasing test coverage to massage coverall

Author: Frederic Massart

telegrambot/test/testcases.py

@@ -84,7 +84,7 @@ def _test_message_ok(self, action, update=None, number=1):
             self.assertBotResponse(mock_send, action)
             self.assertEqual(number, Update.objects.count())
             self.assertUpdate(Update.objects.get(update_id=update.update_id), update)
-            
+
     def _test_message_no_handler(self, action, update=None, number=1):
         if not update:
             update = self.update
@@ -96,4 +96,13 @@ def _test_message_no_handler(self, action, update=None, number=1):
             self.assertEqual(response.status_code, status.HTTP_200_OK)
             self.assertEqual(0, mock_send.call_count)
             self.assertEqual(number, Update.objects.count())
-            self.assertUpdate(Update.objects.get(update_id=update.update_id), update)            
+            self.assertUpdate(Update.objects.get(update_id=update.update_id), update)            
+
+    def _test_no_response(self, action, update=None):
+        if not update:
+            update = self.update
+        with mock.patch("telegram.bot.Bot.sendMessage", callable=mock.MagicMock()) as mock_send:
+            update.message.text = action['in']
+            response = self.client.post(self.webhook_url, update.to_json(), **self.kwargs)
+            self.assertEqual(response.status_code, status.HTTP_200_OK)
+            self.assertEqual(0, mock_send.call_count)

tests/bot_handlers.py

@@ -1,5 +1,5 @@
 from tests.commands_views import StartView, AuthorCommandView, AuthorInverseListView, AuthorCommandQueryView, \
-    UnknownView, AuthorName, MessageView
+    UnknownView, AuthorName, MessageView, MissingTemplateView
 from telegrambot.handlers import command, unknown_command, regex, message 
 from telegrambot.bot_views.decorators import login_required
 from telegrambot.bot_views.generic import SendMessageCommandView, EchoCommandView, HelloWorldCommandView
@@ -13,6 +13,7 @@
     command('author', AuthorCommandView.as_command_view()), 
     command('hello', HelloWorldCommandView.as_command_view()),
     command('how_are_you', SendMessageCommandView.as_command_view(message='Good, thanks!')),
+    command('missing', MissingTemplateView.as_command_view()),
     regex(r'^Echo', EchoCommandView.as_command_view()),
     unknown_command(UnknownView.as_command_view()),
     message(MessageView.as_command_view())

tests/commands_views.py

@@ -48,3 +48,6 @@ class AuthorName(DetailCommandView):
 
     def get_slug(self, **kwargs):
         return kwargs.get('name', None)
+
+class MissingTemplateView(TemplateCommandView):
+    template_text = "i/dont/exist.txt"

tests/test_telegrambot.py

@@ -128,6 +128,10 @@ class TestBotCommands(testcases.BaseTestBot):
                        'text': 'Good, thanks!'}
                }
 
+    template_missing = {'in': '/missing',
+                        'out': None
+                       }
+
     def test_start(self):
         self._test_message_ok(self.start)
 
@@ -177,6 +181,9 @@ def test_hello_world(self):
     def test_message(self):
         self._test_message_ok(self.message)
 
+    def test_missing_template(self):
+        self._test_no_response(self.template_missing)
+
 
 class TestBotMessage(testcases.BaseTestBot):