test_bot.py: unit test setWebhook and deleteWebhook

jh0ker · jh0ker · commit 8ac66698b582 · 2017-01-16T00:01:32.000+01:00
diff --git a/tests/test_bot.py b/tests/test_bot.py
@@ -291,7 +291,7 @@ def testGetChatMember(self):
 
     @flaky(3, 1)
     @timeout(10)
-    def test_forward_channel_messgae(self):
+    def test_forward_channel_message(self):
         text = 'test forward message'
         msg = self._bot.sendMessage(self._channel_id, text)
         self.assertEqual(text, msg.text)
@@ -301,12 +301,25 @@ def test_forward_channel_messgae(self):
 
     @flaky(3, 1)
     @timeout(10)
-    def test_get_webhook_info(self):
+    def test_set_webhook_get_webhook_info(self):
         url = 'https://python-telegram-bot.org/test/webhook'
-        self._bot.set_webhook(url)
+        max_connections = 7
+        allowed_updates = ['message']
+        self._bot.set_webhook(url, max_connections=7, allowed_updates=['message'])
         info = self._bot.getWebhookInfo()
-        self._bot.set_webhook('')
+        self._bot.delete_webhook()
         self.assertEqual(url, info.url)
+        self.assertEqual(max_connections, info.max_connections)
+        self.assertListEqual(allowed_updates, info.allowed_updates)
+
+    @flaky(3, 1)
+    @timeout(10)
+    def test_delete_webhook(self):
+        url = 'https://python-telegram-bot.org/test/webhook'
+        self._bot.set_webhook(url)
+        self._bot.delete_webhook()
+        info = self._bot.getWebhookInfo()
+        self.assertEqual(info.url, '')
 
     @flaky(3, 1)
     @timeout(10)
