Search⌘ K
Home
Courses
Full Stack Django and React

Writing Tests for Django Models

Explore how to create automated tests for Django models such as User, Post, and Comment using pytest and fixtures. This lesson helps you understand the importance of model testing, how to write test functions with decorators, and how fixtures can reduce code repetition for more reliable Django applications.

We'll cover the following...

When applying testing to a Django project, it’s always a good idea to start with writing tests for the models. But why test the models?

Well, it gives us better confidence in our code and the connections to the database. It’ll make sure that methods or attributes on the model are well represented in the database, but it can also help us with code structure, resolving bugs, and building documentation.

Without further ado, let’s start by writing tests for the User model.

Writing tests for the User model

Inside the core/user directory, let’s create a new file called tests.py. We’ll write tests to create a user and a simple user:

Python 3.8
import pytest
from core.user.models import User
data_user = {
    "username": "test_user",
    "email": "test@gmail.com",
    "first_name": "Test",
    "last_name": "User",
    "password": "test_password"
}

Once the imports and the data to create the user have been added, we can write the test function:

Python 3.8
@pytest.mark.django_db
def test_create_user():
    user = User.objects.create_user(**data_user)
    assert user.username == data_user["username"]
    assert user.email == data_user["email"]
    assert user.first_name == data_user["first_name"]
    assert user.last_name == data_user["last_name"]
...