merge 2
This commit is contained in:
@@ -123,36 +123,32 @@ class TestVoiceBotService:
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_send_welcome_messages_success(self, voice_service, mock_bot_db, mock_settings):
|
||||
"""Тест успешной отправки приветственных сообщений"""
|
||||
"""Тест успешной отправки приветственных сообщений."""
|
||||
mock_message = Mock()
|
||||
mock_message.from_user.id = 123
|
||||
mock_message.answer = AsyncMock()
|
||||
mock_message.answer.return_value = Mock()
|
||||
mock_message.answer_sticker = AsyncMock()
|
||||
|
||||
with patch.object(voice_service, 'get_welcome_sticker') as mock_sticker:
|
||||
mock_sticker.return_value = 'test_sticker.tgs'
|
||||
|
||||
await voice_service.send_welcome_messages(mock_message, '😊')
|
||||
|
||||
# Проверяем, что сообщения отправлены
|
||||
assert mock_message.answer.call_count >= 1
|
||||
|
||||
|
||||
with patch.object(voice_service, 'get_welcome_sticker', new_callable=AsyncMock, return_value='test_sticker.tgs'):
|
||||
with patch('helper_bot.handlers.voice.services.asyncio.sleep', new_callable=AsyncMock):
|
||||
await voice_service.send_welcome_messages(mock_message, '😊')
|
||||
|
||||
assert mock_message.answer.call_count >= 1
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_send_welcome_messages_no_sticker(self, voice_service, mock_bot_db, mock_settings):
|
||||
"""Тест отправки приветственных сообщений без стикера"""
|
||||
"""Тест отправки приветственных сообщений без стикера."""
|
||||
mock_message = Mock()
|
||||
mock_message.from_user.id = 123
|
||||
mock_message.answer = AsyncMock()
|
||||
mock_message.answer.return_value = Mock()
|
||||
|
||||
with patch.object(voice_service, 'get_welcome_sticker') as mock_sticker:
|
||||
mock_sticker.return_value = None
|
||||
|
||||
await voice_service.send_welcome_messages(mock_message, '😊')
|
||||
|
||||
# Проверяем, что сообщения отправлены
|
||||
assert mock_message.answer.call_count >= 1
|
||||
|
||||
with patch.object(voice_service, 'get_welcome_sticker', new_callable=AsyncMock, return_value=None):
|
||||
with patch('helper_bot.handlers.voice.services.asyncio.sleep', new_callable=AsyncMock):
|
||||
await voice_service.send_welcome_messages(mock_message, '😊')
|
||||
|
||||
assert mock_message.answer.call_count >= 1
|
||||
|
||||
|
||||
class TestVoiceHandlers:
|
||||
|
||||
Reference in New Issue
Block a user