social-app/backend/tests/unit/services/base/test_supabase_service.py

from __future__ import annotations

from types import SimpleNamespace
from unittest.mock import MagicMock

import pytest

from core.config.settings import SupabaseSettings
from services.base.supabase import SupabaseService


@pytest.mark.asyncio
async def test_initialize_success(monkeypatch: pytest.MonkeyPatch) -> None:
    service = SupabaseService(
        settings=SupabaseSettings(public_url="https://test.supabase.co")
    )
    anon_client = MagicMock()
    admin_client = MagicMock()

    create_calls: list[tuple[str, str]] = []

    def _fake_create_client(url: str, key: str) -> object:
        create_calls.append((url, key))
        return anon_client if len(create_calls) == 1 else admin_client

    monkeypatch.setattr("services.base.supabase.create_client", _fake_create_client)

    result = await service.initialize()

    assert result is True
    assert service.is_initialized is True
    assert service.get_client() is anon_client
    assert service.get_admin_client() is admin_client
    assert len(create_calls) == 2


@pytest.mark.asyncio
async def test_initialize_failure(monkeypatch: pytest.MonkeyPatch) -> None:
    service = SupabaseService(
        settings=SupabaseSettings(public_url="https://test.supabase.co")
    )

    def _fake_create_client(_: str, __: str) -> object:
        raise RuntimeError("boom")

    monkeypatch.setattr("services.base.supabase.create_client", _fake_create_client)

    result = await service.initialize()

    assert result is False
    assert service.is_initialized is False
    with pytest.raises(RuntimeError):
        service.get_client()


@pytest.mark.asyncio
async def test_close_clears_clients(monkeypatch: pytest.MonkeyPatch) -> None:
    service = SupabaseService(
        settings=SupabaseSettings(public_url="https://test.supabase.co")
    )

    def _fake_create_client(_: str, __: str) -> object:
        return MagicMock()

    monkeypatch.setattr("services.base.supabase.create_client", _fake_create_client)

    assert await service.initialize() is True
    assert await service.close() is True
    assert service.is_initialized is False
    assert service.get_client() is not None
    assert service.get_admin_client() is not None


@pytest.mark.asyncio
async def test_health_check_uninitialized() -> None:
    service = SupabaseService(
        settings=SupabaseSettings(public_url="https://test.supabase.co")
    )

    health = await service.health_check()

    assert health["status"] == "unhealthy"


@pytest.mark.asyncio
async def test_health_check_initialized(monkeypatch: pytest.MonkeyPatch) -> None:
    service = SupabaseService(
        settings=SupabaseSettings(public_url="https://test.supabase.co")
    )

    anon_client = MagicMock()
    anon_client.auth.get_session = MagicMock(return_value=None)

    admin_list_users = MagicMock(return_value=SimpleNamespace(users=[]))
    admin_client = MagicMock()
    admin_client.auth.admin = SimpleNamespace(list_users=admin_list_users)

    create_sequence = [anon_client, admin_client]

    def _fake_create_client(_: str, __: str) -> object:
        return create_sequence.pop(0)

    monkeypatch.setattr("services.base.supabase.create_client", _fake_create_client)

    assert await service.initialize() is True

    health = await service.health_check()

    assert health["status"] == "healthy"
    admin_list_users.assert_called_once_with(page=1, per_page=1)


def test_get_client_raises_before_init() -> None:
    service = SupabaseService(
        settings=SupabaseSettings(public_url="https://test.supabase.co")
    )

    assert service.get_client() is not None
    assert service.get_admin_client() is not None


def test_get_client_raises_when_lazy_initialization_fails(
    monkeypatch: pytest.MonkeyPatch,
) -> None:
    service = SupabaseService(
        settings=SupabaseSettings(public_url="https://test.supabase.co")
    )

    def _fake_create_client(_: str, __: str) -> object:
        raise RuntimeError("boom")

    monkeypatch.setattr("services.base.supabase.create_client", _fake_create_client)

    with pytest.raises(RuntimeError):
        service.get_client()
    with pytest.raises(RuntimeError):
        service.get_admin_client()


def test_get_admin_client_lazily_initializes_clients(
    monkeypatch: pytest.MonkeyPatch,
) -> None:
    service = SupabaseService(
        settings=SupabaseSettings(public_url="https://test.supabase.co")
    )
    anon_client = MagicMock(name="anon")
    admin_client = MagicMock(name="admin")
    create_calls: list[tuple[str, str]] = []

    def _fake_create_client(url: str, key: str) -> object:
        create_calls.append((url, key))
        return anon_client if len(create_calls) == 1 else admin_client

    monkeypatch.setattr("services.base.supabase.create_client", _fake_create_client)

    resolved_admin = service.get_admin_client()

    assert resolved_admin is admin_client
    assert service.get_client() is anon_client
    assert service.is_initialized is True
    assert len(create_calls) == 2


def test_validate_bucket_accepts_attachment_and_avatar() -> None:
    service = SupabaseService(
        settings=SupabaseSettings(public_url="https://test.supabase.co")
    )

    service._validate_bucket("agent-chat-attachments")
    service._validate_bucket("avatars")

    with pytest.raises(RuntimeError):
        service._validate_bucket("unexpected-bucket")