#!/usr/bin/env python # # A library that provides a Python interface to the Telegram Bot API # Copyright (C) 2015-2017 # 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 an object that represents Tests for Telegram InlineQueryResultDocument""" import sys import unittest sys.path.append('.') import telegram from tests.base import BaseTest class InlineQueryResultDocumentTest(BaseTest, unittest.TestCase): """This object represents Tests for Telegram InlineQueryResultDocument.""" def setUp(self): self._id = 'id' self.type = 'document' self.document_url = 'document url' self.title = 'title' self.caption = 'caption' self.mime_type = 'mime type' self.description = 'description' self.thumb_url = 'thumb url' self.thumb_width = 10 self.thumb_height = 15 self.input_message_content = telegram.InputTextMessageContent('input_message_content') self.reply_markup = telegram.InlineKeyboardMarkup( [[telegram.InlineKeyboardButton('reply_markup')]]) self.json_dict = { 'id': self._id, 'type': self.type, 'document_url': self.document_url, 'title': self.title, 'caption': self.caption, 'mime_type': self.mime_type, 'description': self.description, 'thumb_url': self.thumb_url, 'thumb_width': self.thumb_width, 'thumb_height': self.thumb_height, 'input_message_content': self.input_message_content.to_dict(), 'reply_markup': self.reply_markup.to_dict(), } def test_document_de_json(self): document = telegram.InlineQueryResultDocument.de_json(self.json_dict, self._bot) self.assertEqual(document.id, self._id) self.assertEqual(document.type, self.type) self.assertEqual(document.document_url, self.document_url) self.assertEqual(document.title, self.title) self.assertEqual(document.caption, self.caption) self.assertEqual(document.mime_type, self.mime_type) self.assertEqual(document.description, self.description) self.assertEqual(document.thumb_url, self.thumb_url) self.assertEqual(document.thumb_width, self.thumb_width) self.assertEqual(document.thumb_height, self.thumb_height) self.assertDictEqual(document.input_message_content.to_dict(), self.input_message_content.to_dict()) self.assertDictEqual(document.reply_markup.to_dict(), self.reply_markup.to_dict()) def test_document_to_json(self): document = telegram.InlineQueryResultDocument.de_json(self.json_dict, self._bot) self.assertTrue(self.is_json(document.to_json())) def test_document_to_dict(self): document = telegram.InlineQueryResultDocument.de_json(self.json_dict, self._bot).to_dict() self.assertTrue(self.is_dict(document)) self.assertDictEqual(self.json_dict, document) def test_equality(self): a = telegram.InlineQueryResultDocument(self._id, self.document_url, self.title, self.mime_type) b = telegram.InlineQueryResultDocument(self._id, self.document_url, self.title, self.mime_type) c = telegram.InlineQueryResultDocument(self._id, "", self.title, self.mime_type) d = telegram.InlineQueryResultDocument("", self.document_url, self.title, self.mime_type) e = telegram.InlineQueryResultArticle(self._id, "", "") self.assertEqual(a, b) self.assertEqual(hash(a), hash(b)) self.assertIsNot(a, b) self.assertEqual(a, c) self.assertEqual(hash(a), hash(c)) self.assertNotEqual(a, d) self.assertNotEqual(hash(a), hash(d)) self.assertNotEqual(a, e) self.assertNotEqual(hash(a), hash(e)) if __name__ == '__main__': unittest.main()