From 7915c66079c6115e759b8a564f0267e7ed93d49f Mon Sep 17 00:00:00 2001 From: Alvaro Mariano Date: Fri, 16 Jun 2023 01:31:19 +0000 Subject: [PATCH] Add user factories --- notifications/tests/factories/__init__.py | 0 notifications/tests/factories/users.py | 26 +++++++++++++++++++++++ 2 files changed, 26 insertions(+) create mode 100644 notifications/tests/factories/__init__.py create mode 100644 notifications/tests/factories/users.py diff --git a/notifications/tests/factories/__init__.py b/notifications/tests/factories/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/notifications/tests/factories/users.py b/notifications/tests/factories/users.py new file mode 100644 index 0000000..c53fa4d --- /dev/null +++ b/notifications/tests/factories/users.py @@ -0,0 +1,26 @@ +from django.conf import settings +import factory + + +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