from typing import Optional

import pytest

import ormar
from tests.lifespan import init_tests
from tests.settings import create_config

base_ormar_config = create_config()


class User(ormar.Model):
    ormar_config = base_ormar_config.copy(tablename="test_users")

    id: int = ormar.Integer(primary_key=True)
    name: str = ormar.String(max_length=50)


class Signup(ormar.Model):
    ormar_config = base_ormar_config.copy(tablename="test_signup")

    id: int = ormar.Integer(primary_key=True)


class Session(ormar.Model):
    ormar_config = base_ormar_config.copy(tablename="test_sessions")

    id: int = ormar.Integer(primary_key=True)
    name: str = ormar.String(max_length=255, index=True)
    some_text: str = ormar.Text()
    some_other_text: Optional[str] = ormar.Text(nullable=True)
    students: Optional[list[User]] = ormar.ManyToMany(User, through=Signup)


create_test_database = init_tests(base_ormar_config)


@pytest.mark.asyncio
async def test_list_sessions_for_user():
    async with base_ormar_config.database:
        for user_id in [1, 2, 3, 4, 5]:
            await User.objects.create(name=f"User {user_id}")

        for name, some_text, some_other_text in [
            ("Session 1", "Some text 1", "Some other text 1"),
            ("Session 2", "Some text 2", "Some other text 2"),
            ("Session 3", "Some text 3", "Some other text 3"),
            ("Session 4", "Some text 4", "Some other text 4"),
            ("Session 5", "Some text 5", "Some other text 5"),
        ]:
            await Session(
                name=name, some_text=some_text, some_other_text=some_other_text
            ).save()

        s1 = await Session.objects.get(pk=1)
        s2 = await Session.objects.get(pk=2)

        users = {}
        for i in range(1, 6):
            user = await User.objects.get(pk=i)
            users[f"user_{i}"] = user
            if i % 2 == 0:
                await s1.students.add(user)
            else:
                await s2.students.add(user)

        assert len(s1.students) == 2
        assert len(s2.students) == 3

        assert [x.pk for x in s1.students] == [2, 4]
        assert [x.pk for x in s2.students] == [1, 3, 5]

        user = await User.objects.select_related("sessions").get(pk=1)

        assert user.sessions is not None
        assert len(user.sessions) > 0
