mirror of
https://github.com/Alishahryar1/free-claude-code.git
synced 2026-04-28 11:30:03 +00:00
115 lines
3.6 KiB
Python
115 lines
3.6 KiB
Python
from unittest.mock import AsyncMock, MagicMock, patch
|
|
|
|
import pytest
|
|
|
|
from messaging.platforms.telegram import TelegramPlatform
|
|
|
|
|
|
@pytest.fixture
|
|
def telegram_platform():
|
|
with patch("messaging.platforms.telegram.TELEGRAM_AVAILABLE", True):
|
|
platform = TelegramPlatform(bot_token="test_token", allowed_user_id="12345")
|
|
return platform
|
|
|
|
|
|
def test_telegram_platform_init_no_token():
|
|
with patch.dict("os.environ", {}, clear=True):
|
|
platform = TelegramPlatform(bot_token=None)
|
|
assert platform.bot_token is None
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_telegram_platform_start_success(telegram_platform):
|
|
with patch("telegram.ext.Application.builder") as mock_builder:
|
|
mock_app = MagicMock()
|
|
mock_app.initialize = AsyncMock()
|
|
mock_app.start = AsyncMock()
|
|
mock_app.updater.start_polling = AsyncMock()
|
|
|
|
mock_builder.return_value.token.return_value.request.return_value.build.return_value = mock_app
|
|
|
|
# Mock MessagingRateLimiter
|
|
with patch("messaging.limiter.MessagingRateLimiter.get_instance", AsyncMock()):
|
|
await telegram_platform.start()
|
|
|
|
assert telegram_platform._connected is True
|
|
mock_app.initialize.assert_called_once()
|
|
mock_app.start.assert_called_once()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_telegram_platform_send_message_success(telegram_platform):
|
|
mock_bot = AsyncMock()
|
|
mock_msg = MagicMock()
|
|
mock_msg.message_id = 999
|
|
mock_bot.send_message.return_value = mock_msg
|
|
|
|
telegram_platform._application = MagicMock()
|
|
telegram_platform._application.bot = mock_bot
|
|
|
|
msg_id = await telegram_platform.send_message("chat_1", "hello")
|
|
|
|
assert msg_id == "999"
|
|
mock_bot.send_message.assert_called_once_with(
|
|
chat_id="chat_1",
|
|
text="hello",
|
|
reply_to_message_id=None,
|
|
parse_mode="MarkdownV2",
|
|
)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_telegram_platform_edit_message_success(telegram_platform):
|
|
mock_bot = AsyncMock()
|
|
telegram_platform._application = MagicMock()
|
|
telegram_platform._application.bot = mock_bot
|
|
|
|
await telegram_platform.edit_message("chat_1", "999", "new text")
|
|
|
|
mock_bot.edit_message_text.assert_called_once_with(
|
|
chat_id="chat_1", message_id=999, text="new text", parse_mode="MarkdownV2"
|
|
)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_telegram_platform_queue_send_message(telegram_platform):
|
|
mock_limiter = AsyncMock()
|
|
telegram_platform._limiter = mock_limiter
|
|
|
|
await telegram_platform.queue_send_message("chat_1", "hello", fire_and_forget=False)
|
|
|
|
mock_limiter.enqueue.assert_called_once()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_on_telegram_message_authorized(telegram_platform):
|
|
handler = AsyncMock()
|
|
telegram_platform.on_message(handler)
|
|
|
|
mock_update = MagicMock()
|
|
mock_update.message.text = "hello"
|
|
mock_update.message.message_id = 1
|
|
mock_update.effective_user.id = 12345
|
|
mock_update.effective_chat.id = 6789
|
|
mock_update.message.reply_to_message = None
|
|
|
|
await telegram_platform._on_telegram_message(mock_update, MagicMock())
|
|
|
|
handler.assert_called_once()
|
|
incoming = handler.call_args[0][0]
|
|
assert incoming.text == "hello"
|
|
assert incoming.user_id == "12345"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_on_telegram_message_unauthorized(telegram_platform):
|
|
handler = AsyncMock()
|
|
telegram_platform.on_message(handler)
|
|
|
|
mock_update = MagicMock()
|
|
mock_update.message.text = "hello"
|
|
mock_update.effective_user.id = 99999 # Unauthorized
|
|
|
|
await telegram_platform._on_telegram_message(mock_update, MagicMock())
|
|
|
|
handler.assert_not_called()
|