Test UserProfilePhotos (python-telegram-bot#796)

jsmnbom · Eldinnie · commit d7929a8de4ee · 2017-09-01T08:44:26.000+02:00
diff --git a/tests/test_userprofilephotos.py b/tests/test_userprofilephotos.py
@@ -0,0 +1,50 @@
+#!/usr/bin/env python
+#
+# A library that provides a Python interface to the Telegram Bot API
+# Copyright (C) 2015-2017
+# 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 Lesser 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 Lesser Public License for more details.
+#
+# You should have received a copy of the GNU Lesser Public License
+# along with this program.  If not, see [http://www.gnu.org/licenses/].
+from telegram import PhotoSize, UserProfilePhotos
+
+
+class TestUserProfilePhotos(object):
+    total_count = 2
+    photos = [
+        [
+            PhotoSize('file_id1', 512, 512),
+            PhotoSize('file_id2', 512, 512)
+        ],
+        [
+            PhotoSize('file_id3', 512, 512),
+            PhotoSize('file_id4', 512, 512)
+        ]
+    ]
+
+    def test_de_json(self, bot):
+        json_dict = {
+            'total_count': 2,
+            'photos': [[y.to_dict() for y in x] for x in self.photos]
+        }
+        user_profile_photos = UserProfilePhotos.de_json(json_dict, bot)
+        assert user_profile_photos.total_count == self.total_count
+        assert user_profile_photos.photos == self.photos
+
+    def test_to_dict(self):
+        user_profile_photos = UserProfilePhotos(self.total_count, self.photos)
+        user_profile_photos_dict = user_profile_photos.to_dict()
+        assert user_profile_photos_dict['total_count'] == user_profile_photos.total_count
+        for ix, x in enumerate(user_profile_photos_dict['photos']):
+            for iy, y in enumerate(x):
+                assert y == user_profile_photos.photos[ix][iy].to_dict()
