mirror of
https://github.com/Alishahryar1/free-claude-code.git
synced 2026-04-28 03:20:01 +00:00
b926f60f64
- Add local web_search/web_fetch SSE handling and optional tool schemas - Extend HeuristicToolParser for JSON-style WebFetch/WebSearch text - Consolidate provider defaults, ids, and exception typing; stream contracts - Messaging: typed options, voice config injection, platform contract cleanup - Tests for web server tools, converters, parsers, contracts; ignore debug-*.log
116 lines
3.9 KiB
Python
116 lines
3.9 KiB
Python
"""Tests for messaging platform factory."""
|
|
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
from messaging.platforms.factory import (
|
|
MessagingPlatformOptions,
|
|
create_messaging_platform,
|
|
)
|
|
|
|
|
|
class TestCreateMessagingPlatform:
|
|
"""Tests for create_messaging_platform factory function."""
|
|
|
|
def test_telegram_with_token(self):
|
|
"""Create Telegram platform when bot_token is provided."""
|
|
mock_platform = MagicMock()
|
|
with (
|
|
patch("messaging.platforms.telegram.TELEGRAM_AVAILABLE", True),
|
|
patch(
|
|
"messaging.platforms.telegram.TelegramPlatform",
|
|
return_value=mock_platform,
|
|
) as platform_cls,
|
|
):
|
|
result = create_messaging_platform(
|
|
"telegram",
|
|
MessagingPlatformOptions(
|
|
telegram_bot_token="test_token",
|
|
allowed_telegram_user_id="12345",
|
|
voice_note_enabled=False,
|
|
whisper_model="large-v3",
|
|
whisper_device="cuda",
|
|
),
|
|
)
|
|
|
|
assert result is mock_platform
|
|
platform_cls.assert_called_once_with(
|
|
bot_token="test_token",
|
|
allowed_user_id="12345",
|
|
voice_note_enabled=False,
|
|
whisper_model="large-v3",
|
|
whisper_device="cuda",
|
|
hf_token="",
|
|
nvidia_nim_api_key="",
|
|
)
|
|
|
|
def test_telegram_without_token(self):
|
|
"""Return None when no bot_token for Telegram."""
|
|
result = create_messaging_platform("telegram")
|
|
assert result is None
|
|
|
|
def test_telegram_empty_token(self):
|
|
"""Return None when bot_token is empty string."""
|
|
result = create_messaging_platform(
|
|
"telegram", MessagingPlatformOptions(telegram_bot_token="")
|
|
)
|
|
assert result is None
|
|
|
|
def test_discord_with_token(self):
|
|
"""Create Discord platform when discord_bot_token is provided."""
|
|
mock_platform = MagicMock()
|
|
with (
|
|
patch("messaging.platforms.discord.DISCORD_AVAILABLE", True),
|
|
patch(
|
|
"messaging.platforms.discord.DiscordPlatform",
|
|
return_value=mock_platform,
|
|
) as platform_cls,
|
|
):
|
|
result = create_messaging_platform(
|
|
"discord",
|
|
MessagingPlatformOptions(
|
|
discord_bot_token="test_token",
|
|
allowed_discord_channels="123,456",
|
|
voice_note_enabled=False,
|
|
whisper_model="small",
|
|
whisper_device="nvidia_nim",
|
|
),
|
|
)
|
|
|
|
assert result is mock_platform
|
|
platform_cls.assert_called_once_with(
|
|
bot_token="test_token",
|
|
allowed_channel_ids="123,456",
|
|
voice_note_enabled=False,
|
|
whisper_model="small",
|
|
whisper_device="nvidia_nim",
|
|
hf_token="",
|
|
nvidia_nim_api_key="",
|
|
)
|
|
|
|
def test_discord_without_token(self):
|
|
"""Return None when no discord_bot_token for Discord."""
|
|
result = create_messaging_platform("discord")
|
|
assert result is None
|
|
|
|
def test_discord_empty_token(self):
|
|
"""Return None when discord_bot_token is empty string."""
|
|
result = create_messaging_platform(
|
|
"discord",
|
|
MessagingPlatformOptions(
|
|
discord_bot_token="",
|
|
allowed_discord_channels="123",
|
|
),
|
|
)
|
|
assert result is None
|
|
|
|
def test_unknown_platform(self):
|
|
"""Return None for unknown platform types."""
|
|
result = create_messaging_platform("slack")
|
|
assert result is None
|
|
|
|
def test_unknown_platform_with_kwargs(self):
|
|
"""Return None for unknown platform even with kwargs."""
|
|
result = create_messaging_platform(
|
|
"slack", MessagingPlatformOptions(telegram_bot_token="token")
|
|
)
|
|
assert result is None
|