How to use FactoryBoy to create model instances in python for testing

Written on January 18, 2018
Categories: Django   | Python   | FactoryBoy |

Testing is one of the basic and most important things that are needed to be done. Also, it is one of the most important developmental phases. I have already discussed testing in one of my earlier posts. In that post, I talked about the way in which we can write tests for models. Today I am going to talk about using FactoryBoy for testing Django application.

The name ‘Factory Boy’ is inspired by the Factory Girl testing tool for Ruby on Rails. Without Factory boy, you have to write a lot of code for the creation of test cases. For writing test, you have to spend a lot of time writing the setup method for the tests.

Without the use of factories, this is how your tests will look like:

from django.test import TestCase
from my_app.models import Person
class DataBaseStore(TestCase):
    def setUp(self):
        # Creating all the models   
        Person.objects.create(gender='male', age=20, name='ranvir')  
    def test_something(self):
        # test all the things that you want to.

On the other hand, it is always nice to separate your setting up stuff from the main stuff. For this sometimes we have to write long code in the beginning which helps us in the future by allowing us to write smaller code in the future.

We can write factories anywhere but it’s always a good idea to write it in a file named inside the root directory of your app.

Write following in the file inside your root directory.

from factory.django import DjangoModelFactory  
from my_app.models import Person  
class PersonFactory(DjangoModelFactory):  
    class Meta:  
        model = Person  
    name = 'ranvir'  
    age = 20  
    gender = 'male'  

Now we can simply call these factories during the test generation and create as many instances of models as we want to. This is a pretty simple case but we really can do a lot more. We can create instances of varies types by making use of FuzzyText and Sequence.

Hope you guys liked the idea of the post. We really can do a lot of stuff using this library. Do share your views regarding this post if you have used factories in the past or want to use it in your project.