lesspass/server/api/tests/factories.py

import factory

from api import models


class UserFactory(factory.DjangoModelFactory):
    class Meta:
        model = models.User

    username = factory.Sequence(lambda n: 'username{0}'.format(n))
    first_name = factory.Faker('first_name')
    last_name = factory.Faker('last_name')
    email = factory.LazyAttribute(lambda a: '{0}.{1}@oslab.fr'.format(a.first_name, a.last_name).lower())
    password = factory.PostGenerationMethodCall('set_password', 'password')
    is_staff = False


class AdminFactory(UserFactory):
    is_staff = True


class SiteFactory(factory.DjangoModelFactory):
    class Meta:
        model = models.Site

    user = factory.SubFactory(UserFactory)
    name = 'facebook'
    password_length = 12
    password_type = 'strong'
    counter = 1