diff --git a/tests/dataset.py b/tests/dataset.py new file mode 100644 index 00000000..be9432df --- /dev/null +++ b/tests/dataset.py @@ -0,0 +1,25 @@ +UPDATE = { + "update_id": 128526, + "message": { + "message_id": 11223, + "from": { + "id": 12345678, + "is_bot": False, + "first_name": "FirstName", + "last_name": "LastName", + "username": "username", + "language_code": "ru" + }, + "chat": { + "id": 12345678, + "first_name": "FirstName", + "last_name": "LastName", + "username": "username", + "type": "private" + }, + "date": 1508709711, + "text": "Hi, world!" + } +} + +MESSAGE = UPDATE['message'] diff --git a/tests/test_message.py b/tests/test_message.py new file mode 100644 index 00000000..a3cec0ad --- /dev/null +++ b/tests/test_message.py @@ -0,0 +1,35 @@ +import datetime +import unittest + +from aiogram import types +from dataset import MESSAGE + + +class TestMessage(unittest.TestCase): + def setUp(self): + self.message = types.Message(**MESSAGE) + + def test_update_id(self): + self.assertEqual(self.message.message_id, MESSAGE['message_id'], 'test') + self.assertEqual(self.message['message_id'], MESSAGE['message_id']) + + def test_from(self): + self.assertIsInstance(self.message.from_user, types.User) + self.assertEqual(self.message.from_user, self.message['from']) + + def test_chat(self): + self.assertIsInstance(self.message.chat, types.Chat) + self.assertEqual(self.message.chat, self.message['chat']) + + def test_date(self): + self.assertIsInstance(self.message.date, datetime.datetime) + self.assertEqual(int(self.message.date.timestamp()), MESSAGE['date']) + self.assertEqual(self.message.date, self.message['date']) + + def test_text(self): + self.assertEqual(self.message.text, MESSAGE['text']) + self.assertEqual(self.message['text'], MESSAGE['text']) + + +if __name__ == '__main__': + unittest.main() diff --git a/tests/utils.py b/tests/utils.py new file mode 100644 index 00000000..f1141634 --- /dev/null +++ b/tests/utils.py @@ -0,0 +1,2 @@ +def out(*message, sep=' '): + print('Test', sep.join(message))