Первый коммит

tests/database/test_messages.py

@@ -0,0 +1,125 @@
+from datetime import datetime, timezone
+from typing import List
+
+import pytest
+from sqlalchemy import select, Sequence
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from database import UserMessage, BotDatabase
+
+
+@pytest.mark.asyncio
+class TestMessageManagement:
+    """Тесты для управления сообщениями с полной строгой типизацией"""
+
+    async def test_message_creation(
+        self, test_db: BotDatabase, test_session: AsyncSession, test_user: int
+    ) -> None:
+        """
+        Тест создания сообщения.
+        Проверяет, что сообщение успешно сохраняется в базе и содержит правильные данные.
+        """
+        user_id: int = test_user
+        test_text: str = "Тестовое сообщение для проверки"
+
+        await test_db.add_message(user_id, test_text)
+
+        stmt = select(UserMessage).where(UserMessage.user_id == user_id)
+        result = await test_session.execute(stmt)
+        messages: Sequence[UserMessage] = result.scalars().all()
+
+        assert len(messages) == 1
+        assert messages[0].message_text == test_text
+        assert messages[0].user_id == user_id
+        assert messages[0].created_at is not None
+
+    async def test_message_with_custom_date(
+        self, test_db: BotDatabase, test_session: AsyncSession
+    ) -> None:
+        """
+        Тест добавления сообщения с кастомной датой.
+        Проверяет, что дата создания сохраняется корректно.
+        """
+        user_id: int = 999888777
+        custom_date: datetime = datetime(2024, 1, 15, 12, 30, 0, tzinfo=timezone.utc)
+
+        await test_db.add_user(user_id, "test_user", "Test User")
+        await test_db.add_message(
+            user_id=user_id,
+            message_text="Сообщение с кастомной датой",
+            created_at=custom_date
+        )
+
+        stmt = select(UserMessage).where(UserMessage.user_id == user_id)
+        result = await test_session.execute(stmt)
+        messages: Sequence[UserMessage] = result.scalars().all()
+
+        assert len(messages) == 1
+        db_date: datetime = messages[0].created_at
+        if db_date.tzinfo is not None:
+            db_date = db_date.replace(tzinfo=None)
+        expected_date: datetime = custom_date.replace(tzinfo=None)
+        assert db_date == expected_date
+
+    async def test_multiple_messages(
+        self, test_db: BotDatabase, test_session: AsyncSession, test_user: int
+    ) -> None:
+        """
+        Тест добавления нескольких сообщений.
+        Проверяет, что все сообщения корректно сохраняются в базе.
+        """
+        user_id: int = test_user
+
+        # Удаляем старые сообщения
+        async with test_db.session_factory() as session:
+            stmt = select(UserMessage).where(UserMessage.user_id == user_id)
+            result = await session.execute(stmt)
+            old_messages: Sequence[UserMessage] = result.scalars().all()
+            for msg in old_messages:
+                await session.delete(msg)
+            await session.commit()
+
+        # Добавляем несколько сообщений
+        for i in range(5):
+            await test_db.add_message(
+                user_id=user_id,
+                message_text=f"Сообщение {i + 1}"
+            )
+
+        stmt = select(UserMessage).where(UserMessage.user_id == user_id)
+        result = await test_session.execute(stmt)
+        messages: Sequence[UserMessage] = result.scalars().all()
+
+        assert len(messages) == 5
+
+    async def test_message_ordering(
+        self, test_db: BotDatabase, test_session: AsyncSession, test_user: int
+    ) -> None:
+        """
+        Тест проверки порядка сообщений по дате создания.
+        Сообщения должны возвращаться в порядке возрастания даты.
+        """
+        user_id: int = test_user
+
+        # Очищаем старые сообщения
+        async with test_db.session_factory() as session:
+            stmt = select(UserMessage).where(UserMessage.user_id == user_id)
+            result = await session.execute(stmt)
+            old_messages: Sequence[UserMessage] = result.scalars().all()
+            for msg in old_messages:
+                await session.delete(msg)
+            await session.commit()
+
+        texts: List[str] = ["Сообщение 1", "Сообщение 2", "Сообщение 3"]
+
+        for text in texts:
+            await test_db.add_message(user_id, text)
+
+        stmt = select(UserMessage).where(UserMessage.user_id == user_id).order_by(UserMessage.created_at.asc())
+        result = await test_session.execute(stmt)
+        messages: Sequence[UserMessage] = result.scalars().all()
+
+        assert len(messages) == 3
+        assert messages[0].message_text == "Сообщение 1"
+        assert messages[1].message_text == "Сообщение 2"
+        assert messages[2].message_text == "Сообщение 3"