from collections.abc import Callable
from uuid import uuid4

import pytest

from generalresearch.managers.gr.authentication import GRTokenManager, GRUserManager
from generalresearch.models.gr.authentication import GRToken, GRUser
from generalresearch.pg_helper import PostgresConfig
from generalresearch.redis_helper import RedisConfig

SSO_ISSUER = ""


class TestGRUserManager:
    def test_create(self, gr_user: GRUser, gr_user_manager: GRUserManager):
        assert gr_user.id is not None
        instance = gr_user_manager.get_by_id(gr_user.id)
        assert isinstance(instance, GRUser)
        assert gr_user.id == instance.id

        assert gr_user.id is not None
        instance2 = gr_user_manager.get_by_id(gr_user.id)
        assert isinstance(instance2, GRUser)
        assert gr_user.model_dump_json() == instance2.model_dump_json()

    def test_get_by_id(self, gr_user: GRUser, gr_user_manager: GRUserManager):
        with pytest.raises(expected_exception=ValueError) as cm:
            gr_user_manager.get_by_id(gr_user_id=999_999_999)
        assert "GRUser not found" in str(cm.value)
        assert gr_user.id is not None
        instance = gr_user_manager.get_by_id(gr_user_id=gr_user.id)
        assert isinstance(instance, GRUser)
        assert instance.email == gr_user.email

    def test_get_by_email(
        self,
        gr_user_factory: Callable[..., GRUser],
        gr_user_manager: GRUserManager,
    ):
        email = f"{uuid4().hex}@example.com"
        gr_user = gr_user_factory(email=email)

        with pytest.raises(expected_exception=ValueError, match="GRUser not found"):
            gr_user_manager.get_by_email(email="missing@example.com")

        instance = gr_user_manager.get_by_email(email=email)
        assert isinstance(instance, GRUser)
        assert instance.id == gr_user.id

    def test_get_all(
        self, gr_user_factory: Callable[..., GRUser], gr_user_manager: GRUserManager
    ):
        res1 = gr_user_manager.get_all()
        assert isinstance(res1, list)

        gr_user_factory(save=True)
        res2 = gr_user_manager.get_all()
        assert len(res1) == len(res2) - 1

    def test_get_by_team(self, gr_user_manager: GRUserManager):
        res = gr_user_manager.get_by_team(team_id=999_999_999)
        assert isinstance(res, list)
        assert res == []


class TestGRTokenManager:
    def test_create(self, gr_user: GRUser, gr_token_manager: GRTokenManager):
        assert gr_user.id is not None
        assert gr_token_manager.create(user_id=gr_user.id) is None

        token = gr_token_manager.get_by_user_id(user_id=gr_user.id)
        assert isinstance(token, GRToken)
        assert gr_user.id == token.user_id

    def test_get_by_user_id(self, gr_user: GRUser, gr_token_manager: GRTokenManager):
        assert gr_user.id is not None
        assert gr_token_manager.create(user_id=gr_user.id) is None

        token = gr_token_manager.get_by_user_id(user_id=gr_user.id)
        assert isinstance(token, GRToken)
        assert gr_user.id == token.user_id

    def test_prefetch_user(
        self,
        gr_user: GRUser,
        gr_token_manager: GRTokenManager,
        gr_db: PostgresConfig,
        gr_redis_config: RedisConfig,
    ):
        assert gr_user.id is not None
        gr_token_manager.create(user_id=gr_user.id)

        token: GRToken | None = gr_token_manager.get_by_user_id(user_id=gr_user.id)
        assert isinstance(token, GRToken)
        assert token.user is None

        token.prefetch_user(pg_config=gr_db, redis_config=gr_redis_config)
        assert token.user.id == gr_user.id

    def test_get_by_key(
        self,
        gr_user: GRUser,
        gr_token_manager: GRTokenManager,
    ):
        assert gr_user.id is not None
        gr_token_manager.create(user_id=gr_user.id)
        token = gr_token_manager.get_by_user_id(user_id=gr_user.id)
        assert isinstance(token, GRToken)

        instance = gr_token_manager.get_by_key(api_key=token.key)
        assert token.created == instance.created

        # Search for non-existent key
        with pytest.raises(expected_exception=Exception) as cm:
            gr_token_manager.get_by_key(api_key=uuid4().hex)
        assert "No GRUser with token of " in str(cm.value)