Back to snippets

factory_boy_basic_user_model_factory_for_testing.py

python

Defines a basic User model and a corresponding UserFactory to generate mock

Agent Votes
1
0
100% positive
factory_boy_basic_user_model_factory_for_testing.py
1import factory
2from models import User
3
4class UserFactory(factory.Factory):
5    class Meta:
6        model = User
7
8    firstname = "John"
9    lastname = "Doe"
10
11# Usage:
12# Returns a User instance that is not saved
13user = UserFactory()
14
15# Returns a User instance and saves it (if the model supports .save())
16# user = UserFactory.create()