lib/extractor/tests/conftest.py

"""
Конфигурация pytest для тестов ntr_text_fragmentation.
"""

from uuid import UUID

import pytest

from ntr_text_fragmentation.models.linker_entity import LinkerEntity
from tests.custom_entity import CustomEntity  # Импортируем наш кастомный класс


@pytest.fixture
def sample_entity():
    """
    Фикстура, возвращающая экземпляр LinkerEntity с предустановленными значениями.
    """
    return LinkerEntity(
        id=UUID('12345678-1234-5678-1234-567812345678'),
        name="Тестовая сущность",
        text="Текст тестовой сущности",
        metadata={"test_key": "test_value"}
    )


@pytest.fixture
def sample_custom_entity():
    """
    Фикстура, возвращающая экземпляр CustomEntity с предустановленными значениями.
    """
    return CustomEntity(
        id=UUID('87654321-8765-4321-8765-432187654321'),
        name="Тестовый кастомный объект",
        text="Текст кастомного объекта",
        metadata={"original_key": "original_value"},
        in_search_text="Текст для поиска кастомного объекта",
        custom_field1="custom_value",
        custom_field2=42
    )


@pytest.fixture
def sample_link():
    """
    Фикстура, возвращающая экземпляр LinkerEntity с предустановленными значениями связи.
    """
    return LinkerEntity(
        id=UUID('98765432-9876-5432-9876-543298765432'),
        name="Тестовая связь",
        text="Текст тестовой связи",
        metadata={"test_key": "test_value"},
        source_id=UUID('12345678-1234-5678-1234-567812345678'),
        target_id=UUID('87654321-8765-4321-8765-432187654321'),
        type="Link"
    )