(Attempt to) fix failing Travis CI tests.

tests/test_telebot.py

@@ -1,21 +1,28 @@
 # -*- coding: utf-8 -*-
 import sys
 import time
-
+import pytest
-sys.path.append('../')
-from telebot import types
-from telebot import apihelper
-import telebot
 import os
 
+import telebot
+from telebot import types
+from telebot import apihelper
+
+sys.path.append('../')
+
+should_skip = 'TOKEN' and 'CHAT_ID' not in os.environ
+
+if not should_skip:
     TOKEN = os.environ['TOKEN']
     CHAT_ID = os.environ['CHAT_ID']
 
+@pytest.mark.skipif(should_skip, reason="No environment variables configured")
+class TestTeleBot:
 
-def test_message_listener():
+    def test_message_listener(self):
         msg_list = []
         for x in range(100):
-        msg_list.append(create_text_message('Message ' + str(x)))
+            msg_list.append(self.create_text_message('Message ' + str(x)))
 
         def listener(messages):
             assert len(messages) == 100
@@ -23,10 +30,9 @@ def test_message_listener():
         tb = telebot.TeleBot('')
         tb.set_update_listener(listener)
 
-
+    def test_message_handler(self):
-def test_message_handler():
         tb = telebot.TeleBot('')
-    msg = create_text_message('/help')
+        msg = self.create_text_message('/help')
 
         @tb.message_handler(commands=['help', 'start'])
         def command_handler(message):
@@ -36,10 +42,9 @@ def test_message_handler():
         time.sleep(1)
         assert msg.text == 'got'
 
-
+    def test_message_handler_reg(self):
-def test_message_handler_reg():
         bot = telebot.TeleBot('')
-    msg = create_text_message(r'https://web.telegram.org/')
+        msg = self.create_text_message(r'https://web.telegram.org/')
 
         @bot.message_handler(regexp='((https?):((//)|(\\\\))+([\w\d:#@%/;$()~_?\+-=\\\.&](#!)?)*)')
         def command_url(message):
@@ -49,10 +54,9 @@ def test_message_handler_reg():
         time.sleep(1)
         assert msg.text == 'got'
 
-
+    def test_message_handler_reg_fail(self):
-def test_message_handler_reg_fail():
         bot = telebot.TeleBot('')
-    msg = create_text_message(r'web.telegram.org/')
+        msg = self.create_text_message(r'web.telegram.org/')
 
         @bot.message_handler(regexp='((https?):((//)|(\\\\))+([\w\d:#@%/;$()~_?\+-=\\\.&](#!)?)*)')
         def command_url(message):
@@ -62,60 +66,52 @@ def test_message_handler_reg_fail():
         time.sleep(1)
         assert not msg.text == 'got'
 
-
+    def test_send_file_by_id(self):
-def test_send_file_by_id():
         file_id = 'BQADBQADjAIAAsYifgbvqwq1he9REAI'
         tb = telebot.TeleBot(TOKEN)
         ret_msg = tb.send_document(CHAT_ID, file_id)
         assert ret_msg.message_id
 
-
+    def test_send_file(self):
-def test_send_file():
         file_data = open('../examples/detailed_example/kitten.jpg', 'rb')
         tb = telebot.TeleBot(TOKEN)
         ret_msg = tb.send_document(CHAT_ID, file_data)
         assert ret_msg.message_id
 
-
+    def test_send_video(self):
-def test_send_video():
         file_data = open('./test_data/test_video.mp4', 'rb')
         tb = telebot.TeleBot(TOKEN)
         ret_msg = tb.send_video(CHAT_ID, file_data)
         assert ret_msg.message_id
 
-
+    def test_send_video_more_params(self):
-def test_send_video_more_params():
         file_data = open('./test_data/test_video.mp4', 'rb')
         tb = telebot.TeleBot(TOKEN)
         ret_msg = tb.send_video(CHAT_ID, file_data, 1)
         assert ret_msg.message_id
 
-
+    def test_send_file_exception(self):
-def test_send_file_exception():
         tb = telebot.TeleBot(TOKEN)
         try:
-        ret_msg = tb.send_document(CHAT_ID, None)
+            tb.send_document(CHAT_ID, None)
             assert False
         except Exception as e:
             print(e)
             assert True
 
-
+    def test_send_photo_by_id(self):
-def test_send_photo_by_id():
         photo_id = 'AgADBQADTKgxG8YifgbcWQAB7Da9yYIx1rEyAAT-HYJ3CrJEqdA2AQABAg'
         tb = telebot.TeleBot(TOKEN)
         ret_msg = tb.send_photo(CHAT_ID, photo_id)
         assert ret_msg.message_id
 
-
+    def test_send_photo(self):
-def test_send_photo():
         file_data = open('../examples/detailed_example/kitten.jpg', 'rb')
         tb = telebot.TeleBot(TOKEN)
         ret_msg = tb.send_photo(CHAT_ID, file_data)
         assert ret_msg.message_id
 
-
+    def test_send_audio(self):
-def test_send_audio():
         file_data = open('./test_data/record.mp3', 'rb')
         tb = telebot.TeleBot(TOKEN)
         ret_msg = tb.send_audio(CHAT_ID, file_data, 1, 'eternnoir', 'pyTelegram')
@@ -123,38 +119,33 @@ def test_send_audio():
         assert ret_msg.audio.performer == 'eternnoir'
         assert ret_msg.audio.title == 'pyTelegram'
 
-
+    def test_send_voice(self):
-def test_send_voice():
         file_data = open('./test_data/record.ogg', 'rb')
         tb = telebot.TeleBot(TOKEN)
         ret_msg = tb.send_voice(CHAT_ID, file_data)
         assert ret_msg.voice.mime_type == 'audio/ogg'
 
-
+    def test_send_message(self):
-def test_send_message():
         text = 'CI Test Message'
         tb = telebot.TeleBot(TOKEN)
         ret_msg = tb.send_message(CHAT_ID, text)
         assert ret_msg.message_id
 
-
+    def test_forward_message(self):
-def test_forward_message():
         text = 'CI forward_message Test Message'
         tb = telebot.TeleBot(TOKEN)
         msg = tb.send_message(CHAT_ID, text)
         ret_msg = tb.forward_message(CHAT_ID, CHAT_ID, msg.message_id)
         assert ret_msg.forward_from
 
-
+    def test_reply_to(self):
-def test_reply_to():
         text = 'CI reply_to Test Message'
         tb = telebot.TeleBot(TOKEN)
         msg = tb.send_message(CHAT_ID, text)
         ret_msg = tb.reply_to(msg, text + ' REPLY')
         assert ret_msg.reply_to_message.message_id == msg.message_id
 
-
+    def test_register_for_reply(self):
-def test_register_for_reply():
         text = 'CI reply_to Test Message'
         tb = telebot.TeleBot(TOKEN)
         msg = tb.send_message(CHAT_ID, text, reply_markup=types.ForceReply())
@@ -167,8 +158,7 @@ def test_register_for_reply():
 
         tb.process_new_messages([reply_msg])
 
-
+    def test_send_location(self):
-def test_send_location():
         tb = telebot.TeleBot(TOKEN)
         lat = 26.3875591
         lon = -161.2901042
@@ -176,25 +166,19 @@ def test_send_location():
         assert int(ret_msg.location.longitude) == int(lon)
         assert int(ret_msg.location.latitude) == int(lat)
 
-
+    def create_text_message(self, text):
-def create_text_message(text):
         params = {'text': text}
         chat = types.User(11, 'test')
         return types.Message(1, None, None, chat, 'text', params)
 
-
+    def test_is_string_unicode(self):
-def test_is_string_unicode():
         s1 = u'string'
         assert apihelper.is_string(s1)
 
-
+    def test_is_string_string(self):
-def test_is_string_string():
         s1 = 'string'
         assert apihelper.is_string(s1)
 
-
+    def test_not_string(self):
-def test_not_string():
         i1 = 10
         assert not apihelper.is_string(i1)
-
-test_send_voice()