mirror of
https://github.com/boostorg/website-v2.git
synced 2026-01-19 04:42:17 +00:00
28 lines
659 B
Python
28 lines
659 B
Python
import factory
|
|
|
|
from django.utils import timezone
|
|
|
|
from .models import User
|
|
|
|
|
|
class UserFactory(factory.django.DjangoModelFactory):
|
|
email = factory.Sequence(lambda n: "user%s@example.com" % n)
|
|
first_name = factory.Sequence(lambda n: "User%s Bob" % n)
|
|
last_name = factory.Sequence(lambda n: "User%s Smith" % n)
|
|
|
|
last_login = factory.LazyFunction(timezone.now)
|
|
|
|
password = factory.PostGenerationMethodCall("set_password", "password")
|
|
|
|
class Meta:
|
|
model = User
|
|
django_get_or_create = ("email",)
|
|
|
|
|
|
class StaffUserFactory(UserFactory):
|
|
is_staff = True
|
|
|
|
|
|
class SuperUserFactory(StaffUserFactory):
|
|
is_superuser = True
|