import faker import pytest from email_campaigns.campaigns.handlers import ( GetCampaignHandler, GetCampaignRequest, ) from email_campaigns.campaigns.models import EmailCampaign from tests.unit.types import CreateModel class TestGetCampaignHandler: @pytest.mark.db def test_get_email_campaign( self, handler: GetCampaignHandler, create_model: CreateModel, faker: faker.Faker, identity_id: str, ) -> None: campaign = create_model( EmailCampaign, sender_name=faker.name(), subject=faker.sentence(), preview_text=faker.sentence(), email_domain=None, email_username=faker.user_name(), ) result = handler.handle( GetCampaignRequest( identity_id=identity_id, campaign_id=campaign.id, ) ) assert result == campaign