mirror of
https://github.com/aiogram/aiogram.git
synced 2025-12-09 09:22:03 +00:00
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
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()
|