Enhance bot functionality and refactor database interactions
- Added `ca-certificates` installation to Dockerfile for improved network security. - Updated health check command in Dockerfile to include better timeout handling. - Refactored `run_helper.py` to implement proper signal handling and logging during shutdown. - Transitioned database operations to an asynchronous model in `async_db.py`, improving performance and responsiveness. - Updated database schema to support new foreign key relationships and optimized indexing for better query performance. - Enhanced various bot handlers to utilize async database methods, improving overall efficiency and user experience. - Removed obsolete database and fix scripts to streamline the project structure.
This commit is contained in:
@@ -6,7 +6,7 @@ from unittest.mock import Mock, AsyncMock, patch
|
||||
from aiogram.types import Message, User, Chat
|
||||
from aiogram.fsm.context import FSMContext
|
||||
|
||||
from database.db import BotDB
|
||||
from database.async_db import AsyncBotDB
|
||||
|
||||
# Импортируем моки в самом начале
|
||||
import tests.mocks
|
||||
@@ -58,15 +58,15 @@ def mock_state():
|
||||
@pytest.fixture
|
||||
def mock_db():
|
||||
"""Создает мок базы данных для тестов"""
|
||||
db = Mock(spec=BotDB)
|
||||
db = Mock(spec=AsyncBotDB)
|
||||
db.user_exists = Mock(return_value=False)
|
||||
db.add_new_user_in_db = Mock()
|
||||
db.update_date_for_user = Mock()
|
||||
db.update_username_and_full_name = Mock()
|
||||
db.add_new_user = Mock()
|
||||
db.update_user_date = Mock()
|
||||
db.update_user_info = Mock()
|
||||
db.add_post_in_db = Mock()
|
||||
db.update_info_about_stickers = Mock()
|
||||
db.update_stickers_info = Mock()
|
||||
db.add_new_message_in_db = Mock()
|
||||
db.get_info_about_stickers = Mock(return_value=False)
|
||||
db.get_stickers_info = Mock(return_value=False)
|
||||
db.get_username_and_full_name = Mock(return_value=("testuser", "Test User"))
|
||||
return db
|
||||
|
||||
|
||||
Reference in New Issue
Block a user