You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
23 lines
638 B
23 lines
638 B
# -*- coding: utf-8 -*-
|
|
from factory import Sequence, PostGenerationMethodCall
|
|
from factory.alchemy import SQLAlchemyModelFactory
|
|
|
|
from {{cookiecutter.app_name}}.user.models import User
|
|
from {{cookiecutter.app_name}}.database import db
|
|
|
|
class BaseFactory(SQLAlchemyModelFactory):
|
|
|
|
class Meta:
|
|
abstract = True
|
|
sqlalchemy_session = db.session
|
|
|
|
|
|
class UserFactory(BaseFactory):
|
|
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:
|
|
model = User
|
|
|
|
|