Django Simple Factory

A simple factory implementation built upon Faker.

Installation

pip install django-simple-factory

Usage

To use factories, create a factories.py inside of a Django app. Then subclass the django_simple_factory.factories.Factory class. To define the model generation, override the definition method.

from django_simple_factory.factories import Factory
from posts import models

class PostFactory(Factory):
    model = models.Post

    def definition(self):
        return {
            "title": self.faker.word(),
            "content": self.faker.text(),
        }

class CommentFactory(Factory):
    model = "comments.Comment" # You can provide a model string

    def definition(self):
        return {
            "content": self.faker.text(),
            "post": PostFactory() # You can provide either a factory instance or a factory string (e.g)
        }

To use these models, instaniate the factories, then call either make or create.

post_factory = PostFactory()
post_factory.make() # Does not persist
post_factory.create() # Does Persist

Factories that have related factories, say a Comment requires a Post, can pass a new dictionary for the post values to override.

comment_factory = CommentFactory()
comment = comment_factory.make(content="Hello", post={"title": "My Post"})

comment.content # Hello
comment.post.title # My Post

Alternatively, you may use the "django-style" of post__title="My Title".

comment_factory = CommentFactory()
comment = comment_factory.make(content="Hello", post__title="My Post")

comment.content # Hello
comment.post.title # My Post

To create related models (those that have a foreign key to the curren model) use the has method. Use the related_name to specify. You can optionally provide arguments and an amount to generate.

post_factory = PostFactory()
post = post_factory.has("comments").create() # Creates a Post with a related comment.

To enable easier testing, a FactoryTestMixin has been included that enables rich definition of factories.

from django_simple_factory.mixins import FactoryTestMixin
from django.test import TestCase
from posts import models

class PostTest(FactoryTestMixin, TestCase):
    factories = ["posts.PostFactory"]

    def test_post_creation(self):
        post = self.factories[models.Post].create()
        ...

API

django_simple_factory.factories.Factory

The primary class of the package, factories allow for simple and easy testing of Django models.

Contact

If there are any issues, please feel free to make an issue. If you have suggested improvements, please make an issue where we can discuss.