# tests/realtime/conftest.py
"""
Pytest fixtures для тестирования WebSocket consumers.
"""

import pytest
from channels.testing import WebsocketCommunicator
from django.contrib.auth import get_user_model
from django.contrib.auth.models import AnonymousUser

from communications.models import Chat, ChatMembership
from realtime.consumers import UserConsumer

User = get_user_model()


@pytest.fixture
def user(db):
    """Создает тестового пользователя."""
    return User.objects.create_user(
        email="testuser@example.com",
        password="testpass123",
        phone_number="+79991234567",
        first_name="Test",
        last_name="User",
    )


@pytest.fixture
def user2(db):
    """Создает второго тестового пользователя."""
    return User.objects.create_user(
        email="testuser2@example.com",
        password="testpass123",
        phone_number="+79991234568",
        first_name="Test2",
        last_name="User2",
    )


@pytest.fixture
def ws_communicator():
    """
    Фабрика для создания WebSocket communicator.

    Использование:
        communicator = await ws_communicator(user=test_user)
        connected, _ = await communicator.connect()
    """

    async def _create_communicator(user=None, path="/ws/"):
        """Создает WebsocketCommunicator с указанным пользователем."""
        application = UserConsumer.as_asgi()

        # Создаем scope с пользователем
        scope = {
            "type": "websocket",
            "path": path,
            "headers": [],
            "query_string": b"",
            "user": user if user else AnonymousUser(),
        }

        communicator = WebsocketCommunicator(application, path)
        communicator.scope.update(scope)

        return communicator

    return _create_communicator


@pytest.fixture
def test_chat(db, user):
    """Создает тестовый чат с пользователем."""
    chat = Chat.objects.create(name="Test Chat", type="group")
    ChatMembership.objects.create(chat=chat, user=user, role="member")
    return chat


@pytest.fixture
def test_chat_with_users(db, user, user2):
    """Создает чат с двумя пользователями."""
    chat = Chat.objects.create(name="Multi User Chat", type="group")
    ChatMembership.objects.create(chat=chat, user=user, role="admin")
    ChatMembership.objects.create(chat=chat, user=user2, role="member")
    return chat