diff --git a/tests/test_messageentity.py b/tests/test_messageentity.py new file mode 100644 index 000000000..33346e187 --- /dev/null +++ b/tests/test_messageentity.py @@ -0,0 +1,71 @@ +#!/usr/bin/env python +# +# A library that provides a Python interface to the Telegram Bot API +# Copyright (C) 2015-2016 +# Leandro Toledo de Souza +# +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program. If not, see [http://www.gnu.org/licenses/]. +"""This module contains a object that represents Tests for Telegram +MessageEntity""" + +import sys + +if sys.version_info[0:2] == (2, 6): + import unittest2 as unittest +else: + import unittest + +sys.path.append('.') + +import telegram +from tests.base import BaseTest + + +class MessageEntityTest(BaseTest, unittest.TestCase): + """This object represents Tests for Telegram MessageEntity.""" + + def setUp(self): + self.type = 'type' + self.offset = 1 + self.length = 2 + self.url = 'url' + + self.json_dict = { + 'type': self.type, + 'offset': self.offset, + 'length': self.length, + 'url': self.url + } + + def test_sticker_de_json(self): + sticker = telegram.MessageEntity.de_json(self.json_dict) + + self.assertEqual(sticker.type, self.type) + self.assertEqual(sticker.offset, self.offset) + self.assertEqual(sticker.length, self.length) + + def test_sticker_to_json(self): + sticker = telegram.MessageEntity.de_json(self.json_dict) + + self.assertTrue(self.is_json(sticker.to_json())) + + def test_sticker_to_dict(self): + sticker = telegram.MessageEntity.de_json(self.json_dict).to_dict() + + self.assertTrue(self.is_dict(sticker)) + self.assertDictEqual(self.json_dict, sticker) + + +if __name__ == '__main__': + unittest.main()