import pytest

from email_campaigns.campaigns.exceptions import EmailCampaignRequiredFieldsError
from email_campaigns.campaigns.handlers import (
    GetCampaignInternalHandler,
    GetCampaignInternalRequest,
)
from email_campaigns.campaigns.models import EmailCampaign

from tests.unit.types import CreateModel


class TestGetCampaignInternalHandler:
    @pytest.mark.db
    def test_get_campaign_without_domain(
        self,
        handler: GetCampaignInternalHandler,
        create_model: CreateModel,
    ) -> None:
        campaign = create_model(EmailCampaign, email_domain=None)

        with pytest.raises(EmailCampaignRequiredFieldsError):
            handler.handle(
                GetCampaignInternalRequest(
                    campaign_id=campaign.id,
                )
            )