import pytest

from dmp.meta.exceptions import MetaConnectionNotFoundError
from dmp.meta.handlers import GetMetaUserAdAccountsHandler, GetMetaUserAdAccountsRequest
from dmp.meta.models import MetaUserConnection

from tests.unit.types import CreateModel


class TestGetMetaUserAdAccountsHandler:
    @pytest.mark.db
    def test_get_user_ad_accounts(
        self,
        handler: GetMetaUserAdAccountsHandler,
        create_model: CreateModel,
        identity_id: str,
    ) -> None:
        create_model(MetaUserConnection, identity_id=identity_id, ad_accounts=[])

        ad_accounts = handler.handle(
            GetMetaUserAdAccountsRequest(identity_id=identity_id),
        )

        assert ad_accounts == []

    @pytest.mark.db
    def test_get_user_ad_accounts_missed_connection(
        self, handler: GetMetaUserAdAccountsHandler, identity_id: str
    ) -> None:
        with pytest.raises(MetaConnectionNotFoundError):
            handler.handle(
                GetMetaUserAdAccountsRequest(identity_id=identity_id),
            )