import factory from django.conf import settings from django.apps import apps from .models import Notification from django.db.models.signals import post_save @factory.django.mute_signals(post_save) class NotificationFactory(factory.django.DjangoModelFactory): class Meta: model = Notification type = factory.Faker('text', max_nb_chars=50) summary = factory.Faker('paragraph', nb_sentences=3, variable_nb_sentences=True) author = factory.Faker('url') user = factory.Iterator(apps.get_model(settings.AUTH_USER_MODEL).objects.all()) date = factory.Faker('past_datetime') unread = factory.Faker('boolean') object = factory.Faker('url')