Skip to content
Snippets Groups Projects
Commit e589ec1d authored by sloria's avatar sloria
Browse files

Add abstract base factory

parent 84123762
No related branches found
No related tags found
No related merge requests found
...@@ -5,13 +5,19 @@ from factory.alchemy import SQLAlchemyModelFactory ...@@ -5,13 +5,19 @@ from factory.alchemy import SQLAlchemyModelFactory
from {{cookiecutter.app_name}}.user.models import User from {{cookiecutter.app_name}}.user.models import User
from {{cookiecutter.app_name}}.database import db from {{cookiecutter.app_name}}.database import db
class BaseFactory(SQLAlchemyModelFactory):
class UserFactory(SQLAlchemyModelFactory):
class Meta: class Meta:
model = User abstract = True
sqlalchemy_session = db.session sqlalchemy_session = db.session
class UserFactory(BaseFactory):
username = Sequence(lambda n: "user{0}".format(n)) username = Sequence(lambda n: "user{0}".format(n))
email = Sequence(lambda n: "user{0}@example.com".format(n)) email = Sequence(lambda n: "user{0}@example.com".format(n))
password = PostGenerationMethodCall('set_password', 'example') password = PostGenerationMethodCall('set_password', 'example')
active = True active = True
class Meta:
model = User
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment