Source code for tests.notifications.factories

from factory import SubFactory
from factory.django import DjangoModelFactory
from faker import Faker

from apps.notifications.constants import NotificationTypeChoices
from apps.notifications.models import Notification
from tests.posts.factories import PostFactory
from tests.users.factories import UserFactory  # type: ignore

fake = Faker()




[docs]
class NotificationFactory(DjangoModelFactory):
    class Meta:
        model = Notification

    sender = SubFactory(UserFactory)
    receiver = SubFactory(UserFactory)
    post = SubFactory(PostFactory)






[docs]
class LikeNotificationFactory(NotificationFactory):
    type = NotificationTypeChoices.LIKE






[docs]
class FollowNotificationFactory(NotificationFactory):
    type = NotificationTypeChoices.FOLLOW






[docs]
class CommentNotificationFactory(NotificationFactory):
    type = NotificationTypeChoices.COMMENT