tests and corrections for inlinequery

jh0ker · jh0ker · commit c55c5408b1fe · 2016-02-07T23:34:15.000+01:00
diff --git a/telegram/inlinequery.py b/telegram/inlinequery.py
@@ -29,7 +29,7 @@ class InlineQuery(TelegramObject):
         * In Python `from` is a reserved word, use `from_user` instead.
 
     Attributes:
-        id (int):
+        id (str):
         from_user (:class:`telegram.User`):
         query (str):
         offset (str):
@@ -48,7 +48,7 @@ def __init__(self,
                  query,
                  offset):
         # Required
-        self.id = int(id)
+        self.id = id
         self.from_user = from_user
         self.query = query
         self.offset = offset
diff --git a/tests/test_inlinequery.py b/tests/test_inlinequery.py
@@ -0,0 +1,84 @@
+#!/usr/bin/env python
+#
+# A library that provides a Python interface to the Telegram Bot API
+# Copyright (C) 2015-2016
+# Leandro Toledo de Souza <devs@python-telegram-bot.org>
+#
+# 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
+InlineQuery"""
+
+import os
+import unittest
+import sys
+sys.path.append('.')
+
+import telegram
+from tests.base import BaseTest
+
+
+class InlineQueryTest(BaseTest, unittest.TestCase):
+    """This object represents Tests for Telegram InlineQuery."""
+
+    def setUp(self):
+
+        user = telegram.User(1, 'First name')
+
+        self.id = 'id'
+        self.from_user = user
+        self.query = 'query text'
+        self.offset = 'offset'
+
+        self.json_dict = {
+            'id': self.id,
+            'from': self.from_user.to_dict(),
+            'query': self.query,
+            'offset': self.offset
+        }
+
+    def test_inlinequery_de_json(self):
+        """Test InlineQuery.de_json() method"""
+        print('Testing InlineQuery.de_json()')
+
+        inlinequery = telegram.InlineQuery.de_json(self.json_dict)
+
+        self.assertEqual(inlinequery.id, self.id)
+        self.assertEqual(inlinequery.from_user.to_dict(), self.from_user.to_dict())
+        self.assertEqual(inlinequery.query, self.query)
+        self.assertEqual(inlinequery.offset, self.offset)
+
+    def test_inlinequery_to_json(self):
+        """Test InlineQuery.to_json() method"""
+        print('Testing InlineQuery.to_json()')
+
+        inlinequery = telegram.InlineQuery.de_json(self.json_dict)
+
+        self.assertTrue(self.is_json(inlinequery.to_json()))
+
+    def test_inlinequery_to_dict(self):
+        """Test InlineQuery.to_dict() method"""
+        print('Testing InlineQuery.to_dict()')
+
+        inlinequery = telegram.InlineQuery.de_json(self.json_dict).to_dict()
+
+        self.assertTrue(self.is_dict(inlinequery))
+        self.assertEqual(inlinequery['id'], self.id)
+        self.assertEqual(inlinequery['from'], self.from_user.to_dict())
+        self.assertEqual(inlinequery['query'], self.query)
+        self.assertEqual(inlinequery['offset'], self.offset)
+
+
+if __name__ == '__main__':
+    unittest.main()
