Remove obsolete configuration management and test settings files
- Deleted the `config.py` file responsible for managing bot configuration via environment variables and `.env` files, as it is no longer needed. - Removed the `test_settings.ini` file used for testing, which contained mock configuration data. - Cleaned up the project structure by eliminating unused files to enhance maintainability.
This commit is contained in:
231
tests/test_voice_services.py
Normal file
231
tests/test_voice_services.py
Normal file
@@ -0,0 +1,231 @@
|
||||
import pytest
|
||||
from unittest.mock import Mock, AsyncMock, patch, MagicMock
|
||||
from pathlib import Path
|
||||
from datetime import datetime
|
||||
|
||||
from helper_bot.handlers.voice.services import VoiceBotService
|
||||
from helper_bot.handlers.voice.exceptions import VoiceMessageError, AudioProcessingError
|
||||
|
||||
|
||||
class TestVoiceBotService:
|
||||
"""Тесты для VoiceBotService"""
|
||||
|
||||
@pytest.fixture
|
||||
def mock_bot_db(self):
|
||||
"""Мок для базы данных"""
|
||||
mock_db = Mock()
|
||||
mock_db.settings = {
|
||||
'Settings': {'logs': True},
|
||||
'Telegram': {'important_logs': 'test_chat_id'}
|
||||
}
|
||||
return mock_db
|
||||
|
||||
@pytest.fixture
|
||||
def mock_settings(self):
|
||||
"""Мок для настроек"""
|
||||
return {
|
||||
'Settings': {'logs': True},
|
||||
'Telegram': {'preview_link': True}
|
||||
}
|
||||
|
||||
@pytest.fixture
|
||||
def voice_service(self, mock_bot_db, mock_settings):
|
||||
"""Экземпляр VoiceBotService для тестов"""
|
||||
return VoiceBotService(mock_bot_db, mock_settings)
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_get_welcome_sticker_success(self, voice_service, mock_settings):
|
||||
"""Тест успешного получения стикера"""
|
||||
with patch('pathlib.Path.rglob') as mock_rglob:
|
||||
mock_rglob.return_value = ['/path/to/sticker1.tgs', '/path/to/sticker2.tgs']
|
||||
|
||||
sticker = await voice_service.get_welcome_sticker()
|
||||
|
||||
assert sticker is not None
|
||||
mock_rglob.assert_called_once()
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_get_welcome_sticker_no_stickers(self, voice_service, mock_settings):
|
||||
"""Тест получения стикера когда их нет"""
|
||||
with patch('pathlib.Path.rglob') as mock_rglob:
|
||||
mock_rglob.return_value = []
|
||||
|
||||
sticker = await voice_service.get_welcome_sticker()
|
||||
|
||||
assert sticker is None
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_get_welcome_sticker_only_webp_files(self, voice_service, mock_settings):
|
||||
"""Тест получения стикера когда есть только webp файлы"""
|
||||
with patch('pathlib.Path.rglob') as mock_rglob:
|
||||
mock_rglob.return_value = ['/path/to/sticker1.webp', '/path/to/sticker2.webp']
|
||||
|
||||
sticker = await voice_service.get_welcome_sticker()
|
||||
|
||||
# Проверяем, что стикер не None (метод ищет файлы по паттерну Hello_*)
|
||||
assert sticker is not None
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_get_welcome_sticker_mixed_files(self, voice_service, mock_settings):
|
||||
"""Тест получения стикера когда есть смешанные файлы"""
|
||||
with patch('pathlib.Path.rglob') as mock_rglob:
|
||||
mock_rglob.return_value = [
|
||||
'/path/to/sticker1.webp',
|
||||
'/path/to/sticker2.tgs',
|
||||
'/path/to/sticker3.webp'
|
||||
]
|
||||
|
||||
sticker = await voice_service.get_welcome_sticker()
|
||||
|
||||
assert sticker is not None
|
||||
# Проверяем, что стикер не None (метод возвращает FSInputFile объект)
|
||||
|
||||
def test_get_random_audio_success(self, voice_service, mock_bot_db):
|
||||
"""Тест успешного получения случайного аудио"""
|
||||
mock_bot_db.check_listen_audio.return_value = ['audio1', 'audio2']
|
||||
mock_bot_db.get_user_id_by_file_name.return_value = 123
|
||||
mock_bot_db.get_date_by_file_name.return_value = '2025-01-01 12:00:00'
|
||||
mock_bot_db.check_emoji_for_user.return_value = '😊'
|
||||
|
||||
result = voice_service.get_random_audio(456)
|
||||
|
||||
assert result is not None
|
||||
assert len(result) == 3
|
||||
# Проверяем, что результат содержит ожидаемые данные, но не проверяем точное значение audio
|
||||
assert result[0] in ['audio1', 'audio2']
|
||||
assert result[1] == '2025-01-01 12:00:00'
|
||||
assert result[2] == '😊'
|
||||
|
||||
def test_get_random_audio_no_audio(self, voice_service, mock_bot_db):
|
||||
"""Тест получения аудио когда их нет"""
|
||||
mock_bot_db.check_listen_audio.return_value = []
|
||||
|
||||
result = voice_service.get_random_audio(456)
|
||||
|
||||
assert result is None
|
||||
|
||||
def test_get_random_audio_single_audio(self, voice_service, mock_bot_db):
|
||||
"""Тест получения аудио когда есть только одно"""
|
||||
mock_bot_db.check_listen_audio.return_value = ['audio1']
|
||||
mock_bot_db.get_user_id_by_file_name.return_value = 123
|
||||
mock_bot_db.get_date_by_file_name.return_value = '2025-01-01 12:00:00'
|
||||
mock_bot_db.check_emoji_for_user.return_value = '😊'
|
||||
|
||||
result = voice_service.get_random_audio(456)
|
||||
|
||||
assert result is not None
|
||||
assert len(result) == 3
|
||||
assert result[0] == 'audio1'
|
||||
|
||||
def test_mark_audio_as_listened_success(self, voice_service, mock_bot_db):
|
||||
"""Тест успешной пометки аудио как прослушанного"""
|
||||
voice_service.mark_audio_as_listened('test_audio', 123)
|
||||
|
||||
mock_bot_db.mark_listened_audio.assert_called_once_with('test_audio', user_id=123)
|
||||
|
||||
def test_clear_user_listenings_success(self, voice_service, mock_bot_db):
|
||||
"""Тест успешной очистки прослушиваний"""
|
||||
voice_service.clear_user_listenings(123)
|
||||
|
||||
mock_bot_db.delete_listen_count_for_user.assert_called_once_with(123)
|
||||
|
||||
def test_get_remaining_audio_count_success(self, voice_service, mock_bot_db):
|
||||
"""Тест получения количества оставшихся аудио"""
|
||||
mock_bot_db.check_listen_audio.return_value = ['audio1', 'audio2', 'audio3']
|
||||
|
||||
result = voice_service.get_remaining_audio_count(123)
|
||||
|
||||
assert result == 3
|
||||
mock_bot_db.check_listen_audio.assert_called_once_with(user_id=123)
|
||||
|
||||
def test_get_remaining_audio_count_zero(self, voice_service, mock_bot_db):
|
||||
"""Тест получения количества оставшихся аудио когда их нет"""
|
||||
mock_bot_db.check_listen_audio.return_value = []
|
||||
|
||||
result = voice_service.get_remaining_audio_count(123)
|
||||
|
||||
assert result == 0
|
||||
mock_bot_db.check_listen_audio.assert_called_once_with(user_id=123)
|
||||
|
||||
@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
|
||||
|
||||
@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
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_send_welcome_messages_with_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()
|
||||
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
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_get_welcome_sticker_with_tgs_files(self, voice_service, mock_settings):
|
||||
"""Тест получения стикера когда есть .tgs файлы"""
|
||||
with patch('pathlib.Path.rglob') as mock_rglob:
|
||||
mock_rglob.return_value = ['/path/to/sticker1.tgs', '/path/to/sticker2.tgs']
|
||||
|
||||
sticker = await voice_service.get_welcome_sticker()
|
||||
|
||||
assert sticker is not None
|
||||
# Проверяем, что стикер не None (метод возвращает FSInputFile объект)
|
||||
|
||||
def test_service_initialization(self, mock_bot_db, mock_settings):
|
||||
"""Тест инициализации сервиса"""
|
||||
service = VoiceBotService(mock_bot_db, mock_settings)
|
||||
|
||||
assert service.bot_db == mock_bot_db
|
||||
assert service.settings == mock_settings
|
||||
|
||||
def test_service_attributes(self, voice_service):
|
||||
"""Тест атрибутов сервиса"""
|
||||
assert hasattr(voice_service, 'bot_db')
|
||||
assert hasattr(voice_service, 'settings')
|
||||
assert hasattr(voice_service, 'get_welcome_sticker')
|
||||
assert hasattr(voice_service, 'get_random_audio')
|
||||
assert hasattr(voice_service, 'mark_audio_as_listened')
|
||||
assert hasattr(voice_service, 'clear_user_listenings')
|
||||
assert hasattr(voice_service, 'get_remaining_audio_count')
|
||||
assert hasattr(voice_service, 'send_welcome_messages')
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
pytest.main([__file__])
|
||||
Reference in New Issue
Block a user