OpenSource/python-telegram-bot
OpenSource/python-telegram-bot
1
0
Fork 0
mirror of https://github.com/python-telegram-bot/python-telegram-bot.git synced 2024-11-21 14:46:29 +01:00
Code Issues Projects Releases Packages Wiki Activity

Maintenance Work on Bot Tests (#4489)

Browse source 
Co-authored-by: Harshil <37377066+harshil21@users.noreply.github.com>
This commit is contained in:
Bibo-Joshi 2024-09-21 11:34:25 +02:00 committed by GitHub
parent 28d19c3b9a
commit 2eae2830f3
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 14 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
tests/test_bot.py
View file

@ -90,7 +90,7 @@ from telegram.constants import (
ParseMode, ParseMode,
ReactionEmoji, ReactionEmoji,
) )
from telegram.error import BadRequest, EndPointNotFound, InvalidToken, NetworkError from telegram.error import BadRequest, EndPointNotFound, InvalidToken
from telegram.ext import ExtBot, InvalidCallbackData from telegram.ext import ExtBot, InvalidCallbackData
from telegram.helpers import escape_markdown from telegram.helpers import escape_markdown
from telegram.request import BaseRequest, HTTPXRequest, RequestData from telegram.request import BaseRequest, HTTPXRequest, RequestData
@ -2264,6 +2264,19 @@ class TestBotWithoutRequest:
obj = await offline_bot.get_business_connection(business_connection_id=bci) obj = await offline_bot.get_business_connection(business_connection_id=bci)
assert isinstance(obj, BusinessConnection) assert isinstance(obj, BusinessConnection)
async def test_send_chat_action_all_args(self, bot, chat_id, monkeypatch):
async def make_assertion(*args, **_):
kwargs = args[1]
return (
kwargs["chat_id"] == chat_id
and kwargs["action"] == "action"
and kwargs["message_thread_id"] == 1
and kwargs["business_connection_id"] == 3
)
monkeypatch.setattr(bot, "_post", make_assertion)
assert await bot.send_chat_action(chat_id, "action", 1, 3)
async def test_refund_star_payment(self, offline_bot, monkeypatch): async def test_refund_star_payment(self, offline_bot, monkeypatch):
# can't make actual request so we just test that the correct data is passed # can't make actual request so we just test that the correct data is passed
async def make_assertion(url, request_data: RequestData, *args, **kwargs): async def make_assertion(url, request_data: RequestData, *args, **kwargs):
@ -2395,8 +2408,6 @@ class TestBotWithRequest:
async def test_delete_message(self, bot, chat_id): async def test_delete_message(self, bot, chat_id):
message = await bot.send_message(chat_id, text="will be deleted") message = await bot.send_message(chat_id, text="will be deleted")
await asyncio.sleep(2)
assert await bot.delete_message(chat_id=chat_id, message_id=message.message_id) is True assert await bot.delete_message(chat_id=chat_id, message_id=message.message_id) is True
async def test_delete_message_old_message(self, bot, chat_id): async def test_delete_message_old_message(self, bot, chat_id):
@ -2488,18 +2499,6 @@ class TestBotWithRequest:
assert message.contact.last_name == last_name assert message.contact.last_name == last_name
assert message.has_protected_content assert message.has_protected_content
async def test_send_chat_action_all_args(self, bot, chat_id, monkeypatch):
async def make_assertion(*args, **_):
kwargs = args[1]
return (
kwargs["chat_id"] == chat_id
and kwargs["action"] == "action"
and kwargs["message_thread_id"] == 1
)
monkeypatch.setattr(bot, "_post", make_assertion)
assert await bot.send_chat_action(chat_id, "action", 1)
# TODO: Add bot to group to test polls too # TODO: Add bot to group to test polls too
@pytest.mark.parametrize( @pytest.mark.parametrize(
"reply_markup", "reply_markup",
@ -3108,9 +3107,6 @@ class TestBotWithRequest:
with pytest.raises(BadRequest, match="Chat not found"): with pytest.raises(BadRequest, match="Chat not found"):
await bot.leave_chat(-123456) await bot.leave_chat(-123456)
with pytest.raises(NetworkError, match="Chat not found"):
await bot.leave_chat(-123456)
async def test_get_chat(self, bot, super_group_id): async def test_get_chat(self, bot, super_group_id):
cfi = await bot.get_chat(super_group_id) cfi = await bot.get_chat(super_group_id)
assert cfi.type == "supergroup" assert cfi.type == "supergroup"