Source code for profiles.factories

__copyright__ = "Copyright © Stichting SciPost (SciPost Foundation)"
__license__ = "AGPL v3"


import factory

from .models import Profile

from scipost.constants import TITLE_CHOICES, SCIPOST_DISCIPLINES

class ProfileFactory(factory.django.DjangoModelFactory):
    title = factory.Iterator(TITLE_CHOICES, getter=lambda c: c[0])
    first_name = factory.Faker('first_name')
    last_name = factory.Faker('last_name')
    discipline = factory.Iterator(SCIPOST_DISCIPLINES[2][1], getter=lambda c: c[0])

    class Meta:
        model = Profile