Testing Django Models Example

tests.py

from datetime import datetime

from django.db import IntegrityError
from django.test import TestCase
from django.urls import reverse
from django.utils.text import slugify

from .models import Genre, Movie


class TestMovie(TestCase):

    MOVIE_TITLE = "test movie"
    RELEASE_DATE = datetime.strptime("2022-01-06", "%Y-%m-%d")

    def setUp(self):
        self.movie = Movie.objects.create(
            title=self.MOVIE_TITLE,
            release_date=self.RELEASE_DATE,
            api_id=1
        )

    def test_instance(self):
        self.assertEqual(self.movie.title, self.MOVIE_TITLE)
        self.assertEqual(self.movie.api_id, 1)

    def test_unique_api_id_is_enforced(self):
        """ Test that two movies with same api_id are not allowed."""
        with self.assertRaises(IntegrityError):
            Movie.objects.create(
                title="another movie",
                release_date=self.RELEASE_DATE,
                api_id=1
            )

    def test_slug_value(self):
        """
        Test that the slug is automatically added to the movie on creation.
        """
        expected = slugify(self.movie.title)
        actual = self.movie.slug
        self.assertEqual(expected, actual)

    def test_slug_value_for_duplicate_title(self):
        """
        Test that two movies with identical titles get unique slugs.
        """
        movie2 = Movie.objects.create(
            title=self.movie.title,
            release_date=self.RELEASE_DATE,
            api_id=99
        )

        self.assertNotEqual(self.movie.slug, movie2.slug)

    def test_added_date_automatically(self):
        """ Test that the date is automatically saved on creation"""
        self.assertTrue(type(self.movie.added), datetime)

    def test_active_false_by_default(self):
        """ Test that our booleans are set to false by default"""
        self.assertTrue(type(self.movie.active) == bool)
        self.assertFalse(self.movie.active)

    def test_deleted_false_by_default(self):
        """ Test that our booleans are set to false by default"""
        self.assertTrue(type(self.movie.deleted) == bool)
        self.assertFalse(self.movie.deleted)

    def test_str(self):
        """ Test the __str__ method"""
        expected = "test movie"
        actual = str(self.movie)

        self.assertEqual(expected, actual)

    def test_get_absolute_url(self):
        """ Test that get_absolute_url returns the expected URL"""

        expected = reverse("movie_detail", kwargs={"slug": self.movie.slug})
        actual = self.movie.get_absolute_url()

        self.assertEqual(expected, actual)


class TestGenre(TestCase):

    @classmethod
    def setUpTestData(cls):
        cls.genre = Genre.objects.create(name="Western", api_id=1)

    def test_has_name(self):
        self.assertEqual(self.genre.name, "Western")

    def test_has_api_id(self):
        self.assertEqual(self.genre.api_id, 1)

    def test_str(self):
        expected = "Western"
        actual = str(self.genre)

        self.assertEqual(expected, actual)