import faker
import pytest

from email_campaigns.profile.models import Profile
from email_campaigns.profile.repositories import ProfileRepository

from tests.unit.types import CreateModel


class TestProfileRepository:
    @pytest.mark.db
    def test_find_profiles(
        self,
        repository: ProfileRepository,
        create_model: CreateModel,
        faker: faker.Faker,
    ) -> None:
        profile = create_model(Profile)
        email = faker.email()
        data = repository.find_by_emails({profile.email, email})

        assert data == [profile]

    @pytest.mark.db
    def test_find_profiles_returns_first_name(
        self,
        repository: ProfileRepository,
        create_model: CreateModel,
    ) -> None:
        profile = create_model(Profile, first_name="Anna")
        [found] = repository.find_by_emails({profile.email})

        assert found.first_name == "Anna"