django-notifications/notifications/tests/factories/users.py
2023-06-24 00:10:23 +00:00

26 lines
703 B
Python

import factory
from django.conf import settings
class Recipient(factory.django.DjangoModelFactory):
username = factory.Sequence(lambda n: f"recipient-{n}")
first_name = factory.SelfAttribute("username")
class Meta:
model = settings.AUTH_USER_MODEL
class Actor(factory.django.DjangoModelFactory):
username = factory.Sequence(lambda n: f"actor-{n}")
first_name = factory.SelfAttribute("username")
class Meta:
model = settings.AUTH_USER_MODEL
class Target(factory.django.DjangoModelFactory):
username = factory.Sequence(lambda n: f"target-{n}")
first_name = factory.SelfAttribute("username")
class Meta:
model = settings.AUTH_USER_MODEL