Full Stack Django and React/

...

/

Writing Tests for Django Models

Writing Tests for Django Models

Write tests for the User, Post, and Comment models using Pytest.

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:

Press + to interact
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:

Press + to interact
@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"]
...