29 lines
695 B
Python
29 lines
695 B
Python
# -*- coding: utf-8 -*-
|
|
"""Factories to help in tests."""
|
|
from factory import PostGenerationMethodCall, Sequence
|
|
from factory.alchemy import SQLAlchemyModelFactory
|
|
|
|
from grant.app import db
|
|
|
|
|
|
class BaseFactory(SQLAlchemyModelFactory):
|
|
"""Base factory."""
|
|
|
|
class Meta:
|
|
"""Factory configuration."""
|
|
|
|
abstract = True
|
|
sqlalchemy_session = db.session
|
|
|
|
|
|
class UserFactory(BaseFactory):
|
|
"""User factory."""
|
|
|
|
username = Sequence(lambda n: 'user{0}'.format(n))
|
|
email = Sequence(lambda n: 'user{0}@example.com'.format(n))
|
|
password = PostGenerationMethodCall('set_password', 'example')
|
|
active = True
|
|
|
|
class Meta:
|
|
"""Factory configuration."""
|