telegram-helper-bot/tests/conftest_post_repository.py

import asyncio
import os
import tempfile
from datetime import datetime
from unittest.mock import AsyncMock, Mock

import pytest

from database.models import MessageContentLink, PostContent, TelegramPost
from database.repositories.post_repository import PostRepository


@pytest.fixture(scope="session")
def event_loop():
    """Создает event loop для асинхронных тестов"""
    loop = asyncio.get_event_loop_policy().new_event_loop()
    yield loop
    loop.close()


@pytest.fixture
def mock_post_repository():
    """Создает мок PostRepository для unit тестов"""
    mock_repo = Mock(spec=PostRepository)
    mock_repo._execute_query = AsyncMock()
    mock_repo._execute_query_with_result = AsyncMock()
    mock_repo.update_status_by_message_id = AsyncMock()
    mock_repo.update_status_for_media_group_by_helper_id = AsyncMock()
    mock_repo.logger = Mock()
    return mock_repo


@pytest.fixture
def sample_telegram_post():
    """Создает тестовый объект TelegramPost"""
    return TelegramPost(
        message_id=12345,
        text="Тестовый пост для unit тестов",
        author_id=67890,
        helper_text_message_id=None,
        created_at=int(datetime.now().timestamp()),
    )


@pytest.fixture
def sample_telegram_post_with_helper():
    """Создает тестовый объект TelegramPost с helper сообщением"""
    return TelegramPost(
        message_id=12346,
        text="Тестовый пост с helper сообщением",
        author_id=67890,
        helper_text_message_id=99999,
        created_at=int(datetime.now().timestamp()),
    )


@pytest.fixture
def sample_telegram_post_no_date():
    """Создает тестовый объект TelegramPost без даты"""
    return TelegramPost(
        message_id=12347,
        text="Тестовый пост без даты",
        author_id=67890,
        helper_text_message_id=None,
        created_at=None,
    )


@pytest.fixture
def sample_post_content():
    """Создает тестовый объект PostContent"""
    return PostContent(
        message_id=12345, content_name="/path/to/test/file.jpg", content_type="photo"
    )


@pytest.fixture
def sample_message_content_link():
    """Создает тестовый объект MessageContentLink"""
    return MessageContentLink(post_id=12345, message_id=67890)


@pytest.fixture
def mock_db_execute_query():
    """Создает мок для _execute_query"""
    return AsyncMock()


@pytest.fixture
def mock_db_execute_query_with_result():
    """Создает мок для _execute_query_with_result"""
    return AsyncMock()


@pytest.fixture
def mock_logger():
    """Создает мок для logger"""
    return Mock()


@pytest.fixture
def temp_db_file():
    """Создает временный файл БД для интеграционных тестов"""
    with tempfile.NamedTemporaryFile(suffix=".db", delete=False) as tmp_file:
        db_path = tmp_file.name

    yield db_path

    # Очищаем временный файл после тестов
    try:
        os.unlink(db_path)
    except OSError:
        pass


@pytest.fixture
def real_post_repository(temp_db_file):
    """Создает реальный PostRepository с временной БД для интеграционных тестов"""
    return PostRepository(temp_db_file)


@pytest.fixture
def sample_posts_batch():
    """Создает набор тестовых постов для batch тестов"""
    return [
        TelegramPost(
            message_id=10001,
            text="Первый тестовый пост",
            author_id=11111,
            helper_text_message_id=None,
            created_at=int(datetime.now().timestamp()),
        ),
        TelegramPost(
            message_id=10002,
            text="Второй тестовый пост",
            author_id=22222,
            helper_text_message_id=None,
            created_at=int(datetime.now().timestamp()),
        ),
        TelegramPost(
            message_id=10003,
            text="Третий тестовый пост",
            author_id=33333,
            helper_text_message_id=88888,
            created_at=int(datetime.now().timestamp()),
        ),
    ]


@pytest.fixture
def sample_content_batch():
    """Создает набор тестового контента для batch тестов"""
    return [
        (10001, "/path/to/photo1.jpg", "photo"),
        (10002, "/path/to/video1.mp4", "video"),
        (10003, "/path/to/audio1.mp3", "audio"),
        (10004, "/path/to/photo2.jpg", "photo"),
        (10005, "/path/to/video2.mp4", "video"),
    ]


@pytest.fixture
def mock_database_connection():
    """Создает мок для DatabaseConnection"""
    mock_conn = Mock()
    mock_conn._execute_query = AsyncMock()
    mock_conn._execute_query_with_result = AsyncMock()
    mock_conn.logger = Mock()
    return mock_conn


@pytest.fixture
def sample_helper_message_ids():
    """Создает набор тестовых helper message ID"""
    return [11111, 22222, 33333, 44444, 55555]


@pytest.fixture
def sample_message_ids():
    """Создает набор тестовых message ID"""
    return [10001, 10002, 10003, 10004, 10005]


@pytest.fixture
def sample_author_ids():
    """Создает набор тестовых author ID"""
    return [11111, 22222, 33333, 44444, 55555]


@pytest.fixture
def mock_sql_queries():
    """Создает мок для SQL запросов"""
    return {
        "create_tables": [
            "CREATE TABLE IF NOT EXISTS post_from_telegram_suggest",
            "CREATE TABLE IF NOT EXISTS content_post_from_telegram",
            "CREATE TABLE IF NOT EXISTS message_link_to_content",
        ],
        "add_post": "INSERT INTO post_from_telegram_suggest",
        "add_post_status": "status",
        "update_helper": "UPDATE post_from_telegram_suggest SET helper_text_message_id",
        "update_status": "UPDATE post_from_telegram_suggest SET status = ?",
        "add_content": "INSERT OR IGNORE INTO content_post_from_telegram",
        "add_link": "INSERT OR IGNORE INTO message_link_to_content",
        "get_content": "SELECT cpft.content_name, cpft.content_type",
        "get_text": "SELECT text FROM post_from_telegram_suggest",
        "get_ids": "SELECT mltc.message_id",
        "get_author": "SELECT author_id FROM post_from_telegram_suggest",
    }