import pytest

from dmp.tiktok.exceptions import TikTokConnectionNotFoundError
from dmp.tiktok.handlers import (
    GetTikTokUserAdAccountsHandler,
    GetTikTokUserAdAccountsRequest,
)
from dmp.tiktok.models import TikTokUserConnection

from tests.unit.types import CreateModel


class TestGetTikTokUserAdAccountsHandler:
    @pytest.mark.db
    def test_get_user_ad_accounts_empty_list(
        self,
        handler: GetTikTokUserAdAccountsHandler,
        create_model: CreateModel,
        identity_id: str,
        user_id: str,
    ) -> None:
        create_model(
            TikTokUserConnection,
            identity_id=identity_id,
            user_id=user_id,
            ad_accounts=[],
        )
        ad_accounts = handler.handle(
            GetTikTokUserAdAccountsRequest(identity_id=identity_id, user_id=user_id)
        )

        assert ad_accounts == []

    @pytest.mark.db
    def test_get_user_ad_accounts_missed_connection(
        self, handler: GetTikTokUserAdAccountsHandler, identity_id: str, user_id: str
    ) -> None:
        with pytest.raises(TikTokConnectionNotFoundError):
            handler.handle(
                GetTikTokUserAdAccountsRequest(
                    identity_id=identity_id, user_id=user_id
                ),
            )