Adding test to check if thumbs are PhotoSize instances

python-telegram-bot · Sep 4, 2015 · 79cdb6c · 79cdb6c
1 parent 49ff02d
commit 79cdb6c
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/test_bot.py b/tests/test_bot.py
@@ -185,6 +185,7 @@ def testSendDocument(self):
 
         self.assertTrue(self.is_json(message.to_json()))
         self.assertEqual(message.document.file_size, 12948)
+        self.assertTrue(isinstance(message.document.thumb, telegram.PhotoSize))
 
     def testSendGIFURLDocument(self):
         '''Test the telegram.Bot sendDocument method'''
@@ -203,6 +204,7 @@ def testResendDocument(self):
 
         self.assertTrue(self.is_json(message.to_json()))
         self.assertEqual(message.document.file_id, 'BQADAQADHAADNTwtBxZxUGKyxYbYAg')
+        self.assertTrue(isinstance(message.document.thumb, telegram.PhotoSize))
 
     def testSendVideo(self):
         '''Test the telegram.Bot sendVideo method'''
@@ -214,6 +216,8 @@ def testSendVideo(self):
         self.assertTrue(self.is_json(message.to_json()))
         self.assertEqual(message.video.file_size, 326534)
         self.assertEqual(message.caption, 'testSendVideo')
+        if message.video.thumb:
+            self.assertTrue(isinstance(message.video.thumb, telegram.PhotoSize))
 
     def testResendVideo(self):
         '''Test the telegram.Bot sendVideo method'''
@@ -223,6 +227,7 @@ def testResendVideo(self):
 
         self.assertTrue(self.is_json(message.to_json()))
         self.assertEqual(message.video.duration, 4)
+        self.assertTrue(isinstance(message.video.thumb, telegram.PhotoSize))
 
     def testResendSticker(self):
         '''Test the telegram.Bot sendSticker method'''
@@ -232,6 +237,7 @@ def testResendSticker(self):
 
         self.assertTrue(self.is_json(message.to_json()))
         self.assertEqual(message.sticker.file_size, 39518)
+        self.assertTrue(isinstance(message.sticker.thumb, telegram.PhotoSize))
 
     def testSendLocation(self):
         '''Test the telegram.Bot sendLocation method'''