import pytest

from preference_center.profile.models import Profile, Subscription
from preference_center.profile.repositories import SubscriptionRepository

from tests.unit.types import CreateModel


class TestSubscriptionRepository:
    @pytest.mark.db
    def test_find_by_profile_id_empty(
        self,
        repository: SubscriptionRepository,
    ) -> None:
        subscriptions = repository.find_by_profile_id("not-exist")
        assert subscriptions == []

    @pytest.mark.db
    def test_find_by_profile_id(
        self,
        repository: SubscriptionRepository,
        create_model: CreateModel,
    ) -> None:
        profile = create_model(Profile)
        subscription = create_model(Subscription, profile_id=profile.id)

        subscriptions = repository.find_by_profile_id(profile.id)

        assert subscriptions == [subscription]