python-telegram-bot · Bibo-Joshi · Sep 21, 2024 · May 21, 2024 · Sep 19, 2024 · Sep 19, 2024
diff --git a/tests/_files/conftest.py b/tests/_files/conftest.py
@@ -0,0 +1,189 @@
+#!/usr/bin/env python
+#
+#  A library that provides a Python interface to the Telegram Bot API
+#  Copyright (C) 2015-2024
+#  Leandro Toledo de Souza <[email protected]>
+#
+#  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/].
+"""Module to provide fixtures most of which are used in test_inputmedia.py."""
+import pytest
+
+from telegram.error import BadRequest
+from tests.auxil.files import data_file
+from tests.auxil.networking import expect_bad_request
+
+
+@pytest.fixture(scope="session")
+async def animation(bot, chat_id):
+    with data_file("game.gif").open("rb") as f, data_file("thumb.jpg").open("rb") as thumb:
+        return (
+            await bot.send_animation(chat_id, animation=f, read_timeout=50, thumbnail=thumb)
+        ).animation
+
+
+@pytest.fixture
+def animation_file():
+    with data_file("game.gif").open("rb") as f:
+        yield f
+
+
+@pytest.fixture(scope="module")
+async def animated_sticker(bot, chat_id):
+    with data_file("telegram_animated_sticker.tgs").open("rb") as f:
+        return (await bot.send_sticker(chat_id, sticker=f, read_timeout=50)).sticker
+
+
+@pytest.fixture
+def animated_sticker_file():
+    with data_file("telegram_animated_sticker.tgs").open("rb") as f:
+        yield f
+
+
+@pytest.fixture
+async def animated_sticker_set(bot):
+    ss = await bot.get_sticker_set(f"animated_test_by_{bot.username}")
+    if len(ss.stickers) > 100:
+        try:
+            for i in range(1, 50):
+                await bot.delete_sticker_from_set(ss.stickers[-i].file_id)
+        except BadRequest as e:
+            if e.message == "Stickerset_not_modified":
+                return ss
+            raise Exception("stickerset is growing too large.") from None
+    return ss
+
+
+@pytest.fixture(scope="session")
+async def audio(bot, chat_id):
+    with data_file("telegram.mp3").open("rb") as f, data_file("thumb.jpg").open("rb") as thumb:
+        return (await bot.send_audio(chat_id, audio=f, read_timeout=50, thumbnail=thumb)).audio
+
+
+@pytest.fixture
+def audio_file():
+    with data_file("telegram.mp3").open("rb") as f:
+        yield f
+
+
+@pytest.fixture(scope="session")
+async def document(bot, chat_id):
+    with data_file("telegram.png").open("rb") as f:
+        return (await bot.send_document(chat_id, document=f, read_timeout=50)).document
+
+
+@pytest.fixture
+def document_file():
+    with data_file("telegram.png").open("rb") as f:
+        yield f
+
+
+@pytest.fixture(scope="session")
+def photo(photolist):
+    return photolist[-1]
+
+
+@pytest.fixture
+def photo_file():
+    with data_file("telegram.jpg").open("rb") as f:
+        yield f
+
+
+@pytest.fixture(scope="session")
+async def photolist(bot, chat_id):
+    async def func():
+        with data_file("telegram.jpg").open("rb") as f:
+            return (await bot.send_photo(chat_id, photo=f, read_timeout=50)).photo
+
+    return await expect_bad_request(
+        func, "Type of file mismatch", "Telegram did not accept the file."
+    )
+
+
+@pytest.fixture(scope="module")
+async def sticker(bot, chat_id):
+    with data_file("telegram.webp").open("rb") as f:
+        sticker = (await bot.send_sticker(chat_id, sticker=f, read_timeout=50)).sticker
+        # necessary to properly test needs_repainting
+        with sticker._unfrozen():
+            sticker.needs_repainting = True
+        return sticker
+
+
+@pytest.fixture
+def sticker_file():
+    with data_file("telegram.webp").open("rb") as file:
+        yield file
+
+
+@pytest.fixture
+async def sticker_set(bot):
+    ss = await bot.get_sticker_set(f"test_by_{bot.username}")
+    if len(ss.stickers) > 100:
+        try:
+            for i in range(1, 50):
+                await bot.delete_sticker_from_set(ss.stickers[-i].file_id)
+        except BadRequest as e:
+            if e.message == "Stickerset_not_modified":
+                return ss
+            raise Exception("stickerset is growing too large.") from None
+    return ss
+
+
+@pytest.fixture
+def sticker_set_thumb_file():
+    with data_file("sticker_set_thumb.png").open("rb") as file:
+        yield file
+
+
+@pytest.fixture(scope="session")
+def thumb(photolist):
+    return photolist[0]
+
+
+@pytest.fixture(scope="session")
+async def video(bot, chat_id):
+    with data_file("telegram.mp4").open("rb") as f:
+        return (await bot.send_video(chat_id, video=f, read_timeout=50)).video
+
+
+@pytest.fixture
+def video_file():
+    with data_file("telegram.mp4").open("rb") as f:
+        yield f
+
+
+@pytest.fixture
+def video_sticker_file():
+    with data_file("telegram_video_sticker.webm").open("rb") as f:
+        yield f
+
+
+@pytest.fixture(scope="module")
+def video_sticker(bot, chat_id):
+    with data_file("telegram_video_sticker.webm").open("rb") as f:
+        return bot.send_sticker(chat_id, sticker=f, timeout=50).sticker
+
+
+@pytest.fixture
+async def video_sticker_set(bot):
+    ss = await bot.get_sticker_set(f"video_test_by_{bot.username}")
+    if len(ss.stickers) > 100:
+        try:
+            for i in range(1, 50):
+                await bot.delete_sticker_from_set(ss.stickers[-i].file_id)
+        except BadRequest as e:
+            if e.message == "Stickerset_not_modified":
+                return ss
+            raise Exception("stickerset is growing too large.") from None
+    return ss
diff --git a/tests/_files/test_animation.py b/tests/_files/test_animation.py
@@ -37,20 +37,6 @@
 from tests.auxil.slots import mro_slots
 
 
-@pytest.fixture
-def animation_file():
-    with data_file("game.gif").open("rb") as f:
-        yield f
-
-
-@pytest.fixture(scope="module")
-async def animation(bot, chat_id):
-    with data_file("game.gif").open("rb") as f, data_file("thumb.jpg").open("rb") as thumb:
-        return (
-            await bot.send_animation(chat_id, animation=f, read_timeout=50, thumbnail=thumb)
-        ).animation
-
-
 class AnimationTestBase:
     animation_file_id = "CgADAQADngIAAuyVeEez0xRovKi9VAI"
     animation_file_unique_id = "adc3145fd2e84d95b64d68eaa22aa33e"

diff --git a/tests/_files/test_audio.py b/tests/_files/test_audio.py
@@ -37,18 +37,6 @@
 from tests.auxil.slots import mro_slots
 
 
-@pytest.fixture
-def audio_file():
-    with data_file("telegram.mp3").open("rb") as f:
-        yield f
-
-
-@pytest.fixture(scope="module")
-async def audio(bot, chat_id):
-    with data_file("telegram.mp3").open("rb") as f, data_file("thumb.jpg").open("rb") as thumb:
-        return (await bot.send_audio(chat_id, audio=f, read_timeout=50, thumbnail=thumb)).audio
-
-
 class AudioTestBase:
     caption = "Test *audio*"
     performer = "Leandro Toledo"

diff --git a/tests/_files/test_document.py b/tests/_files/test_document.py
@@ -37,18 +37,6 @@
 from tests.auxil.slots import mro_slots
 
 
-@pytest.fixture
-def document_file():
-    with data_file("telegram.png").open("rb") as f:
-        yield f
-
-
-@pytest.fixture(scope="module")
-async def document(bot, chat_id):
-    with data_file("telegram.png").open("rb") as f:
-        return (await bot.send_document(chat_id, document=f, read_timeout=50)).document
-
-
 class DocumentTestBase:
     caption = "DocumentTest - *Caption*"
     document_file_url = "https://python-telegram-bot.org/static/testfiles/telegram.gif"